Fix UT

zalando · Mar 8, 2025 · 6f469d1 · 6f469d1
1 parent d7fb626
commit 6f469d1
Showing 1 changed file with 11 additions and 13 deletions.
diff --git a/pkg/cluster/k8sres_test.go b/pkg/cluster/k8sres_test.go
@@ -3988,6 +3988,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{
@@ -4008,9 +4011,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,
 				},
 			},
 		},
@@ -4028,14 +4029,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,
+	},
+	)
 }