add: test case for ordering base on value

k8snetworkplumbingwg · Nov 10, 2024 · 547a5fe · 547a5fe
1 parent 5030cd2
commit 547a5fe
Showing 1 changed file with 55 additions and 0 deletions.
diff --git a/controllers/helper_test.go b/controllers/helper_test.go
@@ -148,6 +148,61 @@ func TestNodeSelectorMerge(t *testing.T) {
 			},
 			expected: []corev1.NodeSelectorTerm{},
 		},
+		{
+			tname: "testordersamekey",
+			policies: []sriovnetworkv1.SriovNetworkNodePolicy{
+				{
+					Spec: sriovnetworkv1.SriovNetworkNodePolicySpec{
+						NodeSelector: map[string]string{
+							"env": "production",
+						},
+					},
+				},
+				{
+					Spec: sriovnetworkv1.SriovNetworkNodePolicySpec{
+						NodeSelector: map[string]string{
+							"env": "staging",
+						},
+					},
+				},
+				{
+					Spec: sriovnetworkv1.SriovNetworkNodePolicySpec{
+						NodeSelector: map[string]string{
+							"env": "dev",
+						},
+					},
+				},
+			},
+			expected: []corev1.NodeSelectorTerm{
+				{
+					MatchExpressions: []corev1.NodeSelectorRequirement{
+						{
+							Operator: corev1.NodeSelectorOpIn,
+							Key:      "env",
+							Values:   []string{"dev"},
+						},
+					},
+				},
+				{
+					MatchExpressions: []corev1.NodeSelectorRequirement{
+						{
+							Operator: corev1.NodeSelectorOpIn,
+							Key:      "env",
+							Values:   []string{"production"},
+						},
+					},
+				},
+				{
+					MatchExpressions: []corev1.NodeSelectorRequirement{
+						{
+							Operator: corev1.NodeSelectorOpIn,
+							Key:      "env",
+							Values:   []string{"staging"},
+						},
+					},
+				},
+			},
+		},
 	}
 
 	for _, tc := range table {