From d36ade1f14fb39b9db546a2277cccad680ab9410 Mon Sep 17 00:00:00 2001
From: "Alexander A. Klimov" <alexander.klimov@icinga.com>
Date: Mon, 25 Mar 2024 14:49:06 +0100
Subject: [PATCH] Test Flatten()

---
 pkg/flatten/flatten_test.go | 44 +++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 pkg/flatten/flatten_test.go

diff --git a/pkg/flatten/flatten_test.go b/pkg/flatten/flatten_test.go
new file mode 100644
index 000000000..0655550f1
--- /dev/null
+++ b/pkg/flatten/flatten_test.go
@@ -0,0 +1,44 @@
+package flatten
+
+import (
+	"github.com/icinga/icingadb/pkg/types"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestFlatten(t *testing.T) {
+	for _, st := range []struct {
+		name   string
+		prefix string
+		value  any
+		output map[string]types.String
+	}{
+		{"nil", "a", nil, map[string]types.String{"a": types.MakeString("null")}},
+		{"bool", "b", true, map[string]types.String{"b": types.MakeString("true")}},
+		{"int", "c", 42, map[string]types.String{"c": types.MakeString("42")}},
+		{"float", "d", 77.7, map[string]types.String{"d": types.MakeString("77.7")}},
+		{"string", "f", "\x00", map[string]types.String{"f": types.MakeString("\x00")}},
+		{"nil_slice", "g", []any(nil), map[string]types.String{"g": {}}},
+		{"empty_slice", "h", []any{}, map[string]types.String{"h": {}}},
+		{"slice", "i", []any{nil}, map[string]types.String{"i[0]": types.MakeString("null")}},
+		{"nil_map", "j", map[string]any(nil), map[string]types.String{"j": {}}},
+		{"empty_map", "k", map[string]any{}, map[string]types.String{"k": {}}},
+		{"map", "l", map[string]any{" ": nil}, map[string]types.String{"l. ": types.MakeString("null")}},
+		{"map_with_slice", "m", map[string]any{"\t": []any{"ä", "ö", "ü"}, "ß": "s"}, map[string]types.String{
+			"m.\t[0]": types.MakeString("ä"),
+			"m.\t[1]": types.MakeString("ö"),
+			"m.\t[2]": types.MakeString("ü"),
+			"m.ß":     types.MakeString("s"),
+		}},
+		{"slice_with_map", "n", []any{map[string]any{"ä": "a", "ö": "o", "ü": "u"}, "ß"}, map[string]types.String{
+			"n[0].ä": types.MakeString("a"),
+			"n[0].ö": types.MakeString("o"),
+			"n[0].ü": types.MakeString("u"),
+			"n[1]":   types.MakeString("ß"),
+		}},
+	} {
+		t.Run(st.name, func(t *testing.T) {
+			assert.Equal(t, st.output, Flatten(st.value, st.prefix))
+		})
+	}
+}