Add eventually to MP reconcile test

Signed-off-by: killianmuldoon <kmuldoon@vmware.com>
kubernetes-sigs · Jun 27, 2023 · 5ee0bf4 · 5ee0bf4
1 parent 632fda1
commit 5ee0bf4
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/exp/internal/controllers/machinepool_controller_phases_test.go b/exp/internal/controllers/machinepool_controller_phases_test.go
@@ -1249,13 +1249,14 @@ func TestReconcileMachinePoolScaleToFromZero(t *testing.T) {
 		g.Expect(res.Requeue).To(BeFalse())
 
 		r.reconcilePhase(machinepool)
-
 		g.Expect(machinepool.Status.GetTypedPhase()).To(Equal(expv1.MachinePoolPhaseRunning))
 
 		delNode := &corev1.Node{}
-		err = env.GetAPIReader().Get(ctx, client.ObjectKeyFromObject(node), delNode)
-		g.Expect(err).ToNot(BeNil())
-		g.Expect(apierrors.IsNotFound(err)).To(BeTrue())
+		g.Eventually(func(g Gomega) {
+			err = fakeClient.Get(ctx, client.ObjectKeyFromObject(node), delNode)
+			g.Expect(err).ToNot(BeNil())
+			g.Expect(apierrors.IsNotFound(err)).To(BeTrue())
+		}).Should(Succeed())
 	})
 
 	t.Run("Should set `Running` when scaled to zero", func(t *testing.T) {