diff --git a/tests/integrations/mcs/discovery/register_test.go b/tests/integrations/mcs/discovery/register_test.go index 01db31dc6e8..27a5a937e4c 100644 --- a/tests/integrations/mcs/discovery/register_test.go +++ b/tests/integrations/mcs/discovery/register_test.go @@ -18,6 +18,7 @@ import ( "context" "strconv" "testing" + "time" "github.com/stretchr/testify/suite" bs "github.com/tikv/pd/pkg/basicserver" @@ -106,6 +107,9 @@ func (suite *serverRegisterTestSuite) checkServerRegister(serviceName string) { endpoints, err = discovery.Discover(client, suite.clusterID, serviceName) re.NoError(err) re.Empty(endpoints) + testutil.Eventually(re, func() bool { + return !s.IsServing() + }, testutil.WithWaitFor(3*time.Second), testutil.WithTickInterval(50*time.Millisecond)) } func (suite *serverRegisterTestSuite) TestServerPrimaryChange() {