diff --git a/tests/integrations/mcs/scheduling/server_test.go b/tests/integrations/mcs/scheduling/server_test.go index eb99411d27e..afd213a624b 100644 --- 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 }) }