package command import ( "bytes" "context" "testing" "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" "github.com/zitadel/zitadel/internal/api/authz" "github.com/zitadel/zitadel/internal/domain" caos_errs "github.com/zitadel/zitadel/internal/errors" "github.com/zitadel/zitadel/internal/eventstore" "github.com/zitadel/zitadel/internal/eventstore/v1/models" "github.com/zitadel/zitadel/internal/repository/instance" "github.com/zitadel/zitadel/internal/repository/policy" "github.com/zitadel/zitadel/internal/static" "github.com/zitadel/zitadel/internal/static/mock" ) func TestCommandSide_AddDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore } type args struct { ctx context.Context primaryColor string backgroundColor string warnColor string fontColor string primaryColorDark string backgroundColorDark string warnColorDark string fontColorDark string hideLoginNameSuffix bool errorMsgPopup bool disableWatermark bool } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "labelpolicy already existing, already exists error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), }, args: args{ ctx: context.Background(), primaryColor: "#ffffff", backgroundColor: "#ffffff", warnColor: "#ffffff", fontColor: "#ffffff", primaryColorDark: "#ffffff", backgroundColorDark: "#ffffff", warnColorDark: "#ffffff", fontColorDark: "#ffffff", hideLoginNameSuffix: true, errorMsgPopup: true, disableWatermark: true, }, res: res{ err: caos_errs.IsErrorAlreadyExists, }, }, { name: "add policy,ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), expectPush( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), }, args: args{ ctx: authz.WithInstanceID(context.Background(), "INSTANCE"), primaryColor: "#ffffff", backgroundColor: "#ffffff", warnColor: "#ffffff", fontColor: "#ffffff", primaryColorDark: "#ffffff", backgroundColorDark: "#ffffff", warnColorDark: "#ffffff", fontColorDark: "#ffffff", hideLoginNameSuffix: true, errorMsgPopup: true, disableWatermark: true, }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, } got, err := r.AddDefaultLabelPolicy( tt.args.ctx, tt.args.primaryColor, tt.args.backgroundColor, tt.args.warnColor, tt.args.fontColor, tt.args.primaryColorDark, tt.args.backgroundColorDark, tt.args.warnColorDark, tt.args.fontColorDark, tt.args.hideLoginNameSuffix, tt.args.errorMsgPopup, tt.args.disableWatermark, ) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_ChangeDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore } type args struct { ctx context.Context policy *domain.LabelPolicy } type res struct { want *domain.LabelPolicy err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "labelpolicy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), policy: &domain.LabelPolicy{ PrimaryColor: "#ffffff", BackgroundColor: "#ffffff", WarnColor: "#ffffff", }, }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "no changes, precondition error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), }, args: args{ ctx: context.Background(), policy: &domain.LabelPolicy{ PrimaryColor: "#ffffff", BackgroundColor: "#ffffff", WarnColor: "#ffffff", FontColor: "#ffffff", PrimaryColorDark: "#ffffff", BackgroundColorDark: "#ffffff", WarnColorDark: "#ffffff", FontColorDark: "#ffffff", HideLoginNameSuffix: true, ErrorMsgPopup: true, DisableWatermark: true, }, }, res: res{ err: caos_errs.IsPreconditionFailed, }, }, { name: "change, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( newDefaultLabelPolicyChangedEvent( context.Background(), "#000000", "#000000", "#000000", "#000000", "#000000", "#000000", "#000000", "#000000", false, false, false), ), ), }, args: args{ ctx: context.Background(), policy: &domain.LabelPolicy{ PrimaryColor: "#000000", BackgroundColor: "#000000", WarnColor: "#000000", FontColor: "#000000", PrimaryColorDark: "#000000", BackgroundColorDark: "#000000", WarnColorDark: "#000000", FontColorDark: "#000000", HideLoginNameSuffix: false, ErrorMsgPopup: false, DisableWatermark: false, }, }, res: res{ want: &domain.LabelPolicy{ ObjectRoot: models.ObjectRoot{ AggregateID: "INSTANCE", ResourceOwner: "INSTANCE", InstanceID: "INSTANCE", }, PrimaryColor: "#000000", BackgroundColor: "#000000", WarnColor: "#000000", FontColor: "#000000", PrimaryColorDark: "#000000", BackgroundColorDark: "#000000", WarnColorDark: "#000000", FontColorDark: "#000000", HideLoginNameSuffix: false, ErrorMsgPopup: false, DisableWatermark: false, }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, } got, err := r.ChangeDefaultLabelPolicy(tt.args.ctx, tt.args.policy) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_ActivateDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore } type args struct { ctx context.Context } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: authz.WithInstanceID(context.Background(), "INSTANCE"), }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "activated, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( instance.NewLabelPolicyActivatedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, ), ), ), }, args: args{ ctx: authz.WithInstanceID(context.Background(), "INSTANCE"), }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, } got, err := r.ActivateDefaultLabelPolicy(tt.args.ctx) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_AddLogoDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context upload *AssetUpload } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "logo", ContentType: "text/css", ObjectType: static.ObjectTypeStyling, File: nil, Size: 0, }, }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "upload failed, error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), storage: mock.NewStorage(t).ExpectPutObjectError(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "logo", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsInternal, }, }, { name: "logo added, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( instance.NewLabelPolicyLogoAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "logo", ), ), ), storage: mock.NewStorage(t).ExpectPutObject(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "logo", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.AddLogoDefaultLabelPolicy(tt.args.ctx, tt.args.upload) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_RemoveLogoDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "asset remove error, internal error", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyLogoAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsInternal, }, }, { name: "logo added, ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectNoError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyLogoAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), expectPush( instance.NewLabelPolicyLogoRemovedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.RemoveLogoDefaultLabelPolicy(tt.args.ctx) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_AddIconDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context upload *AssetUpload } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "icon", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "upload failed, error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), storage: mock.NewStorage(t).ExpectPutObjectError(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "icon", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsInternal, }, }, { name: "icon added, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( instance.NewLabelPolicyIconAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "icon", ), ), ), storage: mock.NewStorage(t).ExpectPutObject(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "icon", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.AddIconDefaultLabelPolicy(tt.args.ctx, tt.args.upload) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_RemoveIconDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "icon removed, ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectNoError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyIconAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), expectPush( instance.NewLabelPolicyIconRemovedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.RemoveIconDefaultLabelPolicy(tt.args.ctx) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_AddLogoDarkDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context instanceID string upload *AssetUpload } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), instanceID: "INSTANCE", upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "logo", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "upload failed, error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), storage: mock.NewStorage(t).ExpectPutObjectError(), }, args: args{ ctx: context.Background(), instanceID: "INSTANCE", upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "logo", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsInternal, }, }, { name: "logo dark added, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( instance.NewLabelPolicyLogoDarkAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "logo", ), ), ), storage: mock.NewStorage(t).ExpectPutObject(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "logo", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.AddLogoDarkDefaultLabelPolicy(tt.args.ctx, tt.args.upload) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_RemoveLogoDarkDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "logo dark removed, not ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyLogoDarkAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsInternal, }, }, { name: "logo dark removed, ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectNoError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyLogoDarkAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), expectPush( instance.NewLabelPolicyLogoDarkRemovedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.RemoveLogoDarkDefaultLabelPolicy(tt.args.ctx) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_AddIconDarkDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context upload *AssetUpload } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "icon", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "upload failed, error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), storage: mock.NewStorage(t).ExpectPutObjectError(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "icon", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsInternal, }, }, { name: "icon dark added, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( instance.NewLabelPolicyIconDarkAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "icon", ), ), ), storage: mock.NewStorage(t).ExpectPutObject(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "icon", ContentType: "image", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.AddIconDarkDefaultLabelPolicy(tt.args.ctx, tt.args.upload) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_RemoveIconDarkDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "icon dark removed, not ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyIconDarkAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsInternal, }, }, { name: "icon dark removed, ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectNoError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyIconDarkAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), expectPush( instance.NewLabelPolicyIconDarkRemovedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.RemoveIconDarkDefaultLabelPolicy(tt.args.ctx) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_AddFontDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context upload *AssetUpload } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "font", ContentType: "ttf", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "upload failed, error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), ), storage: mock.NewStorage(t).ExpectPutObjectError(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "font", ContentType: "ttf", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ err: caos_errs.IsInternal, }, }, { name: "font added, ok", fields: fields{ eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), ), expectPush( instance.NewLabelPolicyFontAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "font", ), ), ), storage: mock.NewStorage(t).ExpectPutObject(), }, args: args{ ctx: context.Background(), upload: &AssetUpload{ ResourceOwner: "IAM", ObjectName: "font", ContentType: "ttf", ObjectType: static.ObjectTypeStyling, File: bytes.NewReader([]byte("test")), Size: 4, }, }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.AddFontDefaultLabelPolicy(tt.args.ctx, tt.args.upload) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func TestCommandSide_RemoveFontDefaultLabelPolicy(t *testing.T) { type fields struct { eventstore *eventstore.Eventstore storage static.Storage } type args struct { ctx context.Context } type res struct { want *domain.ObjectDetails err func(error) bool } tests := []struct { name string fields fields args args res res }{ { name: "label policy not existing, not found error", fields: fields{ eventstore: eventstoreExpect( t, expectFilter(), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsNotFound, }, }, { name: "font remove from storage not possible, internla error", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyFontAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ err: caos_errs.IsInternal, }, }, { name: "font added, ok", fields: fields{ storage: mock.NewMockStorage(gomock.NewController(t)).ExpectRemoveObjectNoError(), eventstore: eventstoreExpect( t, expectFilter( eventFromEventPusher( instance.NewLabelPolicyAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", "#ffffff", true, true, true, ), ), eventFromEventPusher( instance.NewLabelPolicyFontAddedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), expectPush( instance.NewLabelPolicyFontRemovedEvent(context.Background(), &instance.NewAggregate("INSTANCE").Aggregate, "key", ), ), ), }, args: args{ ctx: context.Background(), }, res: res{ want: &domain.ObjectDetails{ ResourceOwner: "INSTANCE", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Commands{ eventstore: tt.fields.eventstore, static: tt.fields.storage, } got, err := r.RemoveFontDefaultLabelPolicy(tt.args.ctx) if tt.res.err == nil { assert.NoError(t, err) } if tt.res.err != nil && !tt.res.err(err) { t.Errorf("got wrong err: %v ", err) } if tt.res.err == nil { assert.Equal(t, tt.res.want, got) } }) } } func newDefaultLabelPolicyChangedEvent(ctx context.Context, primaryColor, backgroundColor, warnColor, fontColor, primaryColorDark, backgroundColorDark, warnColorDark, fontColorDark string, hideLoginNameSuffix, errMsgPopup, disableWatermark bool) *instance.LabelPolicyChangedEvent { event, _ := instance.NewLabelPolicyChangedEvent(ctx, &instance.NewAggregate("INSTANCE").Aggregate, []policy.LabelPolicyChanges{ policy.ChangePrimaryColor(primaryColor), policy.ChangeBackgroundColor(backgroundColor), policy.ChangeWarnColor(warnColor), policy.ChangeFontColor(fontColor), policy.ChangePrimaryColorDark(primaryColorDark), policy.ChangeBackgroundColorDark(backgroundColorDark), policy.ChangeWarnColorDark(warnColorDark), policy.ChangeFontColorDark(fontColorDark), policy.ChangeHideLoginNameSuffix(hideLoginNameSuffix), policy.ChangeErrorMsgPopup(errMsgPopup), policy.ChangeDisableWatermark(disableWatermark), }, ) return event }