fix test

Signed-off-by: Ryan Leung <[email protected]>
tikv · Aug 21, 2024 · dc6b4a7 · dc6b4a7
1 parent 30ebb5d
commit dc6b4a7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/client/tso_dispatcher.go b/client/tso_dispatcher.go
@@ -308,6 +308,7 @@ tsoBatchLoop:
 			}
 			// Because ScheduleCheckMemberChanged is asynchronous, if the leader changes, we better call `updateMember` ASAP.
 			if errs.IsLeaderChange(err) {
+				svcDiscovery.ScheduleCheckServiceModeChanged()
 				if err := bo.Exec(ctx, svcDiscovery.CheckMemberChanged); err != nil {
 					select {
 					case <-ctx.Done():

diff --git a/tests/integrations/mcs/tso/server_test.go b/tests/integrations/mcs/tso/server_test.go
@@ -355,12 +355,16 @@ func TestResignAPIPrimaryForward(t *testing.T) {
 	defer func() {
 		re.NoError(failpoint.Disable("github.com/tikv/pd/pkg/member/skipCampaignLeaderCheck"))
 	}()
+	pdClient, err := pd.NewClientWithContext(context.Background(),
+		[]string{suite.backendEndpoints}, pd.SecurityOption{}, pd.WithMaxErrorRetry(1))
+	re.NoError(err)
+	defer pdClient.Close()
 
 	for j := 0; j < 10; j++ {
 		suite.pdLeader.ResignLeader()
 		suite.pdLeader = suite.cluster.GetServer(suite.cluster.WaitLeader())
 		suite.backendEndpoints = suite.pdLeader.GetAddr()
-		_, _, err = suite.pdClient.GetTS(suite.ctx)
+		_, _, err = pdClient.GetTS(suite.ctx)
 		re.NoError(err)
 	}
 }