Address the comments

Signed-off-by: JmPotato <[email protected]>

client/client_test.go

@@ -23,9 +23,15 @@ import (
 	"github.com/pingcap/errors"
 	"github.com/pingcap/kvproto/pkg/pdpb"
 	"github.com/stretchr/testify/require"
+	"github.com/tikv/pd/client/testutil"
+	"go.uber.org/goleak"
 	"google.golang.org/grpc"
 )
 
+func TestMain(m *testing.M) {
+	goleak.VerifyTestMain(m, testutil.LeakOptions...)
+}
+
 func TestTsLessEqual(t *testing.T) {
 	re := require.New(t)
 	re.True(tsLessEqual(9, 9, 9, 9))

tests/client/client_test.go

@@ -1,4 +1,4 @@
-// Copyright 2018 TiKV Project Authortc.
+// Copyright 2018 TiKV Project Authors.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -43,13 +43,18 @@ import (
 	"github.com/tikv/pd/server/storage/endpoint"
 	"github.com/tikv/pd/server/tso"
 	"github.com/tikv/pd/tests"
+	"go.uber.org/goleak"
 )
 
 const (
 	tsoRequestConcurrencyNumber = 5
 	tsoRequestRound             = 30
 )
 
+func TestMain(m *testing.M) {
+	goleak.VerifyTestMain(m, testutil.LeakOptions...)
+}
+
 type client interface {
 	GetLeaderAddr() string
 	ScheduleCheckLeader()