TheArchive/zitadel
1
0
Fork 0
mirror of https://github.com/zitadel/zitadel.git synced 2025-08-11 10:47:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

org domain tests

Browse Source
This commit is contained in:
adlerhurst
2025-07-23 12:11:51 +02:00
parent e516c1cdf6
commit d9bff64699
3 changed files with 980 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/v3/domain/organization_domain.go
View File

@@ -15,8 +15,8 @@ type OrganizationDomain struct {
IsPrimary bool `json:"isPrimary,omitempty" db:"is_primary"`
ValidationType DomainValidationType `json:"validationType,omitempty" db:"validation_type"`
CreatedAt string `json:"createdAt,omitempty" db:"created_at"`
UpdatedAt string `json:"updatedAt,omitempty" db:"updated_at"`
CreatedAt time.Time `json:"createdAt,omitzero" db:"created_at"`
UpdatedAt time.Time `json:"updatedAt,omitzero" db:"updated_at"`
}
type AddOrganizationDomain struct {

21
backend/v3/storage/database/repository/org_domain.go
View File

@@ -18,8 +18,8 @@ type orgDomain struct {
// repository
// -------------------------------------------------------------
const queryOrganizationDomainStmt = `SELECT instance_id, org_id, domain, is_verified, is_primary, verification_type, created_at, updated_at ` +
`FROM zitadel.organization_domains`
const queryOrganizationDomainStmt = `SELECT instance_id, org_id, domain, is_verified, is_primary, validation_type, created_at, updated_at ` +
`FROM zitadel.org_domains`
// Get implements [domain.OrganizationDomainRepository].
// Subtle: this method shadows the method ([domain.OrganizationRepository]).Get of orgDomain.org.
@@ -55,11 +55,9 @@ func (o *orgDomain) List(ctx context.Context, opts ...database.QueryOption) ([]*
func (o *orgDomain) Add(ctx context.Context, domain *domain.AddOrganizationDomain) error {
var builder database.StatementBuilder
builder.WriteString(`INSERT INTO zitadel.organization_domains (instance_id, org_id, domain, is_verified, is_primary, verification_type) ` +
`VALUES ($1, $2, $3, $4, $5, $6)` +
` RETURNING created_at, updated_at`)
builder.AppendArgs(domain.InstanceID, domain.OrgID, domain.Domain, domain.IsVerified, domain.IsPrimary, domain.ValidationType)
builder.WriteString(`INSERT INTO zitadel.org_domains (instance_id, org_id, domain, is_verified, is_primary, validation_type) VALUES (`)
builder.WriteArgs(domain.InstanceID, domain.OrgID, domain.Domain, domain.IsVerified, domain.IsPrimary, domain.ValidationType)
builder.WriteString(`) RETURNING created_at, updated_at`)
return o.client.QueryRow(ctx, builder.String(), builder.Args()...).Scan(&domain.CreatedAt, &domain.UpdatedAt)
}
@@ -67,11 +65,14 @@ func (o *orgDomain) Add(ctx context.Context, domain *domain.AddOrganizationDomai
// Update implements [domain.OrganizationDomainRepository].
// Subtle: this method shadows the method ([domain.OrganizationRepository]).Update of orgDomain.org.
func (o *orgDomain) Update(ctx context.Context, condition database.Condition, changes ...database.Change) (int64, error) {
if len(changes) == 0 {
return 0, database.NoChangesError
}
var builder database.StatementBuilder
builder.WriteString(`UPDATE zitadel.organization_domains SET `)
builder.WriteString(`UPDATE zitadel.org_domains SET `)
database.Changes(changes).Write(&builder)
writeCondition(&builder, condition)
return o.client.Exec(ctx, builder.String(), builder.Args()...)
@@ -81,7 +82,7 @@ func (o *orgDomain) Update(ctx context.Context, condition database.Condition, ch
func (o *orgDomain) Remove(ctx context.Context, condition database.Condition) (int64, error) {
var builder database.StatementBuilder
builder.WriteString(`DELETE FROM zitadel.organization_domains `)
builder.WriteString(`DELETE FROM zitadel.org_domains `)
writeCondition(&builder, condition)
return o.client.Exec(ctx, builder.String(), builder.Args()...)

967
backend/v3/storage/database/repository/org_domain_test.go Normal file
View File

@@ -0,0 +1,967 @@
package repository_test
import (
"context"
"testing"
"github.com/brianvoe/gofakeit/v6"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/zitadel/zitadel/backend/v3/domain"
"github.com/zitadel/zitadel/backend/v3/storage/database"
"github.com/zitadel/zitadel/backend/v3/storage/database/repository"
)
func TestAddOrganizationDomain(t *testing.T) {
// create instance
instanceID := gofakeit.UUID()
instance := domain.Instance{
ID: instanceID,
Name: gofakeit.Name(),
DefaultOrgID: "defaultOrgId",
IAMProjectID: "iamProject",
ConsoleClientID: "consoleClient",
ConsoleAppID: "consoleApp",
DefaultLanguage: "defaultLanguage",
}
instanceRepo := repository.InstanceRepository(pool)
err := instanceRepo.Create(t.Context(), &instance)
require.NoError(t, err)
// create organization
orgID := gofakeit.UUID()
organization := domain.Organization{
ID: orgID,
Name: gofakeit.Name(),
InstanceID: instanceID,
State: domain.OrgStateActive,
}
tests := []struct {
name string
testFunc func(ctx context.Context, t *testing.T, domainRepo domain.OrganizationDomainRepository) *domain.AddOrganizationDomain
organizationDomain domain.AddOrganizationDomain
err error
}{
{
name: "happy path",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
},
{
name: "add verified domain",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: true,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeHTTP,
},
},
{
name: "add primary domain",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
},
},
{
name: "add domain without domain name",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: "",
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
err: new(database.CheckError),
},
{
name: "add domain with same domain twice",
testFunc: func(ctx context.Context, t *testing.T, domainRepo domain.OrganizationDomainRepository) *domain.AddOrganizationDomain {
domainName := gofakeit.DomainName()
organizationDomain := &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName,
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
}
err := domainRepo.Add(ctx, organizationDomain)
require.NoError(t, err)
// return same domain again
return &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName,
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeHTTP,
}
},
err: new(database.UniqueError),
},
{
name: "add domain with non-existent instance",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: "non-existent-instance",
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
err: new(database.ForeignKeyError),
},
{
name: "add domain with non-existent organization",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: "non-existent-org",
Domain: gofakeit.DomainName(),
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
err: new(database.ForeignKeyError),
},
{
name: "add domain without instance id",
organizationDomain: domain.AddOrganizationDomain{
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
err: new(database.ForeignKeyError),
},
{
name: "add domain without org id",
organizationDomain: domain.AddOrganizationDomain{
InstanceID: instanceID,
Domain: gofakeit.DomainName(),
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
err: new(database.ForeignKeyError),
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ctx := t.Context()
tx, err := pool.Begin(t.Context(), nil)
require.NoError(t, err)
defer func() {
require.NoError(t, tx.Rollback(t.Context()))
}()
orgRepo := repository.OrganizationRepository(tx)
err = orgRepo.Create(t.Context(), &organization)
require.NoError(t, err)
domainRepo := orgRepo.Domains(false)
var organizationDomain *domain.AddOrganizationDomain
if test.testFunc != nil {
organizationDomain = test.testFunc(ctx, t, domainRepo)
} else {
organizationDomain = &test.organizationDomain
}
err = domainRepo.Add(ctx, organizationDomain)
if test.err != nil {
assert.ErrorIs(t, err, test.err)
return
}
require.NoError(t, err)
assert.NotZero(t, organizationDomain.CreatedAt)
assert.NotZero(t, organizationDomain.UpdatedAt)
})
}
}
func TestGetOrganizationDomain(t *testing.T) {
// create instance
instanceID := gofakeit.UUID()
instance := domain.Instance{
ID: instanceID,
Name: gofakeit.Name(),
DefaultOrgID: "defaultOrgId",
IAMProjectID: "iamProject",
ConsoleClientID: "consoleClient",
ConsoleAppID: "consoleApp",
DefaultLanguage: "defaultLanguage",
}
// create organization
orgID := gofakeit.UUID()
organization := domain.Organization{
ID: orgID,
Name: gofakeit.Name(),
InstanceID: instanceID,
State: domain.OrgStateActive,
}
tx, err := pool.Begin(t.Context(), nil)
require.NoError(t, err)
defer func() {
require.NoError(t, tx.Rollback(t.Context()))
}()
instanceRepo := repository.InstanceRepository(tx)
err = instanceRepo.Create(t.Context(), &instance)
require.NoError(t, err)
orgRepo := repository.OrganizationRepository(tx)
err = orgRepo.Create(t.Context(), &organization)
require.NoError(t, err)
// add domains
domainRepo := orgRepo.Domains(false)
domainName1 := gofakeit.DomainName()
domainName2 := gofakeit.DomainName()
domain1 := &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName1,
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
}
domain2 := &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName2,
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeHTTP,
}
err = domainRepo.Add(t.Context(), domain1)
require.NoError(t, err)
err = domainRepo.Add(t.Context(), domain2)
require.NoError(t, err)
tests := []struct {
name string
opts []database.QueryOption
expected *domain.OrganizationDomain
err error
}{
{
name: "get primary domain",
opts: []database.QueryOption{
database.WithCondition(domainRepo.IsPrimaryCondition(true)),
},
expected: &domain.OrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName1,
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
},
},
{
name: "get by domain name",
opts: []database.QueryOption{
database.WithCondition(domainRepo.DomainCondition(database.TextOperationEqual, domainName2)),
},
expected: &domain.OrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName2,
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeHTTP,
},
},
{
name: "get by org ID",
opts: []database.QueryOption{
database.WithCondition(domainRepo.OrgIDCondition(orgID)),
database.WithCondition(domainRepo.IsPrimaryCondition(true)),
},
expected: &domain.OrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName1,
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
},
},
{
name: "get verified domain",
opts: []database.QueryOption{
database.WithCondition(domainRepo.IsVerifiedCondition(true)),
},
expected: &domain.OrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName1,
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
},
},
{
name: "get non-existent domain",
opts: []database.QueryOption{
database.WithCondition(domainRepo.DomainCondition(database.TextOperationEqual, "non-existent.com")),
},
err: new(database.NoRowFoundError),
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ctx := t.Context()
result, err := domainRepo.Get(ctx, test.opts...)
if test.err != nil {
assert.ErrorIs(t, err, test.err)
return
}
require.NoError(t, err)
assert.Equal(t, test.expected.InstanceID, result.InstanceID)
assert.Equal(t, test.expected.OrgID, result.OrgID)
assert.Equal(t, test.expected.Domain, result.Domain)
assert.Equal(t, test.expected.IsVerified, result.IsVerified)
assert.Equal(t, test.expected.IsPrimary, result.IsPrimary)
assert.Equal(t, test.expected.ValidationType, result.ValidationType)
assert.NotEmpty(t, result.CreatedAt)
assert.NotEmpty(t, result.UpdatedAt)
})
}
}
func TestListOrganizationDomains(t *testing.T) {
// create instance
instanceID := gofakeit.UUID()
instance := domain.Instance{
ID: instanceID,
Name: gofakeit.Name(),
DefaultOrgID: "defaultOrgId",
IAMProjectID: "iamProject",
ConsoleClientID: "consoleClient",
ConsoleAppID: "consoleApp",
DefaultLanguage: "defaultLanguage",
}
// create organization
orgID := gofakeit.UUID()
organization := domain.Organization{
ID: orgID,
Name: gofakeit.Name(),
InstanceID: instanceID,
State: domain.OrgStateActive,
}
tx, err := pool.Begin(t.Context(), nil)
require.NoError(t, err)
defer func() {
require.NoError(t, tx.Rollback(t.Context()))
}()
instanceRepo := repository.InstanceRepository(tx)
err = instanceRepo.Create(t.Context(), &instance)
require.NoError(t, err)
orgRepo := repository.OrganizationRepository(tx)
err = orgRepo.Create(t.Context(), &organization)
require.NoError(t, err)
// add multiple domains
domainRepo := orgRepo.Domains(false)
domains := []domain.AddOrganizationDomain{
{
InstanceID: instanceID,
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
},
{
InstanceID: instanceID,
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeHTTP,
},
{
InstanceID: instanceID,
OrgID: orgID,
Domain: gofakeit.DomainName(),
IsVerified: true,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
},
}
for i := range domains {
err = domainRepo.Add(t.Context(), &domains[i])
require.NoError(t, err)
}
tests := []struct {
name string
opts []database.QueryOption
expectedCount int
}{
{
name: "list all domains",
opts: []database.QueryOption{},
expectedCount: 3,
},
{
name: "list verified domains",
opts: []database.QueryOption{
database.WithCondition(domainRepo.IsVerifiedCondition(true)),
},
expectedCount: 2,
},
{
name: "list primary domains",
opts: []database.QueryOption{
database.WithCondition(domainRepo.IsPrimaryCondition(true)),
},
expectedCount: 1,
},
{
name: "list by organization",
opts: []database.QueryOption{
database.WithCondition(domainRepo.OrgIDCondition(orgID)),
},
expectedCount: 3,
},
{
name: "list by instance",
opts: []database.QueryOption{
database.WithCondition(domainRepo.InstanceIDCondition(instanceID)),
},
expectedCount: 3,
},
{
name: "list non-existent organization",
opts: []database.QueryOption{
database.WithCondition(domainRepo.OrgIDCondition("non-existent")),
},
expectedCount: 0,
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ctx := t.Context()
results, err := domainRepo.List(ctx, test.opts...)
require.NoError(t, err)
assert.Len(t, results, test.expectedCount)
for _, result := range results {
assert.Equal(t, instanceID, result.InstanceID)
assert.Equal(t, orgID, result.OrgID)
assert.NotEmpty(t, result.Domain)
assert.NotEmpty(t, result.CreatedAt)
assert.NotEmpty(t, result.UpdatedAt)
}
})
}
}
func TestUpdateOrganizationDomain(t *testing.T) {
// create instance
instanceID := gofakeit.UUID()
instance := domain.Instance{
ID: instanceID,
Name: gofakeit.Name(),
DefaultOrgID: "defaultOrgId",
IAMProjectID: "iamProject",
ConsoleClientID: "consoleClient",
ConsoleAppID: "consoleApp",
DefaultLanguage: "defaultLanguage",
}
// create organization
orgID := gofakeit.UUID()
organization := domain.Organization{
ID: orgID,
Name: gofakeit.Name(),
InstanceID: instanceID,
State: domain.OrgStateActive,
}
tx, err := pool.Begin(t.Context(), nil)
require.NoError(t, err)
defer func() {
require.NoError(t, tx.Rollback(t.Context()))
}()
instanceRepo := repository.InstanceRepository(tx)
err = instanceRepo.Create(t.Context(), &instance)
require.NoError(t, err)
orgRepo := repository.OrganizationRepository(tx)
err = orgRepo.Create(t.Context(), &organization)
require.NoError(t, err)
// add domain
domainRepo := orgRepo.Domains(false)
domainName := gofakeit.DomainName()
organizationDomain := &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName,
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeDNS,
}
err = domainRepo.Add(t.Context(), organizationDomain)
require.NoError(t, err)
tests := []struct {
name string
condition database.Condition
changes []database.Change
expected int64
err error
}{
{
name: "set verified",
condition: domainRepo.DomainCondition(database.TextOperationEqual, domainName),
changes: []database.Change{domainRepo.SetVerified()},
expected: 1,
},
{
name: "set primary",
condition: domainRepo.DomainCondition(database.TextOperationEqual, domainName),
changes: []database.Change{domainRepo.SetPrimary()},
expected: 1,
},
{
name: "set validation type",
condition: domainRepo.DomainCondition(database.TextOperationEqual, domainName),
changes: []database.Change{domainRepo.SetValidationType(domain.DomainValidationTypeHTTP)},
expected: 1,
},
{
name: "multiple changes",
condition: domainRepo.DomainCondition(database.TextOperationEqual, domainName),
changes: []database.Change{
domainRepo.SetVerified(),
domainRepo.SetPrimary(),
domainRepo.SetValidationType(domain.DomainValidationTypeDNS),
},
expected: 1,
},
{
name: "update by org ID and domain",
condition: database.And(domainRepo.OrgIDCondition(orgID), domainRepo.DomainCondition(database.TextOperationEqual, domainName)),
changes: []database.Change{domainRepo.SetVerified()},
expected: 1,
},
{
name: "update non-existent domain",
condition: domainRepo.DomainCondition(database.TextOperationEqual, "non-existent.com"),
changes: []database.Change{domainRepo.SetVerified()},
expected: 0,
},
{
name: "no changes",
condition: domainRepo.DomainCondition(database.TextOperationEqual, domainName),
changes: []database.Change{},
expected: 0,
err: database.NoChangesError,
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ctx := t.Context()
rowsAffected, err := domainRepo.Update(ctx, test.condition, test.changes...)
if test.err != nil {
assert.ErrorIs(t, err, test.err)
return
}
require.NoError(t, err)
assert.Equal(t, test.expected, rowsAffected)
// verify changes were applied if rows were affected
if rowsAffected > 0 && len(test.changes) > 0 {
result, err := domainRepo.Get(ctx, database.WithCondition(test.condition))
require.NoError(t, err)
// We know changes were applied since rowsAffected > 0
// The specific verification of what changed is less important
// than knowing the operation succeeded
assert.NotNil(t, result)
}
})
}
}
func TestRemoveOrganizationDomain(t *testing.T) {
// create instance
instanceID := gofakeit.UUID()
instance := domain.Instance{
ID: instanceID,
Name: gofakeit.Name(),
DefaultOrgID: "defaultOrgId",
IAMProjectID: "iamProject",
ConsoleClientID: "consoleClient",
ConsoleAppID: "consoleApp",
DefaultLanguage: "defaultLanguage",
}
// create organization
orgID := gofakeit.UUID()
organization := domain.Organization{
ID: orgID,
Name: gofakeit.Name(),
InstanceID: instanceID,
State: domain.OrgStateActive,
}
tx, err := pool.Begin(t.Context(), nil)
require.NoError(t, err)
defer func() {
require.NoError(t, tx.Rollback(t.Context()))
}()
instanceRepo := repository.InstanceRepository(tx)
err = instanceRepo.Create(t.Context(), &instance)
require.NoError(t, err)
orgRepo := repository.OrganizationRepository(tx)
err = orgRepo.Create(t.Context(), &organization)
require.NoError(t, err)
// add domains
domainRepo := orgRepo.Domains(false)
domainName1 := gofakeit.DomainName()
domainName2 := gofakeit.DomainName()
domain1 := &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName1,
IsVerified: true,
IsPrimary: true,
ValidationType: domain.DomainValidationTypeDNS,
}
domain2 := &domain.AddOrganizationDomain{
InstanceID: instanceID,
OrgID: orgID,
Domain: domainName2,
IsVerified: false,
IsPrimary: false,
ValidationType: domain.DomainValidationTypeHTTP,
}
err = domainRepo.Add(t.Context(), domain1)
require.NoError(t, err)
err = domainRepo.Add(t.Context(), domain2)
require.NoError(t, err)
tests := []struct {
name string
condition database.Condition
expected int64
}{
{
name: "remove by domain name",
condition: domainRepo.DomainCondition(database.TextOperationEqual, domainName1),
expected: 1,
},
{
name: "remove by primary condition",
condition: domainRepo.IsPrimaryCondition(false),
expected: 1, // domain2 should still exist and be non-primary
},
{
name: "remove by org ID and domain",
condition: database.And(domainRepo.OrgIDCondition(orgID), domainRepo.DomainCondition(database.TextOperationEqual, domainName2)),
expected: 1,
},
{
name: "remove non-existent domain",
condition: domainRepo.DomainCondition(database.TextOperationEqual, "non-existent.com"),
expected: 0,
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
ctx := t.Context()
snapshot, err := tx.Begin(ctx)
require.NoError(t, err)
defer func() {
require.NoError(t, snapshot.Rollback(ctx))
}()
orgRepo := repository.OrganizationRepository(snapshot)
domainRepo := orgRepo.Domains(false)
// count before removal
beforeCount, err := domainRepo.List(ctx)
require.NoError(t, err)
rowsAffected, err := domainRepo.Remove(ctx, test.condition)
require.NoError(t, err)
assert.Equal(t, test.expected, rowsAffected)
// verify removal
afterCount, err := domainRepo.List(ctx)
require.NoError(t, err)
assert.Equal(t, len(beforeCount)-int(test.expected), len(afterCount))
})
}
}
func TestOrganizationDomainConditions(t *testing.T) {
orgRepo := repository.OrganizationRepository(pool)
domainRepo := orgRepo.Domains(false)
tests := []struct {
name string
condition database.Condition
expected string
}{
{
name: "domain condition equal",
condition: domainRepo.DomainCondition(database.TextOperationEqual, "example.com"),
expected: "org_domains.domain = $1",
},
{
name: "domain condition starts with",
condition: domainRepo.DomainCondition(database.TextOperationStartsWith, "example"),
expected: "org_domains.domain LIKE $1 || '%'",
},
{
name: "instance id condition",
condition: domainRepo.InstanceIDCondition("instance-123"),
expected: "org_domains.instance_id = $1",
},
{
name: "org id condition",
condition: domainRepo.OrgIDCondition("org-123"),
expected: "org_domains.org_id = $1",
},
{
name: "is primary true",
condition: domainRepo.IsPrimaryCondition(true),
expected: "org_domains.is_primary = $1",
},
{
name: "is primary false",
condition: domainRepo.IsPrimaryCondition(false),
expected: "org_domains.is_primary = $1",
},
{
name: "is verified true",
condition: domainRepo.IsVerifiedCondition(true),
expected: "org_domains.is_verified = $1",
},
{
name: "is verified false",
condition: domainRepo.IsVerifiedCondition(false),
expected: "org_domains.is_verified = $1",
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
var builder database.StatementBuilder
test.condition.Write(&builder)
assert.Equal(t, test.expected, builder.String())
})
}
}
func TestOrganizationDomainChanges(t *testing.T) {
orgRepo := repository.OrganizationRepository(pool)
domainRepo := orgRepo.Domains(false)
tests := []struct {
name string
change database.Change
expected string
}{
{
name: "set verified",
change: domainRepo.SetVerified(),
expected: "is_verified = $1",
},
{
name: "set primary",
change: domainRepo.SetPrimary(),
expected: "is_primary = $1",
},
{
name: "set validation type DNS",
change: domainRepo.SetValidationType(domain.DomainValidationTypeDNS),
expected: "validation_type = $1",
},
{
name: "set validation type HTTP",
change: domainRepo.SetValidationType(domain.DomainValidationTypeHTTP),
expected: "validation_type = $1",
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
var builder database.StatementBuilder
test.change.Write(&builder)
assert.Equal(t, test.expected, builder.String())
})
}
}
func TestOrganizationDomainColumns(t *testing.T) {
orgRepo := repository.OrganizationRepository(pool)
domainRepo := orgRepo.Domains(false)
tests := []struct {
name string
column database.Column
qualified bool
expected string
}{
{
name: "instance id column qualified",
column: domainRepo.InstanceIDColumn(true),
qualified: true,
expected: "org_domains.instance_id",
},
{
name: "instance id column unqualified",
column: domainRepo.InstanceIDColumn(false),
qualified: false,
expected: "instance_id",
},
{
name: "org id column qualified",
column: domainRepo.OrgIDColumn(true),
qualified: true,
expected: "org_domains.org_id",
},
{
name: "org id column unqualified",
column: domainRepo.OrgIDColumn(false),
qualified: false,
expected: "org_id",
},
{
name: "domain column qualified",
column: domainRepo.DomainColumn(true),
qualified: true,
expected: "org_domains.domain",
},
{
name: "domain column unqualified",
column: domainRepo.DomainColumn(false),
qualified: false,
expected: "domain",
},
{
name: "is verified column qualified",
column: domainRepo.IsVerifiedColumn(true),
qualified: true,
expected: "org_domains.is_verified",
},
{
name: "is verified column unqualified",
column: domainRepo.IsVerifiedColumn(false),
qualified: false,
expected: "is_verified",
},
{
name: "is primary column qualified",
column: domainRepo.IsPrimaryColumn(true),
qualified: true,
expected: "org_domains.is_primary",
},
{
name: "is primary column unqualified",
column: domainRepo.IsPrimaryColumn(false),
qualified: false,
expected: "is_primary",
},
{
name: "validation type column qualified",
column: domainRepo.ValidationTypeColumn(true),
qualified: true,
expected: "org_domains.validation_type",
},
{
name: "validation type column unqualified",
column: domainRepo.ValidationTypeColumn(false),
qualified: false,
expected: "validation_type",
},
{
name: "created at column qualified",
column: domainRepo.CreatedAtColumn(true),
qualified: true,
expected: "org_domains.created_at",
},
{
name: "created at column unqualified",
column: domainRepo.CreatedAtColumn(false),
qualified: false,
expected: "created_at",
},
{
name: "updated at column qualified",
column: domainRepo.UpdatedAtColumn(true),
qualified: true,
expected: "org_domains.updated_at",
},
{
name: "updated at column unqualified",
column: domainRepo.UpdatedAtColumn(false),
qualified: false,
expected: "updated_at",
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
var builder database.StatementBuilder
test.column.Write(&builder)
assert.Equal(t, test.expected, builder.String())
})
}
}