From 555d96ff05e06e5ce5b7b5f4ee4557a5f0cedde6 Mon Sep 17 00:00:00 2001 From: wei liu Date: Mon, 18 Mar 2024 13:51:04 +0800 Subject: [PATCH] fix: Unstable ut TestGetClientFailed (#31296) (#31335) issue: #31295 pr: #31335 This PR fix unstable ut TestGetClientFailed Signed-off-by: Wei Liu --- internal/proxy/look_aside_balancer_test.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/internal/proxy/look_aside_balancer_test.go b/internal/proxy/look_aside_balancer_test.go index ffbd53c409..e3db80dc7b 100644 --- a/internal/proxy/look_aside_balancer_test.go +++ b/internal/proxy/look_aside_balancer_test.go @@ -342,9 +342,10 @@ func (suite *LookAsideBalancerSuite) TestGetClientFailed() { suite.clientMgr.EXPECT().GetClient(mock.Anything, int64(2)).Return(nil, errors.New("shard client not found")) failCounter := atomic.NewInt64(0) suite.balancer.failedHeartBeatCounter.Insert(2, failCounter) - suite.Eventually(func() bool { - return failCounter.Load() == 0 - }, 10*time.Second, 1*time.Second) + + // slepp 10s, wait for checkNodeHealth execute for more than one round + time.Sleep(10 * time.Second) + suite.True(failCounter.Load() == 0) } func (suite *LookAsideBalancerSuite) TestNodeRecover() {