fix failed to apply Label/Annotation Overriders

Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
karmada-io · Dec 16, 2022 · c3f5eaa · c3f5eaa
1 parent 9b626c0
commit c3f5eaa
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 6 deletions.
diff --git a/pkg/util/overridemanager/labelannotationoverrider.go b/pkg/util/overridemanager/labelannotationoverrider.go
@@ -39,6 +39,13 @@ func buildLabelAnnotationOverriderPatches(rawObj *unstructured.Unstructured, ove
 				continue
 			}
 		case policyv1alpha1.OverriderOpAdd:
+			_, exist, _ := unstructured.NestedStringMap(rawObj.Object, path...)
+			if exist {
+				break
+			}
+			if err := unstructured.SetNestedStringMap(rawObj.Object, map[string]string{}, path...); err != nil {
+				continue
+			}
 		}
 		patches = append(patches, overrideOption{
 			Op:    string(overrider.Operator),

diff --git a/pkg/util/overridemanager/labelannotationoverrider_test.go b/pkg/util/overridemanager/labelannotationoverrider_test.go
@@ -23,12 +23,35 @@ func Test_applyLabelsOverriders(t *testing.T) {
 		"foo": "foo",
 		"bar": "bar",
 	}
+	deployment2 := helper.NewDeployment(metav1.NamespaceDefault, "test")
+	deployment2.Labels = nil
 	tests := []struct {
 		name    string
 		args    args
 		want    map[string]string
 		wantErr bool
 	}{
+		{
+			name: "test empty labels",
+			args: args{
+				rawObj: func() *unstructured.Unstructured {
+					deploymentObj, _ := utilhelper.ToUnstructured(deployment2)
+					return deploymentObj
+				}(),
+				commandOverriders: []policyv1alpha1.LabelAnnotationOverrider{
+					{
+						Operator: policyv1alpha1.OverriderOpAdd,
+						Value: map[string]string{
+							"test2": "test2",
+						},
+					},
+				},
+			},
+			want: map[string]string{
+				"test2": "test2",
+			},
+			wantErr: false,
+		},
 		{
 			name: "test labels replace",
 			args: args{
@@ -122,14 +145,37 @@ func Test_applyAnnotationsOverriders(t *testing.T) {
 		"foo": "foo",
 		"bar": "bar",
 	}
+	deployment2 := helper.NewDeployment(metav1.NamespaceDefault, "test")
+	deployment2.Annotations = nil
 	tests := []struct {
 		name    string
 		args    args
 		want    map[string]string
 		wantErr bool
 	}{
 		{
-			name: "test labels replace",
+			name: "test empty annotations",
+			args: args{
+				rawObj: func() *unstructured.Unstructured {
+					deploymentObj, _ := utilhelper.ToUnstructured(deployment2)
+					return deploymentObj
+				}(),
+				commandOverriders: []policyv1alpha1.LabelAnnotationOverrider{
+					{
+						Operator: policyv1alpha1.OverriderOpAdd,
+						Value: map[string]string{
+							"test2": "test2",
+						},
+					},
+				},
+			},
+			want: map[string]string{
+				"test2": "test2",
+			},
+			wantErr: false,
+		},
+		{
+			name: "test annotations replace",
 			args: args{
 				rawObj: func() *unstructured.Unstructured {
 					deploymentObj, _ := utilhelper.ToUnstructured(deployment)
@@ -152,7 +198,7 @@ func Test_applyAnnotationsOverriders(t *testing.T) {
 			wantErr: false,
 		},
 		{
-			name: "test labels add",
+			name: "test annotations add",
 			args: args{
 				rawObj: func() *unstructured.Unstructured {
 					deploymentObj, _ := utilhelper.ToUnstructured(deployment)
@@ -175,7 +221,7 @@ func Test_applyAnnotationsOverriders(t *testing.T) {
 			wantErr: false,
 		},
 		{
-			name: "test labels remove",
+			name: "test annotations remove",
 			args: args{
 				rawObj: func() *unstructured.Unstructured {
 					deploymentObj, _ := utilhelper.ToUnstructured(deployment)
@@ -197,7 +243,7 @@ func Test_applyAnnotationsOverriders(t *testing.T) {
 			wantErr: false,
 		},
 		{
-			name: "test labels remain the same",
+			name: "test annotations remain the same",
 			args: args{
 				rawObj: func() *unstructured.Unstructured {
 					deploymentObj, _ := utilhelper.ToUnstructured(deployment)
@@ -217,7 +263,7 @@ func Test_applyAnnotationsOverriders(t *testing.T) {
 			wantErr: false,
 		},
 		{
-			name: "test labels remain the same",
+			name: "test annotations remain the same",
 			args: args{
 				rawObj: func() *unstructured.Unstructured {
 					deploymentObj, _ := utilhelper.ToUnstructured(deployment)
@@ -239,7 +285,7 @@ func Test_applyAnnotationsOverriders(t *testing.T) {
 			wantErr: false,
 		},
 		{
-			name: "test labels remain the same",
+			name: "test annotations remain the same",
 			args: args{
 				rawObj: func() *unstructured.Unstructured {
 					deploymentObj, _ := utilhelper.ToUnstructured(deployment)