2021-12-07 07:33:52 +00:00
package query
import (
"database/sql"
"database/sql/driver"
"errors"
"fmt"
"regexp"
"testing"
2022-04-26 23:01:45 +00:00
"github.com/zitadel/zitadel/internal/domain"
2021-12-07 07:33:52 +00:00
)
var (
2022-11-30 16:01:17 +00:00
idpUserLinksQuery = regexp . QuoteMeta ( ` SELECT projections.idp_user_links3.idp_id, ` +
` projections.idp_user_links3.user_id, ` +
2023-03-07 10:35:47 +00:00
` projections.idp_templates3.name, ` +
2022-11-30 16:01:17 +00:00
` projections.idp_user_links3.external_user_id, ` +
` projections.idp_user_links3.display_name, ` +
2023-03-07 10:35:47 +00:00
` projections.idp_templates3.type, ` +
2022-11-30 16:01:17 +00:00
` projections.idp_user_links3.resource_owner, ` +
2021-12-07 07:33:52 +00:00
` COUNT(*) OVER () ` +
2022-11-30 16:01:17 +00:00
` FROM projections.idp_user_links3 ` +
2023-03-07 10:35:47 +00:00
` LEFT JOIN projections.idp_templates3 ON projections.idp_user_links3.idp_id = projections.idp_templates3.id AND projections.idp_user_links3.instance_id = projections.idp_templates3.instance_id ` +
2023-02-27 21:36:43 +00:00
` AS OF SYSTEM TIME '-1 ms' ` )
2021-12-08 13:49:19 +00:00
idpUserLinksCols = [ ] string {
2021-12-07 07:33:52 +00:00
"idp_id" ,
"user_id" ,
"name" ,
"external_user_id" ,
"display_name" ,
"type" ,
2022-01-13 07:58:14 +00:00
"resource_owner" ,
2021-12-07 07:33:52 +00:00
"count" ,
}
)
2021-12-08 13:49:19 +00:00
func Test_IDPUserLinkPrepares ( t * testing . T ) {
2021-12-07 07:33:52 +00:00
type want struct {
sqlExpectations sqlExpectation
err checkErr
}
tests := [ ] struct {
name string
prepare interface { }
want want
object interface { }
} {
{
name : "prepareIDPsQuery found" ,
2021-12-08 13:49:19 +00:00
prepare : prepareIDPUserLinksQuery ,
2021-12-07 07:33:52 +00:00
want : want {
sqlExpectations : mockQueries (
2021-12-08 13:49:19 +00:00
idpUserLinksQuery ,
idpUserLinksCols ,
2021-12-07 07:33:52 +00:00
[ ] [ ] driver . Value {
{
"idp-id" ,
"user-id" ,
"idp-name" ,
"external-user-id" ,
"display-name" ,
2023-02-28 20:20:58 +00:00
domain . IDPTypeJWT ,
2022-01-13 07:58:14 +00:00
"ro" ,
2021-12-07 07:33:52 +00:00
} ,
} ,
) ,
} ,
2021-12-08 13:49:19 +00:00
object : & IDPUserLinks {
2021-12-07 07:33:52 +00:00
SearchResponse : SearchResponse {
Count : 1 ,
} ,
2021-12-08 13:49:19 +00:00
Links : [ ] * IDPUserLink {
2021-12-07 07:33:52 +00:00
{
IDPID : "idp-id" ,
UserID : "user-id" ,
IDPName : "idp-name" ,
ProvidedUserID : "external-user-id" ,
ProvidedUsername : "display-name" ,
2023-02-28 20:20:58 +00:00
IDPType : domain . IDPTypeJWT ,
2022-01-13 07:58:14 +00:00
ResourceOwner : "ro" ,
2021-12-07 07:33:52 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareIDPsQuery no idp" ,
2021-12-08 13:49:19 +00:00
prepare : prepareIDPUserLinksQuery ,
2021-12-07 07:33:52 +00:00
want : want {
sqlExpectations : mockQueries (
2021-12-08 13:49:19 +00:00
idpUserLinksQuery ,
idpUserLinksCols ,
2021-12-07 07:33:52 +00:00
[ ] [ ] driver . Value {
{
"idp-id" ,
"user-id" ,
nil ,
"external-user-id" ,
"display-name" ,
nil ,
2022-01-13 07:58:14 +00:00
"ro" ,
2021-12-07 07:33:52 +00:00
} ,
} ,
) ,
} ,
2021-12-08 13:49:19 +00:00
object : & IDPUserLinks {
2021-12-07 07:33:52 +00:00
SearchResponse : SearchResponse {
Count : 1 ,
} ,
2021-12-08 13:49:19 +00:00
Links : [ ] * IDPUserLink {
2021-12-07 07:33:52 +00:00
{
IDPID : "idp-id" ,
UserID : "user-id" ,
IDPName : "" ,
ProvidedUserID : "external-user-id" ,
ProvidedUsername : "display-name" ,
2023-02-28 20:20:58 +00:00
IDPType : domain . IDPTypeUnspecified ,
2022-01-13 07:58:14 +00:00
ResourceOwner : "ro" ,
2021-12-07 07:33:52 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareIDPsQuery sql err" ,
2021-12-08 13:49:19 +00:00
prepare : prepareIDPUserLinksQuery ,
2021-12-07 07:33:52 +00:00
want : want {
sqlExpectations : mockQueryErr (
2021-12-08 13:49:19 +00:00
idpUserLinksQuery ,
2021-12-07 07:33:52 +00:00
sql . ErrConnDone ,
) ,
err : func ( err error ) ( error , bool ) {
if ! errors . Is ( err , sql . ErrConnDone ) {
return fmt . Errorf ( "err should be sql.ErrConnDone got: %w" , err ) , false
}
return nil , true
} ,
} ,
object : nil ,
} ,
}
for _ , tt := range tests {
t . Run ( tt . name , func ( t * testing . T ) {
2023-02-27 21:36:43 +00:00
assertPrepare ( t , tt . prepare , tt . object , tt . want . sqlExpectations , tt . want . err , defaultPrepareArgs ... )
2021-12-07 07:33:52 +00:00
} )
}
}