add podselector labels

powerhome · Feb 8, 2024 · 0c2091e · 0c2091e
1 parent 01ae991
commit 0c2091e
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/operator/redisfailover/service/generator.go b/operator/redisfailover/service/generator.go
@@ -486,6 +486,8 @@ func generateSentinelNetworkPolicy(rf *redisfailoverv1.RedisFailover, labels map
 		Port: &sentinelTargetPort,
 	})
 
+	redisfailoverLabels := map[string]string{"redisfailovers.databases.spotahome.com/name": rf.Name}
+
 	return &np.NetworkPolicy{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:            name,
@@ -496,7 +498,7 @@ func generateSentinelNetworkPolicy(rf *redisfailoverv1.RedisFailover, labels map
 		Spec: np.NetworkPolicySpec{
 			PodSelector: metav1.LabelSelector{
 				MatchLabels: util.MergeLabels(
-					map[string]string{"redisfailovers.databases.spotahome.com/name": rf.Name},
+					redisfailoverLabels,
 					generateComponentLabel("sentinel"),
 				),
 			},
@@ -516,6 +518,11 @@ func generateSentinelNetworkPolicy(rf *redisfailoverv1.RedisFailover, labels map
 								},
 							},
 						},
+						np.NetworkPolicyPeer{
+							PodSelector: &metav1.LabelSelector{
+								MatchLabels: redisfailoverLabels,
+							},
+						},
 					},
 				},
 			},

diff --git a/operator/redisfailover/service/generator_test.go b/operator/redisfailover/service/generator_test.go
@@ -1488,6 +1488,13 @@ func TestSentinelNetworkPolicy(t *testing.T) {
 										},
 									},
 								},
+								networkingv1.NetworkPolicyPeer{
+									PodSelector: &metav1.LabelSelector{
+										MatchLabels: map[string]string{
+											"redisfailovers.databases.spotahome.com/name": name,
+										},
+									},
+								},
 							},
 						},
 					},
@@ -1557,6 +1564,13 @@ func TestSentinelNetworkPolicy(t *testing.T) {
 										},
 									},
 								},
+								networkingv1.NetworkPolicyPeer{
+									PodSelector: &metav1.LabelSelector{
+										MatchLabels: map[string]string{
+											"redisfailovers.databases.spotahome.com/name": name,
+										},
+									},
+								},
 							},
 						},
 					},
@@ -1636,6 +1650,13 @@ func TestSentinelNetworkPolicy(t *testing.T) {
 										},
 									},
 								},
+								networkingv1.NetworkPolicyPeer{
+									PodSelector: &metav1.LabelSelector{
+										MatchLabels: map[string]string{
+											"redisfailovers.databases.spotahome.com/name": name,
+										},
+									},
+								},
 							},
 						},
 					},