resourcemanager: make TestResourceManagerServer more stable (tikv#8418)

close tikv#8417 Signed-off-by: husharp <[email protected]> Co-authored-by: ti-chi-bot[bot] <108142056+ti-chi-bot[bot]@users.noreply.github.com>
HuSharp · Jul 19, 2024 · c514937 · c514937
1 parent 1c86ddb
commit c514937
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/integrations/mcs/resourcemanager/server_test.go b/tests/integrations/mcs/resourcemanager/server_test.go
@@ -25,6 +25,7 @@ import (
 	rmpb "github.com/pingcap/kvproto/pkg/resource_manager"
 	"github.com/stretchr/testify/require"
 	"github.com/tikv/pd/client/grpcutil"
+	bs "github.com/tikv/pd/pkg/basicserver"
 	"github.com/tikv/pd/pkg/utils/tempurl"
 	"github.com/tikv/pd/pkg/versioninfo"
 	"github.com/tikv/pd/tests"
@@ -49,6 +50,7 @@ func TestResourceManagerServer(t *testing.T) {
 	s, cleanup := tests.StartSingleResourceManagerTestServer(ctx, re, leader.GetAddr(), tempurl.Alloc())
 	addr := s.GetAddr()
 	defer cleanup()
+	tests.WaitForPrimaryServing(re, map[string]bs.Server{addr: s})
 
 	// Test registered GRPC Service
 	cc, err := grpcutil.GetClientConn(ctx, addr, nil)