zitadel/internal/query/features_test.go
2021-11-21 20:22:25 +01:00

349 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_FeaturesPrepares(t *testing.T) {
type want struct {
sqlExpectations sqlExpectation
err checkErr
}
tests := []struct {
name string
prepare interface{}
want want
object interface{}
}{
{
name: "prepareFeaturesQuery no result",
prepare: prepareFeaturesQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(`SELECT zitadel.projections.features.aggregate_id,`+
` zitadel.projections.features.change_date,`+
` zitadel.projections.features.sequence,`+
` zitadel.projections.features.is_default,`+
` zitadel.projections.features.tier_name,`+
` zitadel.projections.features.tier_description,`+
` zitadel.projections.features.state,`+
` zitadel.projections.features.state_description,`+
` zitadel.projections.features.audit_log_retention,`+
` zitadel.projections.features.login_policy_factors,`+
` zitadel.projections.features.login_policy_idp,`+
` zitadel.projections.features.login_policy_passwordless,`+
` zitadel.projections.features.login_policy_registration,`+
` zitadel.projections.features.login_policy_username_login,`+
` zitadel.projections.features.login_policy_password_reset,`+
` zitadel.projections.features.password_complexity_policy,`+
` zitadel.projections.features.label_policy_private_label,`+
` zitadel.projections.features.label_policy_watermark,`+
` zitadel.projections.features.custom_domain,`+
` zitadel.projections.features.privacy_policy,`+
` zitadel.projections.features.metadata_user,`+
` zitadel.projections.features.custom_text_message,`+
` zitadel.projections.features.custom_text_login,`+
` zitadel.projections.features.lockout_policy,`+
` zitadel.projections.features.actions`+
` FROM zitadel.projections.features`),
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: (*Features)(nil),
},
{
name: "prepareFeaturesQuery found",
prepare: prepareFeaturesQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(`SELECT zitadel.projections.features.aggregate_id,`+
` zitadel.projections.features.change_date,`+
` zitadel.projections.features.sequence,`+
` zitadel.projections.features.is_default,`+
` zitadel.projections.features.tier_name,`+
` zitadel.projections.features.tier_description,`+
` zitadel.projections.features.state,`+
` zitadel.projections.features.state_description,`+
` zitadel.projections.features.audit_log_retention,`+
` zitadel.projections.features.login_policy_factors,`+
` zitadel.projections.features.login_policy_idp,`+
` zitadel.projections.features.login_policy_passwordless,`+
` zitadel.projections.features.login_policy_registration,`+
` zitadel.projections.features.login_policy_username_login,`+
` zitadel.projections.features.login_policy_password_reset,`+
` zitadel.projections.features.password_complexity_policy,`+
` zitadel.projections.features.label_policy_private_label,`+
` zitadel.projections.features.label_policy_watermark,`+
` zitadel.projections.features.custom_domain,`+
` zitadel.projections.features.privacy_policy,`+
` zitadel.projections.features.metadata_user,`+
` zitadel.projections.features.custom_text_message,`+
` zitadel.projections.features.custom_text_login,`+
` zitadel.projections.features.lockout_policy,`+
` zitadel.projections.features.actions`+
` FROM zitadel.projections.features`),
[]string{
"aggregate_id",
"change_date",
"sequence",
"is_default",
"tier_name",
"tier_description",
"state",
"state_description",
"audit_log_retention",
"login_policy_factors",
"login_policy_idp",
"login_policy_passwordless",
"login_policy_registration",
"login_policy_username_login",
"login_policy_password_reset",
"password_complexity_policy",
"label_policy_private_label",
"label_policy_watermark",
"custom_domain",
"privacy_policy",
"metadata_user",
"custom_text_message",
"custom_text_login",
"lockout_policy",
"actions",
},
[]driver.Value{
"aggregate-id",
testNow,
uint64(20211115),
true,
"tier-name",
"tier-description",
1,
"state-description",
uint(604800000000000), // 7days in nanoseconds
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
},
),
},
object: &Features{
AggregateID: "aggregate-id",
ChangeDate: testNow,
Sequence: 20211115,
IsDefault: true,
TierName: "tier-name",
TierDescription: "tier-description",
State: domain.FeaturesStateActive,
StateDescription: "state-description",
AuditLogRetention: 7 * 24 * time.Hour,
LoginPolicyFactors: true,
LoginPolicyIDP: true,
LoginPolicyPasswordless: true,
LoginPolicyRegistration: true,
LoginPolicyUsernameLogin: true,
LoginPolicyPasswordReset: true,
PasswordComplexityPolicy: true,
LabelPolicyPrivateLabel: true,
LabelPolicyWatermark: true,
CustomDomain: true,
PrivacyPolicy: true,
MetadataUser: true,
CustomTextMessage: true,
CustomTextLogin: true,
LockoutPolicy: true,
Actions: true,
},
},
{
name: "prepareFeaturesQuery found with empty",
prepare: prepareFeaturesQuery,
want: want{
sqlExpectations: mockQuery(
regexp.QuoteMeta(`SELECT zitadel.projections.features.aggregate_id,`+
` zitadel.projections.features.change_date,`+
` zitadel.projections.features.sequence,`+
` zitadel.projections.features.is_default,`+
` zitadel.projections.features.tier_name,`+
` zitadel.projections.features.tier_description,`+
` zitadel.projections.features.state,`+
` zitadel.projections.features.state_description,`+
` zitadel.projections.features.audit_log_retention,`+
` zitadel.projections.features.login_policy_factors,`+
` zitadel.projections.features.login_policy_idp,`+
` zitadel.projections.features.login_policy_passwordless,`+
` zitadel.projections.features.login_policy_registration,`+
` zitadel.projections.features.login_policy_username_login,`+
` zitadel.projections.features.login_policy_password_reset,`+
` zitadel.projections.features.password_complexity_policy,`+
` zitadel.projections.features.label_policy_private_label,`+
` zitadel.projections.features.label_policy_watermark,`+
` zitadel.projections.features.custom_domain,`+
` zitadel.projections.features.privacy_policy,`+
` zitadel.projections.features.metadata_user,`+
` zitadel.projections.features.custom_text_message,`+
` zitadel.projections.features.custom_text_login,`+
` zitadel.projections.features.lockout_policy,`+
` zitadel.projections.features.actions`+
` FROM zitadel.projections.features`),
[]string{
"aggregate_id",
"change_date",
"sequence",
"is_default",
"tier_name",
"tier_description",
"state",
"state_description",
"audit_log_retention",
"login_policy_factors",
"login_policy_idp",
"login_policy_passwordless",
"login_policy_registration",
"login_policy_username_login",
"login_policy_password_reset",
"password_complexity_policy",
"label_policy_private_label",
"label_policy_watermark",
"custom_domain",
"privacy_policy",
"metadata_user",
"custom_text_message",
"custom_text_login",
"lockout_policy",
"actions",
},
[]driver.Value{
"aggregate-id",
testNow,
uint64(20211115),
true,
nil,
nil,
1,
nil,
uint(604800000000000), // 7days in nanoseconds
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
true,
},
),
},
object: &Features{
AggregateID: "aggregate-id",
ChangeDate: testNow,
Sequence: 20211115,
IsDefault: true,
TierName: "",
TierDescription: "",
State: domain.FeaturesStateActive,
StateDescription: "",
AuditLogRetention: 7 * 24 * time.Hour,
LoginPolicyFactors: true,
LoginPolicyIDP: true,
LoginPolicyPasswordless: true,
LoginPolicyRegistration: true,
LoginPolicyUsernameLogin: true,
LoginPolicyPasswordReset: true,
PasswordComplexityPolicy: true,
LabelPolicyPrivateLabel: true,
LabelPolicyWatermark: true,
CustomDomain: true,
PrivacyPolicy: true,
MetadataUser: true,
CustomTextMessage: true,
CustomTextLogin: true,
LockoutPolicy: true,
Actions: true,
},
},
{
name: "prepareFeaturesQuery sql err",
prepare: prepareFeaturesQuery,
want: want{
sqlExpectations: mockQueryErr(
regexp.QuoteMeta(`SELECT zitadel.projections.features.aggregate_id,`+
` zitadel.projections.features.change_date,`+
` zitadel.projections.features.sequence,`+
` zitadel.projections.features.is_default,`+
` zitadel.projections.features.tier_name,`+
` zitadel.projections.features.tier_description,`+
` zitadel.projections.features.state,`+
` zitadel.projections.features.state_description,`+
` zitadel.projections.features.audit_log_retention,`+
` zitadel.projections.features.login_policy_factors,`+
` zitadel.projections.features.login_policy_idp,`+
` zitadel.projections.features.login_policy_passwordless,`+
` zitadel.projections.features.login_policy_registration,`+
` zitadel.projections.features.login_policy_username_login,`+
` zitadel.projections.features.login_policy_password_reset,`+
` zitadel.projections.features.password_complexity_policy,`+
` zitadel.projections.features.label_policy_private_label,`+
` zitadel.projections.features.label_policy_watermark,`+
` zitadel.projections.features.custom_domain,`+
` zitadel.projections.features.privacy_policy,`+
` zitadel.projections.features.metadata_user,`+
` zitadel.projections.features.custom_text_message,`+
` zitadel.projections.features.custom_text_login,`+
` zitadel.projections.features.lockout_policy,`+
` zitadel.projections.features.actions`+
` FROM zitadel.projections.features`),
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)
})
}
}