Use PollUntilContextTimeout to replace PollImmediateWithContext

Signed-off-by: bzsuni <bingzhe.sun@daocloud.io>
openshift · Jun 29, 2024 · 1599658 · 1599658
1 parent 34b8832
commit 1599658
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/e2e/framework/node/helper.go b/test/e2e/framework/node/helper.go
@@ -46,10 +46,11 @@ func WaitForAllNodesSchedulable(ctx context.Context, c clientset.Interface, time
 	}
 
 	framework.Logf("Waiting up to %v for all (but %d) nodes to be schedulable", timeout, framework.TestContext.AllowedNotReadyNodes)
-	return wait.PollImmediateWithContext(
+	return wait.PollUntilContextTimeout(
 		ctx,
 		30*time.Second,
 		timeout,
+		true,
 		CheckReadyForTests(ctx, c, framework.TestContext.NonblockingTaints, framework.TestContext.AllowedNotReadyNodes, largeClusterThreshold),
 	)
 }