make TestScheduler stable

Signed-off-by: Ryan Leung <rleungx@gmail.com>
tikv · Apr 25, 2023 · e3f8525 · e3f8525
1 parent 4f87e9d
commit e3f8525
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/pdctl/scheduler/scheduler_test.go b/tests/pdctl/scheduler/scheduler_test.go
@@ -460,7 +460,15 @@ func TestScheduler(t *testing.T) {
 	checkSchedulerWithStatusCommand(nil, "paused", []string{
 		"balance-leader-scheduler",
 	})
-	checkSchedulerDescribeCommand("balance-leader-scheduler", "paused", "")
+	result := make(map[string]interface{})
+	testutil.Eventually(re, func() bool {
+		mightExec([]string{"-u", pdAddr, "scheduler", "describe", "balance-leader-scheduler"}, &result)
+		return len(result) != 0
+	}, testutil.WithTickInterval(50*time.Millisecond))
+
+	testutil.Eventually(re, func() bool {
+		return result["status"] == "paused" && result["summary"] == ""
+	}, testutil.WithTickInterval(50*time.Millisecond))
 
 	mustUsage([]string{"-u", pdAddr, "scheduler", "resume", "balance-leader-scheduler", "60"})
 	mustExec([]string{"-u", pdAddr, "scheduler", "resume", "balance-leader-scheduler"}, nil)