Merge pull request #8955 from hs0210/work

Add unit test for func JoinSuffixes
kubernetes · Apr 22, 2020 · bb48a8c · bb48a8c
2 parents a57a404 + d3d4afd
commit bb48a8c
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/pkg/model/firewall_test.go b/pkg/model/firewall_test.go
@@ -60,3 +60,34 @@ func Test_SharedGroups(t *testing.T) {
 		}
 	}
 }
+
+func TestJoinSuffixes(t *testing.T) {
+	grid := []struct {
+		src      SecurityGroupInfo
+		dest     SecurityGroupInfo
+		expected string
+	}{
+		{
+			src:      SecurityGroupInfo{Suffix: ""},
+			dest:     SecurityGroupInfo{Suffix: ""},
+			expected: "",
+		},
+		{
+			src:      SecurityGroupInfo{Suffix: "srcSuffix"},
+			dest:     SecurityGroupInfo{Suffix: ""},
+			expected: "srcSuffix-default",
+		},
+		{
+			src:      SecurityGroupInfo{Suffix: ""},
+			dest:     SecurityGroupInfo{Suffix: "destSuffix"},
+			expected: "-defaultdestSuffix",
+		},
+	}
+
+	for _, g := range grid {
+		actual := JoinSuffixes(g.src, g.dest)
+		if actual != g.expected {
+			t.Errorf("unexpected result.  expected %q, got %q", g.expected, actual)
+		}
+	}
+}