zitadel/internal/query/secret_generator_test.go
Fabi e3528ff0b2
feat: Config to eventstore (#3158)
* feat: add default language to eventstore

* feat: add secret generator configs events

* feat: tests

* feat: secret generators in eventstore

* feat: secret generators in eventstore

* feat: smtp config in eventstore

* feat: smtp config in eventstore

* feat: smtp config in eventstore

* feat: smtp config in eventstore

* feat: smtp config in eventstore

* fix: migrations

* fix migration version

* fix test

* feat: change secret generator type to enum

* feat: change smtp attribute names

* feat: change smtp attribute names

* feat: remove engryption algorithms from command side

* feat: remove engryption algorithms from command side

* feat: smtp config

* feat: smtp config

* format smtp from header

Co-authored-by: Livio Amstutz <livio.a@gmail.com>
2022-02-16 15:49:17 +00:00

387 lines
12 KiB
Go

package query
import (
"database/sql"
"database/sql/driver"
"errors"
"fmt"
"regexp"
"testing"
"time"
"github.com/caos/zitadel/internal/domain"
errs "github.com/caos/zitadel/internal/errors"
)
func Test_SecretGeneratorsPrepares(t *testing.T) {
type want struct {
sqlExpectations sqlExpectation
err checkErr
}
tests := []struct {
name string
prepare interface{}
want want
object interface{}
}{
{
name: "prepareSecretGeneratorsQuery no result",
prepare: prepareSecretGeneratorsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols,`+
` COUNT(*) OVER ()`+
` FROM zitadel.projections.secret_generators`),
nil,
nil,
),
},
object: &SecretGenerators{SecretGenerators: []*SecretGenerator{}},
},
{
name: "prepareSecretGeneratorsQuery one result",
prepare: prepareSecretGeneratorsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols,`+
` COUNT(*) OVER ()`+
` FROM zitadel.projections.secret_generators`),
[]string{
"aggregate_id",
"generator_type",
"creation_date",
"change_date",
"resource_owner",
"sequence",
"length",
"expiry",
"include_lower_letters",
"include_upper_letters",
"include_digits",
"include_symbols",
"count",
},
[][]driver.Value{
{
"agg-id",
domain.SecretGeneratorTypeInitCode,
testNow,
testNow,
"ro",
uint64(20211108),
4,
time.Minute * 1,
true,
true,
true,
true,
},
},
),
},
object: &SecretGenerators{
SearchResponse: SearchResponse{
Count: 1,
},
SecretGenerators: []*SecretGenerator{
{
AggregateID: "agg-id",
GeneratorType: 1,
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211108,
Length: 4,
Expiry: time.Minute * 1,
IncludeLowerLetters: true,
IncludeUpperLetters: true,
IncludeDigits: true,
IncludeSymbols: true,
},
},
},
},
{
name: "prepareSecretGeneratorsQuery multiple result",
prepare: prepareSecretGeneratorsQuery,
want: want{
sqlExpectations: mockQueries(
regexp.QuoteMeta(`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols,`+
` COUNT(*) OVER ()`+
` FROM zitadel.projections.secret_generators`),
[]string{
"aggregate_id",
"generator_type",
"creation_date",
"change_date",
"resource_owner",
"sequence",
"length",
"expiry",
"include_lower_letters",
"include_upper_letters",
"include_digits",
"include_symbols",
"count",
},
[][]driver.Value{
{
"agg-id",
domain.SecretGeneratorTypeInitCode,
testNow,
testNow,
"ro",
uint64(20211108),
4,
time.Minute * 1,
true,
true,
true,
true,
},
{
"agg-id",
domain.SecretGeneratorTypeVerifyEmailCode,
testNow,
testNow,
"ro",
uint64(20211108),
4,
time.Minute * 1,
true,
true,
true,
true,
},
},
),
},
object: &SecretGenerators{
SearchResponse: SearchResponse{
Count: 2,
},
SecretGenerators: []*SecretGenerator{
{
AggregateID: "agg-id",
GeneratorType: 1,
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211108,
Length: 4,
Expiry: time.Minute * 1,
IncludeLowerLetters: true,
IncludeUpperLetters: true,
IncludeDigits: true,
IncludeSymbols: true,
},
{
AggregateID: "agg-id",
GeneratorType: 2,
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211108,
Length: 4,
Expiry: time.Minute * 1,
IncludeLowerLetters: true,
IncludeUpperLetters: true,
IncludeDigits: true,
IncludeSymbols: true,
},
},
},
},
{
name: "prepareSecretGeneratorsQuery sql err",
prepare: prepareSecretGeneratorsQuery,
want: want{
sqlExpectations: mockQueryErr(
regexp.QuoteMeta(`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols,`+
` COUNT(*) OVER ()`+
` FROM zitadel.projections.secret_generators`),
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: "prepareSecretGeneratorQuery no result",
prepare: prepareSecretGeneratorQuery,
want: want{
sqlExpectations: mockQueries(
`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols`+
` FROM zitadel.projections.secret_generators`,
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: (*SecretGenerator)(nil),
},
{
name: "prepareSecretGeneratorQuery found",
prepare: prepareSecretGeneratorQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols`+
` FROM zitadel.projections.secret_generators`),
[]string{
"aggregate_id",
"generator_type",
"creation_date",
"change_date",
"resource_owner",
"sequence",
"length",
"expiry",
"include_lower_letters",
"include_upper_letters",
"include_digits",
"include_symbols",
},
[]driver.Value{
"agg-id",
domain.SecretGeneratorTypeInitCode,
testNow,
testNow,
"ro",
uint64(20211108),
4,
time.Minute * 1,
true,
true,
true,
true,
},
),
},
object: &SecretGenerator{
AggregateID: "agg-id",
GeneratorType: domain.SecretGeneratorTypeInitCode,
CreationDate: testNow,
ChangeDate: testNow,
ResourceOwner: "ro",
Sequence: 20211108,
Length: 4,
Expiry: time.Minute * 1,
IncludeLowerLetters: true,
IncludeUpperLetters: true,
IncludeDigits: true,
IncludeSymbols: true,
},
},
{
name: "prepareSecretGeneratorQuery sql err",
prepare: prepareSecretGeneratorQuery,
want: want{
sqlExpectations: mockQueryErr(
regexp.QuoteMeta(`SELECT zitadel.projections.secret_generators.aggregate_id,`+
` zitadel.projections.secret_generators.generator_type,`+
` zitadel.projections.secret_generators.creation_date,`+
` zitadel.projections.secret_generators.change_date,`+
` zitadel.projections.secret_generators.resource_owner,`+
` zitadel.projections.secret_generators.sequence,`+
` zitadel.projections.secret_generators.length,`+
` zitadel.projections.secret_generators.expiry,`+
` zitadel.projections.secret_generators.include_lower_letters,`+
` zitadel.projections.secret_generators.include_upper_letters,`+
` zitadel.projections.secret_generators.include_digits,`+
` zitadel.projections.secret_generators.include_symbols`+
` FROM zitadel.projections.secret_generators`),
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)
})
}
}