diff --git a/test/suites/drift/suite_test.go b/test/suites/drift/suite_test.go index 7d2d2f989c0f..266ab9ac81d4 100644 --- a/test/suites/drift/suite_test.go +++ b/test/suites/drift/suite_test.go @@ -588,6 +588,10 @@ var _ = Describe("Drift", func() { // Remove the startup taints from the new nodes to initialize them Eventually(func(g Gomega) { g.Expect(env.Client.Get(env.Context, client.ObjectKeyFromObject(nodeTwo), nodeTwo)).To(Succeed()) + _, found := lo.Find(nodeTwo.Spec.Taints, func(t corev1.Taint) bool { + return t.MatchTaint(&corev1.Taint{Key: "example.com/another-taint-2", Effect: corev1.TaintEffectNoExecute}) + }) + g.Expect(found).To(BeTrue()) stored := nodeTwo.DeepCopy() nodeTwo.Spec.Taints = lo.Reject(nodeTwo.Spec.Taints, func(t corev1.Taint, _ int) bool { return t.Key == "example.com/another-taint-2" }) g.Expect(env.Client.Patch(env.Context, nodeTwo, client.StrategicMergeFrom(stored))).To(Succeed())