replace tidb

Signed-off-by: Cabinfever_B <[email protected]>
tikv · Nov 30, 2023 · 2c2d3bf · 2c2d3bf
1 parent 26d7ec7
commit 2c2d3bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/integration_tests/split_test.go b/integration_tests/split_test.go
@@ -201,7 +201,7 @@ func (c *mockPDClient) GetRegion(ctx context.Context, key []byte, opts ...pd.Get
 	return c.client.GetRegion(ctx, key, opts...)
 }
 
-func (c *mockPDClient) GetRegionFromMember(ctx context.Context, key []byte, memberURLs []string) (*pd.Region, error) {
+func (c *mockPDClient) GetRegionFromMember(ctx context.Context, key []byte, memberURLs []string, opts ...pd.GetRegionOption) (*pd.Region, error) {
 	return nil, nil
 }
 
@@ -225,7 +225,7 @@ func (c *mockPDClient) GetRegionByID(ctx context.Context, regionID uint64, opts
 	return c.client.GetRegionByID(ctx, regionID, opts...)
 }
 
-func (c *mockPDClient) ScanRegions(ctx context.Context, startKey []byte, endKey []byte, limit int) ([]*pd.Region, error) {
+func (c *mockPDClient) ScanRegions(ctx context.Context, startKey []byte, endKey []byte, limit int, opts ...pd.GetRegionOption) ([]*pd.Region, error) {
 	c.RLock()
 	defer c.RUnlock()