add test for util/Match function (vmware#289)

Signed-off-by: huskykurt <[email protected]>
huskykurt · Dec 14, 2022 · 124278e · 124278e
1 parent f0131e0
commit 124278e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/config-reloader/util/util.go b/config-reloader/util/util.go
@@ -24,7 +24,7 @@ import (
 
 const (
 	maskFile       = 0664
-  maskDirectory = 0775
+	maskDirectory  = 0775
 	MacroLabels    = "$labels"
 	ContainerLabel = "_container"
 )

diff --git a/config-reloader/util/util_test.go b/config-reloader/util/util_test.go
@@ -85,6 +85,29 @@ func TestLabelsParseNotOk(t *testing.T) {
 	}
 }
 
+func TestMatch(t *testing.T) {
+	containerLabels := map[string]string{"key": "value"}
+	containerName := "container-name"
+
+	var labels map[string]string = nil
+	assert.True(t, Match(labels, containerLabels, containerName))
+
+	labels = map[string]string{"_container": containerName}
+	assert.True(t, Match(labels, containerLabels, containerName))
+
+	labels = map[string]string{"a": "a"}
+	assert.False(t, Match(labels, containerLabels, containerName))
+
+	labels = map[string]string{"key": "value"}
+	assert.True(t, Match(labels, containerLabels, containerName))
+
+	labels = map[string]string{"key": "value", "_container": "container-name"}
+	assert.True(t, Match(labels, containerLabels, containerName))
+
+	labels = map[string]string{"a": "a", "key": "value", "_container": "container-name"}
+	assert.False(t, Match(labels, containerLabels, containerName))
+}
+
 func TestEnsureDirExits(t *testing.T) {
 
 	type testDirConfig struct {