fix drift E2E to use ReplaceRequirements (#40)

Co-authored-by: Charlie McBride <[email protected]>
Azure · Nov 18, 2023 · 389dc12 · 389dc12
1 parent 334b902
commit 389dc12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/suites/drift/suite_test.go b/test/suites/drift/suite_test.go
@@ -63,11 +63,11 @@ var _ = Describe("Drift", func() {
 	BeforeEach(func() {
 		env.ExpectSettingsOverridden(v1.EnvVar{Name: "FEATURE_GATES", Value: "Drift=true"})
 
-		nodePool.Spec.Template.Spec.Requirements = []v1.NodeSelectorRequirement{{
+		test.ReplaceRequirements(nodePool, v1.NodeSelectorRequirement{
 			Key:      v1.LabelInstanceTypeStable,
 			Operator: v1.NodeSelectorOpIn,
 			Values:   []string{"Standard_DS2_v2"},
-		}}
+		})
 
 		// Add a do-not-disrupt pod so that we can check node metadata before we disrupt
 		pod = test.Pod(test.PodOptions{