fix test

Signed-off-by: Ryan Leung <[email protected]>

tests/integrations/client/global_config_test.go

@@ -15,6 +15,7 @@
 package client_test
 
 import (
+	"context"
 	"path"
 	"strconv"
 	"testing"
@@ -37,7 +38,8 @@ import (
 const globalConfigPath = "/global/config/"
 
 type testReceiver struct {
-	re *require.Assertions
+	re  *require.Assertions
+	ctx context.Context
 	grpc.ServerStream
 }
 
@@ -49,6 +51,10 @@ func (s testReceiver) Send(m *pdpb.WatchGlobalConfigResponse) error {
 	return nil
 }
 
+func (s testReceiver) Context() context.Context {
+	return s.ctx
+}
+
 type globalConfigTestSuite struct {
 	suite.Suite
 	server  *server.GrpcServer
@@ -199,7 +205,9 @@ func (suite *globalConfigTestSuite) TestWatch() {
 			re.NoError(err)
 		}
 	}()
-	server := testReceiver{re: suite.Require()}
+	ctx, cancel := context.WithCancel(suite.server.Context())
+	defer cancel()
+	server := testReceiver{re: suite.Require(), ctx: ctx}
 	go suite.server.WatchGlobalConfig(&pdpb.WatchGlobalConfigRequest{
 		ConfigPath: globalConfigPath,
 		Revision:   0,