From 389dc120aac1bf17d2fbfb996e258376625fe1aa Mon Sep 17 00:00:00 2001 From: Charlie McBride <33269602+charliedmcb@users.noreply.github.com> Date: Fri, 17 Nov 2023 16:34:18 -0800 Subject: [PATCH] fix drift E2E to use ReplaceRequirements (#40) Co-authored-by: Charlie McBride --- test/suites/drift/suite_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/suites/drift/suite_test.go b/test/suites/drift/suite_test.go index de8f6b127..7e78c647a 100644 --- 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{