chore: fix mockPdClient

Signed-off-by: ekexium <eke@fastmail.com>
pingcap · Mar 13, 2024 · d97c141 · d97c141
1 parent d6da3c1
commit d97c141
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/br/pkg/lightning/backend/local/local_test.go b/br/pkg/lightning/backend/local/local_test.go
@@ -679,6 +679,10 @@ func (c *mockPdClient) GetTS(ctx context.Context) (int64, int64, error) {
 	return 1, 2, nil
 }
 
+func (c *mockPdClient) GetClusterID(ctx context.Context) uint64 {
+	return 1
+}
+
 type mockGrpcErr struct{}
 
 func (e mockGrpcErr) GRPCStatus() *status.Status {

diff --git a/br/pkg/streamhelper/basic_lib_for_test.go b/br/pkg/streamhelper/basic_lib_for_test.go
@@ -752,6 +752,10 @@ func (p *mockPDClient) GetStore(_ context.Context, storeID uint64) (*metapb.Stor
 	}, nil
 }
 
+func (p *mockPDClient) GetClusterID(ctx context.Context) uint64 {
+	return 1
+}
+
 func newMockRegion(regionID uint64, startKey []byte, endKey []byte) *pd.Region {
 	leader := &metapb.Peer{
 		Id:      regionID,