make TestPrimaryChange stable (#7412)

Signed-off-by: Ryan Leung <rleungx@gmail.com>
tikv · Nov 22, 2023 · 7783c74 · 7783c74
1 parent 20da1c7
commit 7783c74
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/integrations/mcs/scheduling/server_test.go b/tests/integrations/mcs/scheduling/server_test.go
@@ -127,21 +127,21 @@ func (suite *serverTestSuite) TestPrimaryChange() {
 	tc.WaitForPrimaryServing(re)
 	primary := tc.GetPrimaryServer()
 	oldPrimaryAddr := primary.GetAddr()
-	re.Len(primary.GetCluster().GetCoordinator().GetSchedulersController().GetSchedulerNames(), 5)
 	testutil.Eventually(re, func() bool {
 		watchedAddr, ok := suite.pdLeader.GetServicePrimaryAddr(suite.ctx, mcs.SchedulingServiceName)
-		return ok && oldPrimaryAddr == watchedAddr
+		return ok && oldPrimaryAddr == watchedAddr &&
+			len(primary.GetCluster().GetCoordinator().GetSchedulersController().GetSchedulerNames()) == 5
 	})
-	// transfer leader
+	// change primary
 	primary.Close()
 	tc.WaitForPrimaryServing(re)
 	primary = tc.GetPrimaryServer()
 	newPrimaryAddr := primary.GetAddr()
 	re.NotEqual(oldPrimaryAddr, newPrimaryAddr)
-	re.Len(primary.GetCluster().GetCoordinator().GetSchedulersController().GetSchedulerNames(), 5)
 	testutil.Eventually(re, func() bool {
 		watchedAddr, ok := suite.pdLeader.GetServicePrimaryAddr(suite.ctx, mcs.SchedulingServiceName)
-		return ok && newPrimaryAddr == watchedAddr
+		return ok && newPrimaryAddr == watchedAddr &&
+			len(primary.GetCluster().GetCoordinator().GetSchedulersController().GetSchedulerNames()) == 5
 	})
 }