Dynamic Configuration Refactoring

old/provider/label/segment_test.go

@@ -0,0 +1,95 @@
+package label
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestExtractTraefikLabels(t *testing.T) {
+	testCases := []struct {
+		desc         string
+		prefix       string
+		originLabels map[string]string
+		expected     SegmentProperties
+	}{
+		{
+			desc:         "nil labels map",
+			prefix:       "traefik",
+			originLabels: nil,
+			expected:     SegmentProperties{"": {}},
+		},
+		{
+			desc:   "container labels",
+			prefix: "traefik",
+			originLabels: map[string]string{
+				"frontend.priority": "foo", // missing prefix: skip
+				"traefik.port":      "bar",
+			},
+			expected: SegmentProperties{
+				"": {
+					"traefik.port": "bar",
+				},
+			},
+		},
+		{
+			desc:   "segment labels: only segment no default",
+			prefix: "traefik",
+			originLabels: map[string]string{
+				"traefik.goo.frontend.priority": "A",
+				"traefik.goo.port":              "D",
+				"traefik.port":                  "C",
+			},
+			expected: SegmentProperties{
+				"goo": {
+					"traefik.frontend.priority": "A",
+					"traefik.port":              "D",
+				},
+			},
+		},
+		{
+			desc:   "segment labels: use default",
+			prefix: "traefik",
+			originLabels: map[string]string{
+				"traefik.guu.frontend.priority": "B",
+				"traefik.port":                  "C",
+			},
+			expected: SegmentProperties{
+				"guu": {
+					"traefik.frontend.priority": "B",
+					"traefik.port":              "C",
+				},
+			},
+		},
+		{
+			desc:   "segment labels: several segments",
+			prefix: "traefik",
+			originLabels: map[string]string{
+				"traefik.goo.frontend.priority": "A",
+				"traefik.goo.port":              "D",
+				"traefik.guu.frontend.priority": "B",
+				"traefik.port":                  "C",
+			},
+			expected: SegmentProperties{
+				"goo": {
+					"traefik.frontend.priority": "A",
+					"traefik.port":              "D",
+				},
+				"guu": {
+					"traefik.frontend.priority": "B",
+					"traefik.port":              "C",
+				},
+			},
+		},
+	}
+
+	for _, test := range testCases {
+		test := test
+		t.Run(test.desc, func(t *testing.T) {
+			t.Parallel()
+
+			actual := ExtractTraefikLabels(test.originLabels)
+			assert.Equal(t, test.expected, actual)
+		})
+	}
+}