1 files changed, 44 insertions, 6 deletions
diff --git a/plugin/kubernetes/watch_test.go b/plugin/kubernetes/watch_test.go
index 46b2e5dc4..08eda894b 100644
--- a/plugin/kubernetes/watch_test.go
+++ b/plugin/kubernetes/watch_test.go
@@ -1,15 +1,53 @@
 package kubernetes
 
 import (
+	"strconv"
+	"strings"
 	"testing"
 
-	"github.com/coredns/coredns/plugin/pkg/watch"
+	"github.com/coredns/coredns/plugin/kubernetes/object"
 )
 
-func TestIsWatchable(t *testing.T) {
-	k := &Kubernetes{}
-	var i interface{} = k
-	if _, ok := i.(watch.Watchable); !ok {
-		t.Error("Kubernetes should implement watch.Watchable and does not")
+func endpointSubsets(addrs ...string) (eps []object.EndpointSubset) {
+	for _, ap := range addrs {
+		apa := strings.Split(ap, ":")
+		address := apa[0]
+		port, _ := strconv.Atoi(apa[1])
+		eps = append(eps, object.EndpointSubset{Addresses: []object.EndpointAddress{{IP: address}}, Ports: []object.EndpointPort{{Port: int32(port)}}})
+	}
+	return eps
+}
+
+func TestEndpointsSubsetDiffs(t *testing.T) {
+	var tests = []struct {
+		a, b, expected object.Endpoints
+	}{
+		{ // From a->b: Nothing changes
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80", "10.0.0.2:8080")},
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80", "10.0.0.2:8080")},
+			object.Endpoints{},
+		},
+		{ // From a->b: Everything goes away
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80", "10.0.0.2:8080")},
+			object.Endpoints{},
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80", "10.0.0.2:8080")},
+		},
+		{ // From a->b: Everything is new
+			object.Endpoints{},
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80", "10.0.0.2:8080")},
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80", "10.0.0.2:8080")},
+		},
+		{ // From a->b: One goes away, one is new
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.2:8080")},
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.1:80")},
+			object.Endpoints{Subsets: endpointSubsets("10.0.0.2:8080", "10.0.0.1:80")},
+		},
+	}
+
+	for i, te := range tests {
+		got := endpointsSubsetDiffs(&te.a, &te.b)
+		if !endpointsEquivalent(got, &te.expected) {
+			t.Errorf("Expected '%v' for test %v, got '%v'.", te.expected, i, got)
+		}
 	}
 }