milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

issue: #31295
pr: #31335

This PR fix unstable ut TestGetClientFailed

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
pull/31344/head
wei liu 2024-03-18 13:51:04 +08:00 committed by GitHub
parent 3987cd69d7
commit 555d96ff05
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
internal/proxy/look_aside_balancer_test.go
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() {