fix: Unstable ut TestGetClientFailed (#31296) (#31472)

issue: #31295
pr: #31296

This PR fix unstable ut TestGetClientFailed

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
This commit is contained in:
wei liu 2024-03-22 11:01:07 +08:00 committed by GitHub
parent 0bf595a513
commit ef523bfef3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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() {