From 1599658fc6b5b19781c28be7b8e87d59568fd654 Mon Sep 17 00:00:00 2001 From: bzsuni Date: Sat, 29 Jun 2024 18:50:34 +0800 Subject: [PATCH] Use PollUntilContextTimeout to replace PollImmediateWithContext Signed-off-by: bzsuni --- test/e2e/framework/node/helper.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/e2e/framework/node/helper.go b/test/e2e/framework/node/helper.go index f473d4ffa76..9bb76d97159 100644 --- 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), ) }