tests: add 'Random' label templates coverage

Signed-off-by: Francesco Giudici <francesco.giudici@suse.com>
rancher · Aug 13, 2024 · 048e880 · 048e880
1 parent 9494210
commit 048e880
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/pkg/templater/templater_test.go b/pkg/templater/templater_test.go
@@ -18,6 +18,7 @@ package templater
 
 import (
 	"fmt"
+	"strings"
 	"testing"
 
 	"gotest.tools/v3/assert"
@@ -130,3 +131,37 @@ func TestIsValueNotFoundError(t *testing.T) {
 	assert.Equal(t, IsValueNotFoundError(errValueNotFound), true)
 	assert.Equal(t, IsValueNotFoundError(errRandom), false)
 }
+
+func TestRandomTemplate(t *testing.T) {
+	testCase := []struct {
+		tmplVal []string
+		isRand  bool
+		isValid bool
+	}{
+		{[]string{"Custom"}, false, false},
+		{[]string{tmplRandomKey, "Custom"}, false, false},
+		{[]string{tmplRandomKey, tmplUUIDKey}, true, true},
+		{[]string{tmplRandomKey, tmplUUIDKey, "ABC"}, true, false},
+		{[]string{tmplRandomKey, tmplUUIDKey, "16"}, true, false},
+		{[]string{tmplRandomKey, tmplHexKey}, true, false},
+		{[]string{tmplRandomKey, tmplHexKey, "0"}, true, false},
+		{[]string{tmplRandomKey, tmplHexKey, "33"}, true, false},
+		{[]string{tmplRandomKey, tmplHexKey, "32"}, true, true},
+		{[]string{tmplRandomKey, tmplIntKey}, true, false},
+		{[]string{tmplRandomKey, tmplIntKey, "12123123"}, true, true},
+		{[]string{tmplRandomKey, tmplIntKey, "0"}, true, false},
+		{[]string{tmplRandomKey, tmplIntKey, "pippo"}, true, false},
+		{[]string{tmplRandomKey, tmplIntKey, "1000", "extraarg"}, true, false},
+	}
+
+	for _, testCase := range testCase {
+		assert.Equal(t, isRandomTemplate(testCase.tmplVal), testCase.isRand,
+			"template: %s, expected ret: %t", strings.Join(testCase.tmplVal, "/"), testCase.isRand)
+		val, err := randomTemplateToString(testCase.tmplVal)
+		if testCase.isValid {
+			assert.NilError(t, err)
+		} else {
+			assert.Assert(t, err != nil, "template '%s' got converted to '%s'", strings.Join(testCase.tmplVal, "/"), val)
+		}
+	}
+}