fix test

Signed-off-by: Ryan Leung <[email protected]>
tikv · Jun 3, 2024 · 3ae952f · 3ae952f
1 parent e16e3c3
commit 3ae952f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/integrations/client/client_test.go b/tests/integrations/client/client_test.go
@@ -1294,6 +1294,7 @@ func (suite *clientTestSuite) TestGetRegion() {
 		if r == nil {
 			return false
 		}
+		region.Leader = peers[0]
 		return reflect.DeepEqual(region, r.Meta) &&
 			reflect.DeepEqual(peers[0], r.Leader) &&
 			r.Buckets == nil
@@ -1375,6 +1376,7 @@ func (suite *clientTestSuite) TestGetPrevRegion() {
 			r, err := suite.client.GetPrevRegion(context.Background(), []byte{byte(i)})
 			re.NoError(err)
 			if i > 0 && i < regionLen {
+				regions[i-1].Leader = peers[0]
 				return reflect.DeepEqual(peers[0], r.Leader) &&
 					reflect.DeepEqual(regions[i-1], r.Meta)
 			}
@@ -1487,6 +1489,7 @@ func (suite *clientTestSuite) TestGetRegionByID() {
 		if r == nil {
 			return false
 		}
+		region.Leader = peers[0]
 		return reflect.DeepEqual(region, r.Meta) &&
 			reflect.DeepEqual(peers[0], r.Leader)
 	})