test: add tests for npm/util (#1303)

Author: snguyen64

npm/util/util_test.go

@@ -323,3 +323,199 @@ func TestCompareSlices(t *testing.T) {
 		t.Errorf("TestCompareSlices failed @ slice comparison 4")
 	}
 }
+
+func TestExists(t *testing.T) {
+	type args struct {
+		filePath string
+	}
+	dir := t.TempDir()
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "Test for filepath exists",
+			args: args{
+				dir,
+			},
+			want: true,
+		},
+		{
+			name: "Test for directory/file not exist",
+			args: args{
+				"unknown_directory",
+			},
+			want: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := Exists(tt.args.filePath); got != tt.want {
+				t.Errorf("Exists() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestGetClusterID(t *testing.T) {
+	type args struct {
+		nodeName string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "Test to get cluster id for invalid azure node name",
+			args: args{
+				"nodename-test111",
+			},
+			want: "",
+		},
+		{
+			name: "Test to get cluster id for valid azure node name",
+			args: args{
+				"aks-agentpool-vmss000000",
+			},
+			want: "vmss000000",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := GetClusterID(tt.args.nodeName); got != tt.want {
+				t.Errorf("GetClusterID() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestGetIPSetListFromLabels(t *testing.T) {
+	labels := make(map[string]string)
+	labels["test-key"] = "test-val"
+	expected := []string{
+		"test-key",
+		"test-key:test-val",
+	}
+	got := GetIPSetListFromLabels(labels)
+	if len(got) != 2 || expected[0] != got[0] || expected[1] != got[1] {
+		t.Errorf("GetIPSetListFromLabels(labels map[string]string) = %v, want %v", got, expected)
+	}
+}
+
+func TestClearAndAppendMap(t *testing.T) {
+	base := map[string]string{
+		"base-key": "base-val",
+	}
+	newmap := map[string]string{
+		"one": "uno",
+		"two": "dos",
+	}
+	if got := ClearAndAppendMap(base, newmap); !reflect.DeepEqual(got, newmap) {
+		t.Errorf("ClearAndAppendMap() = %v, want %v", got, newmap)
+	}
+}
+
+func TestAppendMap(t *testing.T) {
+	base := map[string]string{
+		"one": "uno",
+	}
+	mapAppend := map[string]string{
+		"two": "two",
+	}
+	result := map[string]string{
+		"one": "uno",
+		"two": "two",
+	}
+	if got := AppendMap(base, mapAppend); !reflect.DeepEqual(got, result) {
+		t.Errorf("AppendMap() = %v, want %v", got, result)
+	}
+}
+
+func TestGetOperatorAndLabel(t *testing.T) {
+	type args struct {
+		label string
+	}
+	tests := []struct {
+		name  string
+		args  args
+		want0 string
+		want1 string
+	}{
+		{
+			name: "Test for empty input",
+			args: args{
+				"",
+			},
+			want0: "",
+			want1: "",
+		},
+		{
+			name: "Test for iptables not flag",
+			args: args{
+				"!test",
+			},
+			want0: "!",
+			want1: "test",
+		},
+		{
+			name: "Test for normal label",
+			args: args{
+				"test",
+			},
+			want0: "",
+			want1: "test",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, got1 := GetOperatorAndLabel(tt.args.label)
+			if got != tt.want0 {
+				t.Errorf("GetOperatorAndLabel() got = %v, want %v", got, tt.want0)
+			}
+			if got1 != tt.want1 {
+				t.Errorf("GetOperatorAndLabel() got1 = %v, want %v", got1, tt.want1)
+			}
+		})
+	}
+}
+
+func TestGetLabelsWithoutOperators(t *testing.T) {
+	want := []string{
+		"res",
+		"res2",
+	}
+	labels := []string{
+		"!res",
+		"res2",
+	}
+	if got := GetLabelsWithoutOperators(labels); !reflect.DeepEqual(want, got) {
+		t.Errorf("GetLabelsWithoutOperators() got = %v, want %v", got, want)
+	}
+}
+
+func TestGetSetsFromLabels(t *testing.T) {
+	labels := map[string]string{
+		"key": "val",
+	}
+	want := []string{
+		"key",
+		"key:val",
+	}
+	if got := GetSetsFromLabels(labels); !reflect.DeepEqual(want, got) {
+		t.Errorf("GetSetsFromLabels() got = %v, want %v", got, want)
+	}
+}
+
+func TestSliceToString(t *testing.T) {
+	want := "test,test2"
+	list := []string{
+		"test",
+		"test2",
+	}
+	if got := SliceToString(list); want != got {
+		t.Errorf("SliceToString() got = %v, want %v, using delimiter %v", got, want, SetPolicyDelimiter)
+	}
+}