Follow up in tests

Signed-off-by: György Krajcsovits <[email protected]>
grafana · Sep 5, 2022 · d236af4 · d236af4
1 parent 2e1676f
commit d236af4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/controller/controller_test.go b/pkg/controller/controller_test.go
@@ -365,6 +365,11 @@ func TestRolloutController_ReconcileShouldDeleteMetricsForDecommissionedRolloutG
 }
 
 func mockStatefulSet(name string, overrides ...func(sts *v1.StatefulSet)) *v1.StatefulSet {
+	labelSelector := metav1.LabelSelector{
+		MatchLabels: map[string]string{
+				"name": name,
+		},
+	};
 	sts := &v1.StatefulSet{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      name,
@@ -387,6 +392,7 @@ func mockStatefulSet(name string, overrides ...func(sts *v1.StatefulSet)) *v1.St
 					},
 				},
 			},
+			Selector: &labelSelector,
 		},
 		Status: v1.StatefulSetStatus{
 			Replicas:        3,