fix test

Signed-off-by: Ryan Leung <[email protected]>
tikv · Jun 3, 2024 · 2af142b · 2af142b
1 parent 3ae952f
commit 2af142b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/integrations/client/client_test.go b/tests/integrations/client/client_test.go
@@ -1418,6 +1418,7 @@ func (suite *clientTestSuite) TestScanRegions() {
 	})
 
 	// Set leader of region3 to nil.
+	regions[3].Leader = nil
 	region3 := core.NewRegionInfo(regions[3])
 	suite.srv.GetRaftCluster().HandleRegionHeartbeat(region3)
 
@@ -1439,6 +1440,9 @@ func (suite *clientTestSuite) TestScanRegions() {
 		t.Log("scanRegions", scanRegions)
 		t.Log("expect", expect)
 		for i := range expect {
+			if scanRegions[i].Meta.GetId() != region3.GetID() {
+				expect[i].Leader = expect[i].Peers[0]
+			}
 			re.Equal(expect[i], scanRegions[i].Meta)
 
 			if scanRegions[i].Meta.GetId() == region3.GetID() {