From 760bcac716cc40f99e3ebd5169c2f475d79a1a90 Mon Sep 17 00:00:00 2001
From: Bazaah <github@luxolus.com>
Date: Fri, 9 Dec 2022 12:24:23 +0000
Subject: [PATCH] config: tests for WatchConfig

---
 config/config_test.go | 127 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 127 insertions(+)
 create mode 100644 config/config_test.go

diff --git a/config/config_test.go b/config/config_test.go
new file mode 100644
index 0000000..a5817a1
--- /dev/null
+++ b/config/config_test.go
@@ -0,0 +1,127 @@
+/*
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+package config_test
+
+import (
+	"fmt"
+	"testing"
+	"text/template"
+
+	"github.com/stretchr/testify/assert"
+
+	"git.st8l.com/luxolus/kdnotify/config"
+	"git.st8l.com/luxolus/kdnotify/schema/notify/state"
+	"git.st8l.com/luxolus/kdnotify/schema/notify/ty"
+)
+
+func Test_WatchConfig(t *testing.T) {
+	t.Parallel()
+
+	cases := []struct {
+		Name        string
+		Yaml        string
+		Expected    config.WatchConfig
+		ShouldError bool
+		CompareFunc func(t *testing.T, expected, actual config.WatchConfig) bool
+	}{
+		{
+			Name: "Empty",
+			Yaml: "{}",
+		},
+		{
+			Name: "EmptyWithKey",
+			Yaml: "{ watch: {} }",
+		},
+		{
+			Name: "Minimal",
+			Yaml: `
+watch:
+    rules:
+      - exec: /bin/echo hello test
+`,
+			Expected: config.WatchConfig{
+				Rules: []config.WatchRule{
+					{
+						Exec: s2t("/bin/echo hello test"),
+					},
+				},
+			},
+			CompareFunc: CompareWatchConfig,
+		},
+		{
+			Name: "StringArgs",
+			Yaml: `
+watch:
+    rules:
+      - exec: /bin/echo hello test
+        type: INSTANCE
+        state: STOP
+        instance: foo_instance
+`,
+			Expected: config.WatchConfig{
+				Rules: []config.WatchRule{
+					{
+						Exec:     s2t("/bin/echo hello test"),
+						Type:     ty.INSTANCE,
+						State:    state.STOP,
+						Instance: []string{"foo_instance"},
+					},
+				},
+			},
+			CompareFunc: CompareWatchConfig,
+		},
+		{
+			Name: "ArrayArgs",
+			Yaml: `
+watch:
+    rules:
+      - exec: /bin/echo hello test
+        type: [INSTANCE, GROUP]
+        state: [MASTER]
+        instance: [foo_instance, bar_instance]
+`,
+			Expected: config.WatchConfig{
+				Rules: []config.WatchRule{
+					{
+						Exec:     s2t("/bin/echo hello test"),
+						Type:     ty.INSTANCE | ty.GROUP,
+						State:    state.MASTER,
+						Instance: []string{"foo_instance", "bar_instance"},
+					},
+				},
+			},
+			CompareFunc: CompareWatchConfig,
+		},
+	}
+
+	for i, test := range cases {
+		test := test // Avoid closure capture by ref
+		tname := fmt.Sprintf("%s@%d:%d", test.Name, i+1, len(cases))
+
+		t.Run(tname, func(t *testing.T) {
+			t.Parallel()
+
+			cfg, err := config.WatchConfigFromYAML([]byte(test.Yaml))
+			if !test.ShouldError && assert.NoError(t, err, "failed to unmarshal yaml") {
+				if test.CompareFunc != nil {
+					test.CompareFunc(t, test.Expected, cfg)
+				}
+			} else {
+				assert.Error(t, err)
+			}
+		})
+	}
+
+}
+
+func CompareWatchConfig(t *testing.T, expected, actual config.WatchConfig) bool {
+	return assert.Equal(t, expected, actual)
+}
+
+func s2t(str string) *template.Template {
+	return template.Must(template.New("WatchRule").Parse(str))
+}