ci: Flaky Start-up Taint Drift Test (#4386)

aws · Aug 7, 2023 · 0e0783f · 0e0783f
1 parent d035561
commit 0e0783f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test/suites/drift/suite_test.go b/test/suites/drift/suite_test.go
@@ -272,8 +272,13 @@ var _ = Describe("Drift", Label("AWS"), func() {
 				return nodes[i].CreationTimestamp.Before(&nodes[j].CreationTimestamp)
 			})
 			nodeTwo := nodes[1]
-			nodeTwo.Spec.Taints = []v1.Taint{}
-			env.ExpectCreatedOrUpdated(nodeTwo)
+			Eventually(func(g Gomega) {
+				n := &v1.Node{}
+				g.Expect(env.Client.Get(env.Context, client.ObjectKeyFromObject(nodeTwo), n)).To(Succeed())
+				stored := n.DeepCopy()
+				n.Spec.Taints = []v1.Taint{}
+				g.Expect(env.Client.Patch(env.Context, n, client.MergeFrom(stored))).To(Succeed())
+			}).Should(Succeed())
 		}
 
 		env.EventuallyExpectNotFound(pod, node)