Merge pull request #7108 from kofj/policy_manager_ng_ut

Policy manager ng UT
goharbor · Mar 11, 2019 · 2911ac7 · 2911ac7
2 parents c632c1e + 1039800
commit 2911ac7
Show file tree

Hide file tree

Showing 4 changed files with 535 additions and 2 deletions.
diff --git a/make/migrations/postgresql/0006_ng_replication.up.sql b/make/migrations/postgresql/0006_ng_replication.up.sql
@@ -12,4 +12,25 @@ CREATE TABLE registry (
  creation_time timestamp default CURRENT_TIMESTAMP,
  update_time timestamp default CURRENT_TIMESTAMP,
  CONSTRAINT unique_registry_name UNIQUE (name)
-);
+);
+
+CREATE TABLE "replication_policy_ng" (
+  "id" SERIAL PRIMARY KEY NOT NULL,
+  "name" varchar(256),
+  "description" text,
+  "creator" varchar(256),
+  "src_registry_id" int4,
+  "src_namespaces" varchar(256),
+  "dest_registry_id" int4,
+  "dest_namespace" varchar(256),
+  "override" bool NOT NULL DEFAULT false,
+  "enabled" bool NOT NULL DEFAULT true,
+  "cron_str" varchar(256),
+  "filters" varchar(1024),
+  "replicate_deletion" bool NOT NULL DEFAULT false,
+  "start_time" timestamp(6),
+  "deleted" bool NOT NULL DEFAULT false,
+  "creation_time" timestamp(6) DEFAULT now(),
+  "update_time" timestamp(6) DEFAULT now(),
+  CONSTRAINT unique_policy_ng_name UNIQUE ("name")
+);
diff --git a/src/replication/ng/dao/dao_test.go b/src/replication/ng/dao/dao_test.go
@@ -23,5 +23,19 @@ import (
 
 func TestMain(m *testing.M) {
 	dao.PrepareTestForPostgresSQL()
-	os.Exit(m.Run())
+
+	var code = m.Run()
+
+	// clear test database
+	var clearSqls = []string{
+		`DROP TABLE "access", "access_log", "admin_job", "alembic_version", "clair_vuln_timestamp",
+  "harbor_label", "harbor_resource_label", "harbor_user", "img_scan_job", "img_scan_overview",
+  "job_log", "project", "project_member", "project_metadata", "properties", "registry",
+  "replication_immediate_trigger", "replication_job", "replication_policy", "replication_policy_ng",
+  "replication_target", "repository", "robot", "role", "schema_migrations", "user_group";`,
+		`DROP FUNCTION "update_update_time_at_column"();`,
+	}
+	dao.PrepareTestData(clearSqls, nil)
+
+	os.Exit(code)
 }
diff --git a/src/replication/ng/dao/policy_test.go b/src/replication/ng/dao/policy_test.go
@@ -0,0 +1,311 @@
+package dao
+
+import (
+	"testing"
+
+	"github.com/astaxie/beego/orm"
+	"github.com/goharbor/harbor/src/replication/ng/dao/models"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+var (
+	testPolic1 = &models.RepPolicy{
+		// ID:                999,
+		Name:              "Policy Test 1",
+		Description:       "Policy Description",
+		Creator:           "someone",
+		SrcRegistryID:     123,
+		SrcNamespaces:     "ns1,ns2,ns3",
+		DestRegistryID:    456,
+		DestNamespace:     "target_ns",
+		ReplicateDeletion: true,
+		Override:          true,
+		Enabled:           true,
+		Trigger:           "{\"type\":\"\",\"trigger_settings\":null}",
+		Filters:           "[{\"type\":\"registry\",\"value\":\"abc\"}]",
+	}
+
+	testPolic2 = &models.RepPolicy{
+		// ID:                999,
+		Name:              "Policy Test 2",
+		Description:       "Policy Description",
+		Creator:           "someone",
+		SrcRegistryID:     123,
+		SrcNamespaces:     "ns1,ns2,ns3",
+		DestRegistryID:    456,
+		DestNamespace:     "target_ns",
+		ReplicateDeletion: true,
+		Override:          true,
+		Enabled:           true,
+		Trigger:           "{\"type\":\"\",\"trigger_settings\":null}",
+		Filters:           "[{\"type\":\"registry\",\"value\":\"abc\"}]",
+	}
+
+	testPolic3 = &models.RepPolicy{
+		// ID:                999,
+		Name:              "Policy Test 3",
+		Description:       "Policy Description",
+		Creator:           "someone",
+		SrcRegistryID:     123,
+		SrcNamespaces:     "ns1,ns2,ns3",
+		DestRegistryID:    456,
+		DestNamespace:     "target_ns",
+		ReplicateDeletion: true,
+		Override:          true,
+		Enabled:           true,
+		Trigger:           "{\"type\":\"\",\"trigger_settings\":null}",
+		Filters:           "[{\"type\":\"registry\",\"value\":\"abc\"}]",
+	}
+)
+
+func TestAddRepPolicy(t *testing.T) {
+	tests := []struct {
+		name    string
+		policy  *models.RepPolicy
+		want    int64
+		wantErr bool
+	}{
+		{name: "AddRepPolicy 1", policy: testPolic1, want: 1},
+		{name: "AddRepPolicy 2", policy: testPolic2, want: 2},
+		{name: "AddRepPolicy 3", policy: testPolic3, want: 3},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := AddRepPolicy(tt.policy)
+
+			if tt.wantErr {
+				require.NotNil(t, err, "wantErr: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestGetTotalOfRepPolicies(t *testing.T) {
+	type args struct {
+		name      string
+		namespace string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    int64
+		wantErr bool
+	}{
+		{name: "GetTotalOfRepPolicies 1", args: args{name: "Test 1"}, want: 1},
+		{name: "GetTotalOfRepPolicies 2", args: args{name: "Test"}, want: 3},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := GetTotalOfRepPolicies(tt.args.name, tt.args.namespace)
+			if tt.wantErr {
+				require.NotNil(t, err, "wantErr: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestGetPolicies(t *testing.T) {
+	type args struct {
+		name      string
+		namespace string
+		page      int64
+		pageSize  int64
+	}
+	tests := []struct {
+		name         string
+		args         args
+		wantPolicies []*models.RepPolicy
+		wantErr      bool
+	}{
+		{name: "GetTotalOfRepPolicies nil", args: args{name: "Test 0"}, wantPolicies: []*models.RepPolicy{}},
+		{name: "GetTotalOfRepPolicies 1", args: args{name: "Test 1"}, wantPolicies: []*models.RepPolicy{testPolic1}},
+		{name: "GetTotalOfRepPolicies 2", args: args{name: "Test", page: 1, pageSize: 2}, wantPolicies: []*models.RepPolicy{testPolic1, testPolic2}},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotPolicies, err := GetPolicies(tt.args.name, tt.args.namespace, tt.args.page, tt.args.pageSize)
+			if tt.wantErr {
+				require.NotNil(t, err, "wantErr: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			for i, gotPolicy := range gotPolicies {
+				assert.Equal(t, tt.wantPolicies[i].Name, gotPolicy.Name)
+				assert.Equal(t, tt.wantPolicies[i].Description, gotPolicy.Description)
+				assert.Equal(t, tt.wantPolicies[i].Creator, gotPolicy.Creator)
+				assert.Equal(t, tt.wantPolicies[i].SrcRegistryID, gotPolicy.SrcRegistryID)
+				assert.Equal(t, tt.wantPolicies[i].SrcNamespaces, gotPolicy.SrcNamespaces)
+				assert.Equal(t, tt.wantPolicies[i].DestRegistryID, gotPolicy.DestRegistryID)
+				assert.Equal(t, tt.wantPolicies[i].DestNamespace, gotPolicy.DestNamespace)
+				assert.Equal(t, tt.wantPolicies[i].ReplicateDeletion, gotPolicy.ReplicateDeletion)
+				assert.Equal(t, tt.wantPolicies[i].Override, gotPolicy.Override)
+				assert.Equal(t, tt.wantPolicies[i].Enabled, gotPolicy.Enabled)
+				assert.Equal(t, tt.wantPolicies[i].Trigger, gotPolicy.Trigger)
+				assert.Equal(t, tt.wantPolicies[i].Filters, gotPolicy.Filters)
+			}
+		})
+	}
+}
+
+func TestGetRepPolicy(t *testing.T) {
+	tests := []struct {
+		name       string
+		id         int64
+		wantPolicy *models.RepPolicy
+		wantErr    bool
+	}{
+		{name: "GetRepPolicy 1", id: 1, wantPolicy: testPolic1},
+		{name: "GetRepPolicy 2", id: 2, wantPolicy: testPolic2},
+		{name: "GetRepPolicy 3", id: 3, wantPolicy: testPolic3},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotPolicy, err := GetRepPolicy(tt.id)
+			if tt.wantErr {
+				require.NotNil(t, err, "wantErr: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			assert.Equal(t, tt.wantPolicy.Name, gotPolicy.Name)
+			assert.Equal(t, tt.wantPolicy.Description, gotPolicy.Description)
+			assert.Equal(t, tt.wantPolicy.Creator, gotPolicy.Creator)
+			assert.Equal(t, tt.wantPolicy.SrcRegistryID, gotPolicy.SrcRegistryID)
+			assert.Equal(t, tt.wantPolicy.SrcNamespaces, gotPolicy.SrcNamespaces)
+			assert.Equal(t, tt.wantPolicy.DestRegistryID, gotPolicy.DestRegistryID)
+			assert.Equal(t, tt.wantPolicy.DestNamespace, gotPolicy.DestNamespace)
+			assert.Equal(t, tt.wantPolicy.ReplicateDeletion, gotPolicy.ReplicateDeletion)
+			assert.Equal(t, tt.wantPolicy.Override, gotPolicy.Override)
+			assert.Equal(t, tt.wantPolicy.Enabled, gotPolicy.Enabled)
+			assert.Equal(t, tt.wantPolicy.Trigger, gotPolicy.Trigger)
+			assert.Equal(t, tt.wantPolicy.Filters, gotPolicy.Filters)
+		})
+	}
+}
+
+func TestGetRepPolicyByName(t *testing.T) {
+	type args struct {
+		name string
+	}
+	tests := []struct {
+		name       string
+		args       args
+		wantPolicy *models.RepPolicy
+		wantErr    bool
+	}{
+		{name: "GetRepPolicyByName 1", args: args{name: testPolic1.Name}, wantPolicy: testPolic1},
+		{name: "GetRepPolicyByName 2", args: args{name: testPolic2.Name}, wantPolicy: testPolic2},
+		{name: "GetRepPolicyByName 3", args: args{name: testPolic3.Name}, wantPolicy: testPolic3},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotPolicy, err := GetRepPolicyByName(tt.args.name)
+			if tt.wantErr {
+				require.NotNil(t, err, "wantErr: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			assert.Equal(t, tt.wantPolicy.Name, gotPolicy.Name)
+			assert.Equal(t, tt.wantPolicy.Description, gotPolicy.Description)
+			assert.Equal(t, tt.wantPolicy.Creator, gotPolicy.Creator)
+			assert.Equal(t, tt.wantPolicy.SrcRegistryID, gotPolicy.SrcRegistryID)
+			assert.Equal(t, tt.wantPolicy.SrcNamespaces, gotPolicy.SrcNamespaces)
+			assert.Equal(t, tt.wantPolicy.DestRegistryID, gotPolicy.DestRegistryID)
+			assert.Equal(t, tt.wantPolicy.DestNamespace, gotPolicy.DestNamespace)
+			assert.Equal(t, tt.wantPolicy.ReplicateDeletion, gotPolicy.ReplicateDeletion)
+			assert.Equal(t, tt.wantPolicy.Override, gotPolicy.Override)
+			assert.Equal(t, tt.wantPolicy.Enabled, gotPolicy.Enabled)
+			assert.Equal(t, tt.wantPolicy.Trigger, gotPolicy.Trigger)
+			assert.Equal(t, tt.wantPolicy.Filters, gotPolicy.Filters)
+		})
+	}
+}
+
+func TestUpdateRepPolicy(t *testing.T) {
+	type args struct {
+		policy *models.RepPolicy
+		props  []string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		wantErr bool
+	}{
+		{name: "UpdateRepPolicy Want Error", args: args{policy: nil}, wantErr: true},
+		{
+			name: "UpdateRepPolicy 1",
+			args: args{
+				policy: &models.RepPolicy{ID: 1, Description: "Policy Description 1", Creator: "Someone 1"},
+				props:  []string{"description", "creator"},
+			},
+		},
+		{
+			name: "UpdateRepPolicy 2",
+			args: args{
+				policy: &models.RepPolicy{ID: 2, Description: "Policy Description 2", Creator: "Someone 2"},
+				props:  []string{"description", "creator"},
+			},
+		},
+		{
+			name: "UpdateRepPolicy 3",
+			args: args{
+				policy: &models.RepPolicy{ID: 3, Description: "Policy Description 3", Creator: "Someone 3"},
+				props:  []string{"description", "creator"},
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := UpdateRepPolicy(tt.args.policy, tt.args.props...)
+
+			if tt.wantErr {
+				require.NotNil(t, err, "Error: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			gotPolicy, err := GetRepPolicy(tt.args.policy.ID)
+
+			require.Nil(t, err)
+			assert.Equal(t, tt.args.policy.Description, gotPolicy.Description)
+			assert.Equal(t, tt.args.policy.Creator, gotPolicy.Creator)
+		})
+	}
+}
+
+func TestDeleteRepPolicy(t *testing.T) {
+	tests := []struct {
+		name    string
+		id      int64
+		wantErr bool
+	}{
+		{name: "DeleteRepPolicy 1", id: 1, wantErr: false},
+		{name: "DeleteRepPolicy 2", id: 2, wantErr: false},
+		{name: "DeleteRepPolicy 3", id: 3, wantErr: false},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := DeleteRepPolicy(tt.id)
+			if tt.wantErr {
+				require.NotNil(t, err, "wantErr: %s", err)
+				return
+			}
+
+			require.Nil(t, err)
+			_, err = GetRepPolicy(tt.id)
+			require.NotNil(t, err)
+			assert.Equal(t, err, orm.ErrNoRows)
+		})
+	}
+}