Add tests

pkg/controllers/cassandra/seedlabeller/control.go

@@ -54,11 +54,11 @@ func (c *defaultSeedLabeller) labelSeedNodes(
 			return nil
 		}
 
-		// default to not a seed
-		desiredLabel := "false"
-
 		// label first n as seeds
-		if i < np.Seeds {
+		isSeed := i < np.Seeds
+
+		desiredLabel := ""
+		if isSeed {
 			desiredLabel = seedprovider.SeedLabelValue
 		}
 
@@ -70,7 +70,13 @@ func (c *defaultSeedLabeller) labelSeedNodes(
 		if labels == nil {
 			labels = map[string]string{}
 		}
-		labels[seedprovider.SeedLabelKey] = desiredLabel
+
+		if isSeed {
+			labels[seedprovider.SeedLabelKey] = desiredLabel
+		} else {
+			delete(labels, seedprovider.SeedLabelKey)
+		}
+
 		podCopy := pod.DeepCopy()
 		podCopy.SetLabels(labels)
 		_, err = c.kubeClient.CoreV1().Pods(podCopy.Namespace).Update(podCopy)
@@ -87,8 +93,10 @@ func (c *defaultSeedLabeller) Sync(cluster *v1alpha1.CassandraCluster) error {
 
 		set, err := c.statefulSetLister.StatefulSets(cluster.Namespace).Get(setName)
 		if err != nil {
-			return err
+			glog.Warningf("Couldn't get stateful set: %v", err)
+			return nil
 		}
+
 		err = c.labelSeedNodes(cluster, &np, set)
 		if err != nil {
 			return err

pkg/controllers/cassandra/seedlabeller/control_test.go

@@ -16,15 +16,15 @@ import (
 	casstesting "github.com/jetstack/navigator/pkg/controllers/cassandra/testing"
 )
 
-func CheckSeedLabel(podName, podNamespace string, t *testing.T, state *controllers.State) {
+func CheckSeedLabel(podName, seedLabelValue string, podNamespace string, t *testing.T, state *controllers.State) {
 	p, err := state.Clientset.
 		CoreV1().
 		Pods(podNamespace).
 		Get(podName, metav1.GetOptions{})
 	if err != nil {
 		t.Fatal(err)
 	}
-	if p.Labels[seedprovider.SeedLabelKey] != seedprovider.SeedLabelValue {
+	if p.Labels[seedprovider.SeedLabelKey] != seedLabelValue {
 		t.Errorf("unexpected seed label: %s", p.Labels)
 	}
 }
@@ -44,6 +44,20 @@ func TestSeedLabellerSync(t *testing.T) {
 	pod0ValueIncorrect := pod0LabelMissing.DeepCopy()
 	pod0ValueIncorrect.Labels[seedprovider.SeedLabelKey] = "blah"
 
+	pod1 := &v1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "cass-cassandra-1-RingNodes-1",
+			Namespace: cluster.Namespace,
+		},
+	}
+
+	pod2 := &v1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "cass-cassandra-1-RingNodes-2",
+			Namespace: cluster.Namespace,
+		},
+	}
+
 	type testT struct {
 		kubeObjects []runtime.Object
 		navObjects  []runtime.Object
@@ -63,23 +77,39 @@ func TestSeedLabellerSync(t *testing.T) {
 			navObjects:  []runtime.Object{cluster},
 			cluster:     cluster,
 			assertions: func(t *testing.T, state *controllers.State) {
-				CheckSeedLabel(pod0.Name, pod0.Namespace, t, state)
+				CheckSeedLabel(pod0.Name, seedprovider.SeedLabelValue, pod0.Namespace, t, state)
 			},
 		},
 		"add label if key missing": {
 			kubeObjects: []runtime.Object{ss0, pod0LabelMissing},
 			navObjects:  []runtime.Object{cluster},
 			cluster:     cluster,
 			assertions: func(t *testing.T, state *controllers.State) {
-				CheckSeedLabel(pod0.Name, pod0.Namespace, t, state)
+				CheckSeedLabel(pod0.Name, seedprovider.SeedLabelValue, pod0.Namespace, t, state)
 			},
 		},
 		"fix label if value incorrect": {
 			kubeObjects: []runtime.Object{ss0, pod0ValueIncorrect},
 			navObjects:  []runtime.Object{cluster},
 			cluster:     cluster,
 			assertions: func(t *testing.T, state *controllers.State) {
-				CheckSeedLabel(pod0.Name, pod0.Namespace, t, state)
+				CheckSeedLabel(pod0.Name, seedprovider.SeedLabelValue, pod0.Namespace, t, state)
+			},
+		},
+		"add multiple seeds": {
+			kubeObjects: []runtime.Object{ss0, pod0, pod1, pod2},
+			navObjects:  []runtime.Object{cluster},
+			cluster:     cluster,
+			assertions: func(t *testing.T, state *controllers.State) {
+				CheckSeedLabel(pod1.Name, seedprovider.SeedLabelValue, pod1.Namespace, t, state)
+			},
+		},
+		"don't add too many seeds": {
+			kubeObjects: []runtime.Object{ss0, pod0, pod1, pod2},
+			navObjects:  []runtime.Object{cluster},
+			cluster:     cluster,
+			assertions: func(t *testing.T, state *controllers.State) {
+				CheckSeedLabel(pod2.Name, "", pod2.Namespace, t, state)
 			},
 		},
 	}

pkg/controllers/cassandra/testing/testing.go

@@ -39,6 +39,7 @@ func ClusterForTest() *v1alpha1.CassandraCluster {
 				v1alpha1.CassandraClusterNodePool{
 					Name:     "RingNodes",
 					Replicas: 3,
+					Seeds:    2,
 				},
 			},
 		},