Add possibility to use multiple names for one service (#31)

Co-authored-by: Geoff Bourne <itzgeoff@gmail.com>
2021-12-04 07:25:28 -08:00
parent 6bf14043bb
commit 1b1f8e5f22
5 changed files with 257 additions and 46 deletions
@@ -0,0 +1,172 @@
+package server
+
+import (
+	"encoding/json"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	v1 "k8s.io/api/core/v1"
+	"testing"
+)
+
+func TestK8sWatcherImpl_handleAddThenUpdate(t *testing.T) {
+	type scenario struct {
+		given  string
+		expect string
+	}
+	type svcAndScenarios struct {
+		svc       string
+		scenarios []scenario
+	}
+	tests := []struct {
+		name    string
+		initial svcAndScenarios
+		update  svcAndScenarios
+	}{
+		{
+			name: "a to b",
+			initial: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "a.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: "1.1.1.1:25565"},
+					{given: "b.com", expect: ""},
+				},
+			},
+			update: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "b.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: ""},
+					{given: "b.com", expect: "1.1.1.1:25565"},
+				},
+			},
+		},
+		{
+			name: "a to a,b",
+			initial: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "a.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: "1.1.1.1:25565"},
+					{given: "b.com", expect: ""},
+				},
+			},
+			update: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "a.com,b.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: "1.1.1.1:25565"},
+					{given: "b.com", expect: "1.1.1.1:25565"},
+				},
+			},
+		},
+		{
+			name: "a,b to b",
+			initial: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "a.com,b.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: "1.1.1.1:25565"},
+					{given: "b.com", expect: "1.1.1.1:25565"},
+				},
+			},
+			update: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "b.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: ""},
+					{given: "b.com", expect: "1.1.1.1:25565"},
+				},
+			},
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			// reset the routes
+			Routes.RegisterAll(map[string]string{})
+
+			watcher := &k8sWatcherImpl{}
+			initialSvc := v1.Service{}
+			err := json.Unmarshal([]byte(test.initial.svc), &initialSvc)
+			require.NoError(t, err)
+
+			watcher.handleAdd(&initialSvc)
+			for _, s := range test.initial.scenarios {
+				backend, _ := Routes.FindBackendForServerAddress(s.given)
+				assert.Equal(t, s.expect, backend, "initial: given=%s", s.given)
+			}
+
+			updatedSvc := v1.Service{}
+			err = json.Unmarshal([]byte(test.update.svc), &updatedSvc)
+			require.NoError(t, err)
+
+			watcher.handleUpdate(&initialSvc, &updatedSvc)
+			for _, s := range test.update.scenarios {
+				backend, _ := Routes.FindBackendForServerAddress(s.given)
+				assert.Equal(t, s.expect, backend, "update: given=%s", s.given)
+			}
+		})
+	}
+}
+
+func TestK8sWatcherImpl_handleAddThenDelete(t *testing.T) {
+	type scenario struct {
+		given  string
+		expect string
+	}
+	type svcAndScenarios struct {
+		svc       string
+		scenarios []scenario
+	}
+	tests := []struct {
+		name    string
+		initial svcAndScenarios
+		delete  []scenario
+	}{
+		{
+			name: "single",
+			initial: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "a.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: "1.1.1.1:25565"},
+					{given: "b.com", expect: ""},
+				},
+			},
+			delete: []scenario{
+				{given: "a.com", expect: ""},
+				{given: "b.com", expect: ""},
+			},
+		},
+		{
+			name: "multi",
+			initial: svcAndScenarios{
+				svc: ` {"metadata": {"annotations": {"mc-router.itzg.me/externalServerName": "a.com,b.com"}}, "spec":{"clusterIP": "1.1.1.1"}}`,
+				scenarios: []scenario{
+					{given: "a.com", expect: "1.1.1.1:25565"},
+					{given: "b.com", expect: "1.1.1.1:25565"},
+				},
+			},
+			delete: []scenario{
+				{given: "a.com", expect: ""},
+				{given: "b.com", expect: ""},
+			},
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			// reset the routes
+			Routes.RegisterAll(map[string]string{})
+
+			watcher := &k8sWatcherImpl{}
+			initialSvc := v1.Service{}
+			err := json.Unmarshal([]byte(test.initial.svc), &initialSvc)
+			require.NoError(t, err)
+
+			watcher.handleAdd(&initialSvc)
+			for _, s := range test.initial.scenarios {
+				backend, _ := Routes.FindBackendForServerAddress(s.given)
+				assert.Equal(t, s.expect, backend, "initial: given=%s", s.given)
+			}
+
+			watcher.handleDelete(&initialSvc)
+			for _, s := range test.delete {
+				backend, _ := Routes.FindBackendForServerAddress(s.given)
+				assert.Equal(t, s.expect, backend, "update: given=%s", s.given)
+			}
+		})
+	}
+}