From c7dfc82845131d1d8bf6dd7a510f11cfc34e3bc4 Mon Sep 17 00:00:00 2001
From: KK <68334452+healthjyk@users.noreply.github.com>
Date: Wed, 10 Jan 2024 17:44:22 +0800
Subject: [PATCH] fix: a bug of the map assertion (#743)

---
 .../generators/workload/job_generator_test.go      | 12 ++++++------
 .../workload/network/ports_generator_test.go       | 12 ++++++------
 .../generators/workload/service_generator_test.go  | 14 +++++++-------
 .../generators/workload/workload_generator_test.go |  8 ++++----
 pkg/workspace/util.go                              |  2 +-
 pkg/workspace/util_test.go                         |  4 ++--
 6 files changed, 26 insertions(+), 26 deletions(-)

diff --git a/pkg/modules/generators/workload/job_generator_test.go b/pkg/modules/generators/workload/job_generator_test.go
index 270cf351..0c380370 100644
--- a/pkg/modules/generators/workload/job_generator_test.go
+++ b/pkg/modules/generators/workload/job_generator_test.go
@@ -45,10 +45,10 @@ func TestNewJobGenerator(t *testing.T) {
 	expectedAppName := "test"
 	expectedJob := &workload.Job{}
 	expectedJobConfig := apiv1.GenericConfig{
-		"labels": map[string]any{
+		"labels": apiv1.GenericConfig{
 			"workload-type": "Job",
 		},
-		"annotations": map[string]any{
+		"annotations": apiv1.GenericConfig{
 			"workload-type": "Job",
 		},
 	}
@@ -72,10 +72,10 @@ func TestNewJobGeneratorFunc(t *testing.T) {
 	expectedAppName := "test"
 	expectedJob := &workload.Job{}
 	expectedJobConfig := apiv1.GenericConfig{
-		"labels": map[string]any{
+		"labels": apiv1.GenericConfig{
 			"workload-type": "Job",
 		},
-		"annotations": map[string]any{
+		"annotations": apiv1.GenericConfig{
 			"workload-type": "Job",
 		},
 	}
@@ -110,10 +110,10 @@ func TestJobGenerator_Generate(t *testing.T) {
 			expectedAppName: "test",
 			expectedJob:     &workload.Job{},
 			expectedJobConfig: apiv1.GenericConfig{
-				"labels": map[string]any{
+				"labels": apiv1.GenericConfig{
 					"workload-type": "Job",
 				},
-				"annotations": map[string]any{
+				"annotations": apiv1.GenericConfig{
 					"workload-type": "Job",
 				},
 			},
diff --git a/pkg/modules/generators/workload/network/ports_generator_test.go b/pkg/modules/generators/workload/network/ports_generator_test.go
index a211e26e..07a4ff87 100644
--- a/pkg/modules/generators/workload/network/ports_generator_test.go
+++ b/pkg/modules/generators/workload/network/ports_generator_test.go
@@ -87,10 +87,10 @@ func TestValidatePortConfig(t *testing.T) {
 			name: "valid port config",
 			portConfig: apiv1.GenericConfig{
 				"type": "alicloud",
-				"labels": map[string]any{
+				"labels": apiv1.GenericConfig{
 					"kusionstack.io/control": "true",
 				},
-				"annotations": map[string]any{
+				"annotations": apiv1.GenericConfig{
 					"service.beta.kubernetes.io/alibaba-cloud-loadbalancer-spec": "slb.s1.small",
 				},
 			},
@@ -144,10 +144,10 @@ func TestCompletePort(t *testing.T) {
 			},
 			portConfig: apiv1.GenericConfig{
 				"type": "alicloud",
-				"labels": map[string]any{
+				"labels": apiv1.GenericConfig{
 					"kusionstack.io/control": "true",
 				},
-				"annotations": map[string]any{
+				"annotations": apiv1.GenericConfig{
 					"service.beta.kubernetes.io/alibaba-cloud-loadbalancer-spec": "slb.s1.small",
 				},
 			},
@@ -262,10 +262,10 @@ func TestPortsGenerator_Generate(t *testing.T) {
 					},
 					portConfig: apiv1.GenericConfig{
 						"type": "alicloud",
-						"labels": map[string]any{
+						"labels": apiv1.GenericConfig{
 							"kusionstack.io/control": "true",
 						},
-						"annotations": map[string]any{
+						"annotations": apiv1.GenericConfig{
 							"service.beta.kubernetes.io/alibaba-cloud-loadbalancer-spec": "slb.s1.small",
 						},
 					},
diff --git a/pkg/modules/generators/workload/service_generator_test.go b/pkg/modules/generators/workload/service_generator_test.go
index 56d58fe2..eea1f350 100644
--- a/pkg/modules/generators/workload/service_generator_test.go
+++ b/pkg/modules/generators/workload/service_generator_test.go
@@ -218,19 +218,19 @@ status: {}
 				},
 				serviceConfig: apiv1.GenericConfig{
 					"type": "CollaSet",
-					"labels": map[string]any{
+					"labels": apiv1.GenericConfig{
 						"service-workload-type": "CollaSet",
 					},
-					"annotations": map[string]any{
+					"annotations": apiv1.GenericConfig{
 						"service-workload-type": "CollaSet",
 					},
 				},
 				portConfig: apiv1.GenericConfig{
 					"type": "alicloud",
-					"labels": map[string]any{
+					"labels": apiv1.GenericConfig{
 						"kusionstack.io/control": "true",
 					},
-					"annotations": map[string]any{
+					"annotations": apiv1.GenericConfig{
 						"service.beta.kubernetes.io/alibaba-cloud-loadbalancer-spec": "slb.s1.small",
 					},
 				},
@@ -276,16 +276,16 @@ status: {}
 				},
 				serviceConfig: apiv1.GenericConfig{
 					"replicas": 4,
-					"labels": map[string]any{
+					"labels": apiv1.GenericConfig{
 						"service-workload-type": "Deployment",
 					},
 				},
 				portConfig: apiv1.GenericConfig{
 					"type": "alicloud",
-					"labels": map[string]any{
+					"labels": apiv1.GenericConfig{
 						"kusionstack.io/control": "true",
 					},
-					"annotations": map[string]any{
+					"annotations": apiv1.GenericConfig{
 						"service.beta.kubernetes.io/alibaba-cloud-loadbalancer-spec": "slb.s1.small",
 					},
 				},
diff --git a/pkg/modules/generators/workload/workload_generator_test.go b/pkg/modules/generators/workload/workload_generator_test.go
index f6e85950..581a28f5 100644
--- a/pkg/modules/generators/workload/workload_generator_test.go
+++ b/pkg/modules/generators/workload/workload_generator_test.go
@@ -151,10 +151,10 @@ func TestWorkloadGenerator_Generate(t *testing.T) {
 				},
 				"port": {
 					"type": "alicloud",
-					"labels": map[string]any{
+					"labels": apiv1.GenericConfig{
 						"kusionstack.io/control": "true",
 					},
-					"annotations": map[string]any{
+					"annotations": apiv1.GenericConfig{
 						"service.beta.kubernetes.io/alibaba-cloud-loadbalancer-spec": "slb.s1.small",
 					},
 				},
@@ -392,11 +392,11 @@ func TestCompleteBaseWorkload(t *testing.T) {
 				},
 			},
 			config: apiv1.GenericConfig{
-				"labels": map[string]any{
+				"labels": apiv1.GenericConfig{
 					"k1": "v1-ws",
 					"k3": "v3-ws",
 				},
-				"annotations": map[string]any{
+				"annotations": apiv1.GenericConfig{
 					"k1": "v1-ws",
 				},
 				"replicas": 4,
diff --git a/pkg/workspace/util.go b/pkg/workspace/util.go
index bd0e1dbf..ba9b5bbd 100644
--- a/pkg/workspace/util.go
+++ b/pkg/workspace/util.go
@@ -247,7 +247,7 @@ func GetMapFromGenericConfig(config v1.GenericConfig, key string) (map[string]an
 	if !ok {
 		return nil, nil
 	}
-	m, ok := value.(map[string]any)
+	m, ok := value.(v1.GenericConfig)
 	if !ok {
 		return nil, fmt.Errorf("the value of %s is not map", key)
 	}
diff --git a/pkg/workspace/util_test.go b/pkg/workspace/util_test.go
index 8e48b876..756f366c 100644
--- a/pkg/workspace/util_test.go
+++ b/pkg/workspace/util_test.go
@@ -12,11 +12,11 @@ func mockGenericConfig() v1.GenericConfig {
 	return v1.GenericConfig{
 		"int_type_field":    2,
 		"string_type_field": "kusion",
-		"map_type_field": map[string]any{
+		"map_type_field": v1.GenericConfig{
 			"k1": "v1",
 			"k2": 2,
 		},
-		"string_map_type_field": map[string]any{
+		"string_map_type_field": v1.GenericConfig{
 			"k1": "v1",
 			"k2": "v2",
 		},