Fix UT

zalando · Nov 18, 2024 · 27f69f5 · 27f69f5
1 parent c88bbe6
commit 27f69f5
Showing 1 changed file with 11 additions and 13 deletions.
diff --git a/pkg/cluster/k8sres_test.go b/pkg/cluster/k8sres_test.go
@@ -3890,6 +3890,9 @@ func TestGenerateCapabilities(t *testing.T) {
 func TestTopologySpreadConstraints(t *testing.T) {
 	clusterName := "acid-test-cluster"
 	namespace := "default"
+	labelSelector := &metav1.LabelSelector{
+		MatchLabels: cluster.labelsSet(true),
+	}
 
 	pg := acidv1.Postgresql{
 		ObjectMeta: metav1.ObjectMeta{
@@ -3910,9 +3913,7 @@ func TestTopologySpreadConstraints(t *testing.T) {
 					MaxSkew:           1,
 					TopologyKey:       "topology.kubernetes.io/zone",
 					WhenUnsatisfiable: v1.DoNotSchedule,
-					LabelSelector: &metav1.LabelSelector{
-						MatchLabels: cluster.labelsSet(true),
-					},
+					LabelSelector:     labelSelector,
 				},
 			},
 		},
@@ -3930,14 +3931,11 @@ func TestTopologySpreadConstraints(t *testing.T) {
 
 	s, err := cluster.generateStatefulSet(&pg.Spec)
 	assert.NoError(t, err)
-	assert.Contains(t, s.Spec.Template.Spec.TopologySpreadConstraints, []v1.TopologySpreadConstraint{
-		{
-			MaxSkew:           int32(1),
-			TopologyKey:       "topology.kubernetes.io/zone",
-			WhenUnsatisfiable: v1.DoNotSchedule,
-			LabelSelector: &metav1.LabelSelector{
-				MatchLabels: cluster.labelsSet(true),
-			},
-		},
-	})
+	assert.Contains(t, s.Spec.Template.Spec.TopologySpreadConstraints, v1.TopologySpreadConstraint{
+		MaxSkew:           int32(1),
+		TopologyKey:       "topology.kubernetes.io/zone",
+		WhenUnsatisfiable: v1.DoNotSchedule,
+		LabelSelector:     labelSelector,
+	},
+	)
 }