2022-01-14 09:45:50 +00:00
package query
import (
"database/sql"
"database/sql/driver"
"errors"
"fmt"
"regexp"
"testing"
2022-08-31 07:52:43 +00:00
"github.com/zitadel/zitadel/internal/database"
2022-04-26 23:01:45 +00:00
"github.com/zitadel/zitadel/internal/domain"
errs "github.com/zitadel/zitadel/internal/errors"
2022-01-14 09:45:50 +00:00
)
var (
userGrantStmt = regexp . QuoteMeta (
2022-08-31 07:52:43 +00:00
"SELECT projections.user_grants2.id" +
", projections.user_grants2.creation_date" +
", projections.user_grants2.change_date" +
", projections.user_grants2.sequence" +
", projections.user_grants2.grant_id" +
", projections.user_grants2.roles" +
", projections.user_grants2.state" +
", projections.user_grants2.user_id" +
2022-11-07 09:33:22 +00:00
", projections.users5.username" +
", projections.users5.type" +
", projections.users5.resource_owner" +
", projections.users5_humans.first_name" +
", projections.users5_humans.last_name" +
", projections.users5_humans.email" +
", projections.users5_humans.display_name" +
", projections.users5_humans.avatar_key" +
2022-03-24 13:00:24 +00:00
", projections.login_names.login_name" +
2022-08-31 07:52:43 +00:00
", projections.user_grants2.resource_owner" +
2022-03-23 08:02:39 +00:00
", projections.orgs.name" +
", projections.orgs.primary_domain" +
2022-08-31 07:52:43 +00:00
", projections.user_grants2.project_id" +
", projections.projects2.name" +
" FROM projections.user_grants2" +
2022-11-07 09:33:22 +00:00
" LEFT JOIN projections.users5 ON projections.user_grants2.user_id = projections.users5.id AND projections.user_grants2.instance_id = projections.users5.instance_id" +
" LEFT JOIN projections.users5_humans ON projections.user_grants2.user_id = projections.users5_humans.user_id AND projections.user_grants2.instance_id = projections.users5_humans.instance_id" +
2022-10-27 06:08:36 +00:00
" LEFT JOIN projections.orgs ON projections.user_grants2.resource_owner = projections.orgs.id AND projections.user_grants2.instance_id = projections.orgs.instance_id" +
" LEFT JOIN projections.projects2 ON projections.user_grants2.project_id = projections.projects2.id AND projections.user_grants2.instance_id = projections.projects2.instance_id" +
" LEFT JOIN projections.login_names ON projections.user_grants2.user_id = projections.login_names.user_id AND projections.user_grants2.instance_id = projections.login_names.instance_id" +
2022-03-24 13:00:24 +00:00
" WHERE projections.login_names.is_primary = $1" )
2022-01-14 09:45:50 +00:00
userGrantCols = [ ] string {
"id" ,
"creation_date" ,
"change_date" ,
"sequence" ,
"grant_id" ,
"roles" ,
"state" ,
"user_id" ,
"username" ,
"type" ,
"resource_owner" , //user resource owner
"first_name" ,
"last_name" ,
"email" ,
"display_name" ,
"avatar_key" ,
2022-03-24 13:00:24 +00:00
"login_name" ,
2022-01-14 09:45:50 +00:00
"resource_owner" , //user_grant resource owner
"name" , //org name
"primary_domain" ,
"project_id" ,
"name" , //project name
}
userGrantsStmt = regexp . QuoteMeta (
2022-08-31 07:52:43 +00:00
"SELECT projections.user_grants2.id" +
", projections.user_grants2.creation_date" +
", projections.user_grants2.change_date" +
", projections.user_grants2.sequence" +
", projections.user_grants2.grant_id" +
", projections.user_grants2.roles" +
", projections.user_grants2.state" +
", projections.user_grants2.user_id" +
2022-11-07 09:33:22 +00:00
", projections.users5.username" +
", projections.users5.type" +
", projections.users5.resource_owner" +
", projections.users5_humans.first_name" +
", projections.users5_humans.last_name" +
", projections.users5_humans.email" +
", projections.users5_humans.display_name" +
", projections.users5_humans.avatar_key" +
2022-03-24 13:00:24 +00:00
", projections.login_names.login_name" +
2022-08-31 07:52:43 +00:00
", projections.user_grants2.resource_owner" +
2022-03-23 08:02:39 +00:00
", projections.orgs.name" +
", projections.orgs.primary_domain" +
2022-08-31 07:52:43 +00:00
", projections.user_grants2.project_id" +
", projections.projects2.name" +
2022-01-14 09:45:50 +00:00
", COUNT(*) OVER ()" +
2022-08-31 07:52:43 +00:00
" FROM projections.user_grants2" +
2022-11-07 09:33:22 +00:00
" LEFT JOIN projections.users5 ON projections.user_grants2.user_id = projections.users5.id AND projections.user_grants2.instance_id = projections.users5.instance_id" +
" LEFT JOIN projections.users5_humans ON projections.user_grants2.user_id = projections.users5_humans.user_id AND projections.user_grants2.instance_id = projections.users5_humans.instance_id" +
2022-10-27 06:08:36 +00:00
" LEFT JOIN projections.orgs ON projections.user_grants2.resource_owner = projections.orgs.id AND projections.user_grants2.instance_id = projections.orgs.instance_id" +
" LEFT JOIN projections.projects2 ON projections.user_grants2.project_id = projections.projects2.id AND projections.user_grants2.instance_id = projections.projects2.instance_id" +
" LEFT JOIN projections.login_names ON projections.user_grants2.user_id = projections.login_names.user_id AND projections.user_grants2.instance_id = projections.login_names.instance_id" +
2022-03-24 13:00:24 +00:00
" WHERE projections.login_names.is_primary = $1" )
2022-01-14 09:45:50 +00:00
userGrantsCols = append (
userGrantCols ,
"count" ,
)
)
func Test_UserGrantPrepares ( t * testing . T ) {
type want struct {
sqlExpectations sqlExpectation
err checkErr
}
tests := [ ] struct {
name string
prepare interface { }
want want
object interface { }
} {
{
name : "prepareUserGrantQuery no result" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantStmt ,
nil ,
nil ,
) ,
err : func ( err error ) ( error , bool ) {
if ! errs . IsNotFound ( err ) {
return fmt . Errorf ( "err should be zitadel.NotFoundError got: %w" , err ) , false
}
return nil , true
} ,
} ,
object : ( * UserGrant ) ( nil ) ,
} ,
{
name : "prepareUserGrantQuery found" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQuery (
userGrantStmt ,
userGrantCols ,
[ ] driver . Value {
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
) ,
} ,
object : & UserGrant {
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
{
name : "prepareUserGrantQuery machine user found" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQuery (
userGrantStmt ,
userGrantCols ,
[ ] driver . Value {
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeMachine ,
"resource-owner" ,
nil ,
nil ,
nil ,
nil ,
nil ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
) ,
} ,
object : & UserGrant {
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeMachine ,
UserResourceOwner : "resource-owner" ,
FirstName : "" ,
LastName : "" ,
Email : "" ,
DisplayName : "" ,
AvatarURL : "" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
{
name : "prepareUserGrantQuery (no org) found" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQuery (
userGrantStmt ,
userGrantCols ,
[ ] driver . Value {
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
nil ,
nil ,
"project-id" ,
"project-name" ,
} ,
) ,
} ,
object : & UserGrant {
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "" ,
OrgPrimaryDomain : "" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
{
name : "prepareUserGrantQuery (no project) found" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQuery (
userGrantStmt ,
userGrantCols ,
[ ] driver . Value {
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
nil ,
} ,
) ,
} ,
object : & UserGrant {
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "" ,
} ,
} ,
{
name : "prepareUserGrantQuery (no loginname) found" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQuery (
userGrantStmt ,
userGrantCols ,
[ ] driver . Value {
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
nil ,
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
) ,
} ,
object : & UserGrant {
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
{
name : "prepareUserGrantQuery sql err" ,
prepare : prepareUserGrantQuery ,
want : want {
sqlExpectations : mockQueryErr (
userGrantStmt ,
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 ,
} ,
{
name : "prepareUserGrantsQuery no result" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
nil ,
nil ,
) ,
} ,
object : & UserGrants { UserGrants : [ ] * UserGrant { } } ,
} ,
{
name : "prepareUserGrantsQuery one grant" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
userGrantsCols ,
[ ] [ ] driver . Value {
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
} ,
) ,
} ,
object : & UserGrants {
SearchResponse : SearchResponse {
Count : 1 ,
} ,
UserGrants : [ ] * UserGrant {
{
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareUserGrantsQuery one grant (machine user)" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
userGrantsCols ,
[ ] [ ] driver . Value {
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeMachine ,
"resource-owner" ,
nil ,
nil ,
nil ,
nil ,
nil ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
} ,
) ,
} ,
object : & UserGrants {
SearchResponse : SearchResponse {
Count : 1 ,
} ,
UserGrants : [ ] * UserGrant {
{
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeMachine ,
UserResourceOwner : "resource-owner" ,
FirstName : "" ,
LastName : "" ,
Email : "" ,
DisplayName : "" ,
AvatarURL : "" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareUserGrantsQuery one grant (no org)" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
userGrantsCols ,
[ ] [ ] driver . Value {
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeMachine ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
nil ,
nil ,
"project-id" ,
"project-name" ,
} ,
} ,
) ,
} ,
object : & UserGrants {
SearchResponse : SearchResponse {
Count : 1 ,
} ,
UserGrants : [ ] * UserGrant {
{
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeMachine ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "" ,
OrgPrimaryDomain : "" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareUserGrantsQuery one grant (no project)" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
userGrantsCols ,
[ ] [ ] driver . Value {
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
nil ,
} ,
} ,
) ,
} ,
object : & UserGrants {
SearchResponse : SearchResponse {
Count : 1 ,
} ,
UserGrants : [ ] * UserGrant {
{
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "" ,
} ,
} ,
} ,
} ,
{
name : "prepareUserGrantsQuery one grant (no loginname)" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
userGrantsCols ,
[ ] [ ] driver . Value {
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
nil ,
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
} ,
) ,
} ,
object : & UserGrants {
SearchResponse : SearchResponse {
Count : 1 ,
} ,
UserGrants : [ ] * UserGrant {
{
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareUserGrantsQuery multiple grants" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueries (
userGrantsStmt ,
userGrantsCols ,
[ ] [ ] driver . Value {
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
{
"id" ,
testNow ,
testNow ,
20211111 ,
"grant-id" ,
2022-08-31 07:52:43 +00:00
database . StringArray { "role-key" } ,
2022-01-14 09:45:50 +00:00
domain . UserGrantStateActive ,
"user-id" ,
"username" ,
domain . UserTypeHuman ,
"resource-owner" ,
"first-name" ,
"last-name" ,
"email" ,
"display-name" ,
"avatar-key" ,
2022-03-24 13:00:24 +00:00
"login-name" ,
2022-01-14 09:45:50 +00:00
"ro" ,
"org-name" ,
"primary-domain" ,
"project-id" ,
"project-name" ,
} ,
} ,
) ,
} ,
object : & UserGrants {
SearchResponse : SearchResponse {
Count : 2 ,
} ,
UserGrants : [ ] * UserGrant {
{
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
{
2022-03-24 13:00:24 +00:00
ID : "id" ,
CreationDate : testNow ,
ChangeDate : testNow ,
Sequence : 20211111 ,
2022-08-31 07:52:43 +00:00
Roles : database . StringArray { "role-key" } ,
2022-03-24 13:00:24 +00:00
GrantID : "grant-id" ,
State : domain . UserGrantStateActive ,
UserID : "user-id" ,
Username : "username" ,
UserType : domain . UserTypeHuman ,
UserResourceOwner : "resource-owner" ,
FirstName : "first-name" ,
LastName : "last-name" ,
Email : "email" ,
DisplayName : "display-name" ,
AvatarURL : "avatar-key" ,
PreferredLoginName : "login-name" ,
ResourceOwner : "ro" ,
OrgName : "org-name" ,
OrgPrimaryDomain : "primary-domain" ,
ProjectID : "project-id" ,
ProjectName : "project-name" ,
2022-01-14 09:45:50 +00:00
} ,
} ,
} ,
} ,
{
name : "prepareUserGrantsQuery sql err" ,
prepare : prepareUserGrantsQuery ,
want : want {
sqlExpectations : mockQueryErr (
userGrantsStmt ,
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 ) {
assertPrepare ( t , tt . prepare , tt . object , tt . want . sqlExpectations , tt . want . err )
} )
}
}