From a76ec43bef2e7c10d332669cea38d29e8ab1e35a Mon Sep 17 00:00:00 2001 From: fabriziopandini Date: Tue, 19 Mar 2024 14:51:37 +0100 Subject: [PATCH] Fix TestPatchNode flake --- .../machine/machine_controller_noderef_test.go | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/internal/controllers/machine/machine_controller_noderef_test.go b/internal/controllers/machine/machine_controller_noderef_test.go index a805e57201aa..48ebcabf3b02 100644 --- a/internal/controllers/machine/machine_controller_noderef_test.go +++ b/internal/controllers/machine/machine_controller_noderef_test.go @@ -913,12 +913,12 @@ func TestPatchNode(t *testing.T) { ms := tc.ms.DeepCopy() md := tc.md.DeepCopy() - g.Expect(env.Create(ctx, oldNode)).To(Succeed()) - g.Expect(env.Create(ctx, machine)).To(Succeed()) - g.Expect(env.Create(ctx, ms)).To(Succeed()) - g.Expect(env.Create(ctx, md)).To(Succeed()) + g.Expect(env.CreateAndWait(ctx, oldNode)).To(Succeed()) + g.Expect(env.CreateAndWait(ctx, machine)).To(Succeed()) + g.Expect(env.CreateAndWait(ctx, ms)).To(Succeed()) + g.Expect(env.CreateAndWait(ctx, md)).To(Succeed()) t.Cleanup(func() { - _ = env.Cleanup(ctx, oldNode, machine, ms, md) + _ = env.CleanupAndWait(ctx, oldNode, machine, ms, md) }) err := r.patchNode(ctx, env, oldNode, tc.newLabels, tc.newAnnotations, tc.machine)