zitadel/internal/query/project_grant_test.go
Livio Amstutz 56b916a2b0
feat: projections auto create their tables (#3324)
* begin init checks for projections

* first projection checks

* debug notification providers with query fixes

* more projections and first index

* more projections

* more projections

* finish projections

* fix tests (remove db name)

* create tables in setup

* fix logging / error handling

* add tenant to views

* rename tenant to instance_id

* add instance_id to all projections

* add instance_id to all queries

* correct instance_id on projections

* add instance_id to failed_events

* use separate context for instance

* implement features projection

* implement features projection

* remove unique constraint from setup when migration failed

* add error to failed setup event

* add instance_id to primary keys

* fix IAM projection

* remove old migrations folder

* fix keysFromYAML test
2022-03-23 09:02:39 +01:00

838 lines
25 KiB
Go

package query
import (
"database/sql"
"database/sql/driver"
"errors"
"fmt"
"regexp"
"testing"
"github.com/lib/pq"
"github.com/caos/zitadel/internal/domain"
errs "github.com/caos/zitadel/internal/errors"
)
func Test_ProjectGrantPrepares(t *testing.T) {
type want struct {
sqlExpectations sqlExpectation
err checkErr
}
tests := []struct {
name string
prepare interface{}
want want
object interface{}
}{
{
name: "prepareProjectGrantsQuery no result",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER () `+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
nil,
nil,
),
},
object: &ProjectGrants{ProjectGrants: []*ProjectGrant{}},
},
{
name: "prepareProjectGrantsQuery one result",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER ()`+
` FROM projections.project_grants`+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id`+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
"count",
},
[][]driver.Value{
{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
"org-name",
pq.StringArray{"role-key"},
"ro-name",
},
},
),
},
object: &ProjectGrants{
SearchResponse: SearchResponse{
Count: 1,
},
ProjectGrants: []*ProjectGrant{
{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
},
},
{
name: "prepareProjectGrantsQuery no project",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER () `+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
"count",
},
[][]driver.Value{
{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
nil,
"org-id",
"org-name",
pq.StringArray{"role-key"},
"ro-name",
},
},
),
},
object: &ProjectGrants{
SearchResponse: SearchResponse{
Count: 1,
},
ProjectGrants: []*ProjectGrant{
{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
},
},
{
name: "prepareProjectGrantsQuery no org",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER () `+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
"count",
},
[][]driver.Value{
{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
nil,
pq.StringArray{"role-key"},
"ro-name",
},
},
),
},
object: &ProjectGrants{
SearchResponse: SearchResponse{
Count: 1,
},
ProjectGrants: []*ProjectGrant{
{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
},
},
{
name: "prepareProjectGrantsQuery no resource owner",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER () `+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
"count",
},
[][]driver.Value{
{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
"org-name",
pq.StringArray{"role-key"},
nil,
},
},
),
},
object: &ProjectGrants{
SearchResponse: SearchResponse{
Count: 1,
},
ProjectGrants: []*ProjectGrant{
{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "",
},
},
},
},
{
name: "prepareProjectGrantsQuery multiple result",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER () `+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
"count",
},
[][]driver.Value{
{
"project-id",
"grant-id-1",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
"org-name",
pq.StringArray{"role-key"},
"ro-name",
},
{
"project-id",
"grant-id-2",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
"org-name",
pq.StringArray{"role-key"},
"ro-name",
},
},
),
},
object: &ProjectGrants{
SearchResponse: SearchResponse{
Count: 2,
},
ProjectGrants: []*ProjectGrant{
{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id-1",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id-2",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
},
},
{
name: "prepareProjectGrantsQuery sql err",
prepare: prepareProjectGrantsQuery,
want: want{
sqlExpectations: mockQueryErr(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name,`+
` COUNT(*) OVER () `+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
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: "prepareProjectGrantQuery no result",
prepare: prepareProjectGrantQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name`+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
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: (*ProjectGrant)(nil),
},
{
name: "prepareProjectGrantQuery found",
prepare: prepareProjectGrantQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name`+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
},
[]driver.Value{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
"org-name",
pq.StringArray{"role-key"},
"ro-name",
},
),
},
object: &ProjectGrant{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
{
name: "prepareProjectGrantQuery no org",
prepare: prepareProjectGrantQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name`+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
},
[]driver.Value{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
nil,
pq.StringArray{"role-key"},
"ro-name",
},
),
},
object: &ProjectGrant{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
{
name: "prepareProjectGrantQuery no resource owner",
prepare: prepareProjectGrantQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name`+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
},
[]driver.Value{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
"project-name",
"org-id",
"org-name",
pq.StringArray{"role-key"},
nil,
},
),
},
object: &ProjectGrant{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "project-name",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "",
},
},
{
name: "prepareProjectGrantQuery no project",
prepare: prepareProjectGrantQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name`+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
[]string{
"project_id",
"grant_id",
"creation_date",
"change_date",
"resource_owner",
"state",
"sequence",
"name",
"granted_org_id",
"name",
"granted_role_keys",
"name",
},
[]driver.Value{
"project-id",
"grant-id",
testNow,
testNow,
"ro",
domain.ProjectGrantStateActive,
20211111,
nil,
"org-id",
"org-name",
pq.StringArray{"role-key"},
"ro-name",
},
),
},
object: &ProjectGrant{
ProjectID: "project-id",
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211111,
GrantID: "grant-id",
State: domain.ProjectGrantStateActive,
ProjectName: "",
GrantedOrgID: "org-id",
OrgName: "org-name",
GrantedRoleKeys: pq.StringArray{"role-key"},
ResourceOwnerName: "ro-name",
},
},
{
name: "prepareProjectGrantQuery sql err",
prepare: prepareProjectGrantQuery,
want: want{
sqlExpectations: mockQueryErr(
regexp.QuoteMeta(` SELECT projections.project_grants.project_id,`+
` projections.project_grants.grant_id,`+
` projections.project_grants.creation_date,`+
` projections.project_grants.change_date,`+
` projections.project_grants.resource_owner,`+
` projections.project_grants.state,`+
` projections.project_grants.sequence,`+
` projections.projects.name,`+
` projections.project_grants.granted_org_id,`+
` o.name,`+
` projections.project_grants.granted_role_keys,`+
` r.name`+
` FROM projections.project_grants `+
` LEFT JOIN projections.projects ON projections.project_grants.project_id = projections.projects.id `+
` LEFT JOIN projections.orgs as r ON projections.project_grants.resource_owner = r.id`+
` LEFT JOIN projections.orgs as o ON projections.project_grants.granted_org_id = o.id`),
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)
})
}
}