From 4371db4dbf384c7ef7c259e8d94994ed6621be07 Mon Sep 17 00:00:00 2001
From: Ryan Leung <rleungx@gmail.com>
Date: Mon, 3 Jun 2024 15:40:27 +0800
Subject: [PATCH] fix test

Signed-off-by: Ryan Leung <rleungx@gmail.com>
---
 tests/integrations/client/client_test.go | 7 +++++++
 tools/pd-ctl/tests/region/region_test.go | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/tests/integrations/client/client_test.go b/tests/integrations/client/client_test.go
index d0f5ad87ba9..a8a119fdfd6 100644
--- 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)
 			}
@@ -1416,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)
 
@@ -1437,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() {
@@ -1487,6 +1493,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)
 	})
diff --git a/tools/pd-ctl/tests/region/region_test.go b/tools/pd-ctl/tests/region/region_test.go
index 2952e137f3b..bcf6d79950b 100644
--- a/tools/pd-ctl/tests/region/region_test.go
+++ b/tools/pd-ctl/tests/region/region_test.go
@@ -306,7 +306,7 @@ func TestRegionNoLeader(t *testing.T) {
 			{Id: 6, StoreId: 3}},
 		RegionEpoch: &metapb.RegionEpoch{ConfVer: 1, Version: 1},
 	}
-	r := core.NewRegionInfo(metaRegion, nil)
+	r := core.NewRegionInfo(metaRegion)
 
 	cluster.GetLeaderServer().GetRaftCluster().GetBasicCluster().SetRegion(r)