From 8bb5bc8225ab4f19d81a9fa3cebeacfc394c8568 Mon Sep 17 00:00:00 2001 From: ko_oler <kooler89@gmail.com> Date: Fri, 21 Jul 2023 10:53:07 +0300 Subject: [PATCH] =?UTF-8?q?=D0=BF=D1=80=D0=B0=D0=B2=D0=BA=D0=B8=20=D0=BF?= =?UTF-8?q?=D0=BE=20=D0=9F=D0=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/setup/collection.go | 16 ++++++------- pkg/setup/collection_test.go | 45 ++---------------------------------- pkg/setup/setup_test.go | 10 ++++---- 3 files changed, 14 insertions(+), 57 deletions(-) diff --git a/pkg/setup/collection.go b/pkg/setup/collection.go index 7f2e1821..cee9d620 100644 --- a/pkg/setup/collection.go +++ b/pkg/setup/collection.go @@ -22,7 +22,7 @@ var ( // todo добавлено в релизе v0.0.16: временный флаг для установки метадаты во все коллекции расширений const ( - alwaysSetSchema = true + _alwaysSetSchema = true ) type CollectionsOption func(c *CollectionConfig) @@ -76,24 +76,22 @@ func DeleteCollectionIfRemove() CollectionsOption { } } -func isCollsSchemaMetadataEqual(s1, s2 *schema.Schema) bool { +func isMetadataExtensionEqual(s1, s2 *schema.Schema) bool { if s1.Metadata == nil && s2.Metadata == nil { return true } - if s1.Metadata != nil && s1.Metadata[extension.ExtensionMetadataKey] != "" { - if s2.Metadata == nil || s2.Metadata[extension.ExtensionMetadataKey] != s1.Metadata[extension.ExtensionMetadataKey] { - return false - } + if s1.Metadata == nil || s2.Metadata == nil { + return false } - return true + return s1.Metadata[extension.ExtensionMetadataKey] == s2.Metadata[extension.ExtensionMetadataKey] } func DefaultUpdateCollectionStrategy() CollectionsOption { return func(c *CollectionConfig) { c.UpdateFn = func(s *Setup, exist, collection *collections.Collection) (*collections.Collection, bool, bool, error) { - if !alwaysSetSchema && !s.IsForce() && !collection.IsView() && !exist.IsView() && !isCollsSchemaMetadataEqual(collection.Schema, exist.Schema) { + if !_alwaysSetSchema && !s.IsForce() && !collection.IsView() && !exist.IsView() && !isMetadataExtensionEqual(collection.Schema, exist.Schema) { return nil, false, false, errors.WithDetailf(collections.ErrAlreadyExists, "Коллекция с идентификатором '%s' "+ "уже существует. Удалите ее или вызовите установку расширения с флагом Force", collection.ID) } @@ -110,7 +108,7 @@ func DefaultUpdateCollectionStrategy() CollectionsOption { update = update && *exist.View == *collection.View } - setSchema = alwaysSetSchema || !collection.IsView() && !reflect.DeepEqual(exist.Schema, collection.Schema) + setSchema = _alwaysSetSchema || !collection.IsView() && !reflect.DeepEqual(exist.Schema, collection.Schema) return collection, update, setSchema, nil } diff --git a/pkg/setup/collection_test.go b/pkg/setup/collection_test.go index 2c8cea2b..a88dad0a 100644 --- a/pkg/setup/collection_test.go +++ b/pkg/setup/collection_test.go @@ -108,7 +108,7 @@ func TestSetup_InstallCollections(t *testing.T) { assert.NoError(t, err) }, }, - //todo после удаления флага alwaysSetSchema данный тест будет работать, сейчас он будет всегда падать, поэтому пока закомментирован + //todo после удаления флага _alwaysSetSchema данный тест будет работать, сейчас он будет всегда падать, поэтому пока закомментирован //{ //name: "Fail to update user collection with same id as in extensions collection", //collections: []*collections.Collection{{ID: "1", SpaceID: "sp", EnvID: "env", Schema: schema.New("name", field.String()).WithMetadata(extension.ExtensionMetadataKey, "test-extension")}}, @@ -151,7 +151,7 @@ func TestSetup_InstallCollections(t *testing.T) { assert.NoError(t, err) }, }, - //todo после удаления флага alwaysSetSchema данный тест будет работать, сейчас он будет всегда падать, поэтому пока закомментирован + //todo после удаления флага _alwaysSetSchema данный тест будет работать, сейчас он будет всегда падать, поэтому пока закомментирован //{ // name: "Update view collection with the same id to new view collection", // collections: []*collections.Collection{{ID: "1", SpaceID: "sp", EnvID: "env", View: &collections.View{SpaceID: "sp3", EnvID: environments.DefaultEnvironment, CollectionID: "3"}}}, @@ -202,44 +202,3 @@ func TestSetup_InstallCollections(t *testing.T) { }) } } - -func Test_isCollsSchemaMetadataEqual(t *testing.T) { - type args struct { - } - tests := []struct { - name string - s1 *schema.Schema - s2 *schema.Schema - want bool - }{ - { - "Not equal #1 (no metadata)", - schema.New("name", field.String()).WithMetadata(extension.ExtensionMetadataKey, "test"), - schema.New("name", field.String()), - false, - }, - { - "Not equal #2 (different metadata)", - schema.New("name", field.String()).WithMetadata(extension.ExtensionMetadataKey, "test"), - schema.New("name", field.String()).WithMetadata("test", "test"), - false, - }, - { - "Equal #1 (no metadata)", - schema.New("name", field.String()), - schema.New("name", field.String()), - true, - }, - { - "Equal #2 (equal metadata)", - schema.New("name", field.String()).WithMetadata(extension.ExtensionMetadataKey, "test"), - schema.New("name", field.String()).WithMetadata(extension.ExtensionMetadataKey, "test"), - true, - }, - } - for _, tt := range tests { - t.Run(tt.name, func(t *testing.T) { - assert.Equalf(t, tt.want, isCollsSchemaMetadataEqual(tt.s1, tt.s2), "isCollsSchemaMetadataEqual(%v, %v)", tt.s1, tt.s2) - }) - } -} diff --git a/pkg/setup/setup_test.go b/pkg/setup/setup_test.go index e722041e..6a1b6f7b 100644 --- a/pkg/setup/setup_test.go +++ b/pkg/setup/setup_test.go @@ -179,7 +179,7 @@ func TestSetupInstall(t *testing.T) { t.Run("Success, update existing records", func(t *testing.T) { envMocks := &environmentMock.Environments{} - //todo после удаления флага alwaysSetSchema убрать лишний вызов + //todo после удаления флага _alwaysSetSchema убрать лишний вызов envMocks.On("Migrate", mock.Anything, spaceID, envID, &environments.MigrateOptions{Wait: true}). Return(nil).Once() @@ -187,7 +187,7 @@ func TestSetupInstall(t *testing.T) { collsMock.On("Get", mock.Anything, spaceID, envID, "coll1"). Return(&collections.Collection{ID: "coll1", SpaceID: spaceID, EnvID: envID, Schema: schema.New(), Name: "Коллекция старая"}, nil).Once() collsMock.On("Update", mock.Anything, &collections.Collection{ID: "coll1", SpaceID: spaceID, EnvID: envID, Schema: schema.New(), Name: "Коллекция"}).Return(nil).Once() - //todo после удаления флага alwaysSetSchema убрать лишний вызов + //todo после удаления флага _alwaysSetSchema убрать лишний вызов collsMock.On("SetSchema", mock.Anything, "sp", "env", "coll1", schema.New()).Return(nil).Once() rMock := &rolesMock.Roles{} @@ -245,7 +245,7 @@ func TestSetupInstall(t *testing.T) { }) t.Run("Success, with force", func(t *testing.T) { - //todo после удаления флага alwaysSetSchema убрать лишний вызов + //todo после удаления флага _alwaysSetSchema убрать лишний вызов envMocks := &environmentMock.Environments{} envMocks.On("Migrate", mock.Anything, spaceID, envID, &environments.MigrateOptions{Wait: true}). Return(nil).Once() @@ -254,7 +254,7 @@ func TestSetupInstall(t *testing.T) { collsMock.On("Get", mock.Anything, spaceID, envID, "coll1"). Return(&collections.Collection{ID: "coll1", SpaceID: spaceID, EnvID: envID, Schema: schema.New(), Name: "Коллекция старая"}, nil).Once() collsMock.On("Update", mock.Anything, &collections.Collection{ID: "coll1", SpaceID: spaceID, EnvID: envID, Schema: schema.New(), Name: "Коллекция"}).Return(nil).Once() - //todo после удаления флага alwaysSetSchema убрать лишний вызов + //todo после удаления флага _alwaysSetSchema убрать лишний вызов collsMock.On("SetSchema", mock.Anything, "sp", "env", "coll1", schema.New()).Return(nil).Once() rMock := &rolesMock.Roles{} @@ -290,7 +290,7 @@ func TestSetupInstall(t *testing.T) { Clients: clMock, Roles: rMock, Items: itmMock, - //todo после удаления флага alwaysSetSchema убрать + //todo после удаления флага _alwaysSetSchema убрать Environments: envMocks, }, t) setup = setup.WithForce(true) -- GitLab