cluster: make TestStoreOverloadedWithReplace stable (#8068)

ref #7969 Signed-off-by: husharp <[email protected]> Co-authored-by: ti-chi-bot[bot] <108142056+ti-chi-bot[bot]@users.noreply.github.com>
tikv · Apr 15, 2024 · a109f9c · a109f9c
1 parent 0214778
commit a109f9c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/server/cluster/cluster_test.go b/server/cluster/cluster_test.go
@@ -3480,10 +3480,11 @@ func TestStoreOverloadedWithReplace(t *testing.T) {
 	re.False(oc.AddOperator(op3))
 	ops, _ := lb.Schedule(tc, false /* dryRun */)
 	re.Empty(ops)
-	// sleep 2 seconds to make sure that token is filled up
-	time.Sleep(2 * time.Second)
-	ops, _ = lb.Schedule(tc, false /* dryRun */)
-	re.NotEmpty(ops)
+	// make sure that token is filled up
+	testutil.Eventually(re, func() bool {
+		ops, _ = lb.Schedule(tc, false /* dryRun */)
+		return len(ops) != 0
+	})
 }
 
 func TestDownStoreLimit(t *testing.T) {