From 0069252d45e3cd530cdda0491246eb758bdb0caf Mon Sep 17 00:00:00 2001 From: Ryan Leung Date: Tue, 18 Apr 2023 17:44:32 +0800 Subject: [PATCH] make TestServerRegister stable Signed-off-by: Ryan Leung --- tests/integrations/mcs/discovery/register_test.go | 4 ++++ 1 file changed, 4 insertions(+) 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() {