// Code generated by MockGen. DO NOT EDIT.
// Source: github.com/tidbcloud/devops/pkg/service/tidbtools/api (interfaces: TidbToolsServiceAPI)
//
// Generated by this command:
//
//	mockgen -destination service_mock.go -package=api github.com/tidbcloud/devops/pkg/service/tidbtools/api TidbToolsServiceAPI
//

// Package api is a generated GoMock package.
package api

import (
	context "context"
	reflect "reflect"

	v1beta2 "github.com/tidbcloud/dataflow-service-ng/idl/pbgen/proto/import_into/v1beta2"
	v1beta20 "github.com/tidbcloud/dataflow-service-ng/idl/pbgen/proto/premium_dm/v1beta2"
	dataflow "github.com/tidbcloud/dataflow-service/idl/pbgen/proto/dataflow"
	dm "github.com/tidbcloud/dataflow-service/idl/pbgen/proto/dm"
	nextgenv1 "github.com/tidbcloud/tidb-management-service/api/spec/global/nextgen/v1"
	gomock "go.uber.org/mock/gomock"
)

// MockTidbToolsServiceAPI is a mock of TidbToolsServiceAPI interface.
type MockTidbToolsServiceAPI struct {
	ctrl     *gomock.Controller
	recorder *MockTidbToolsServiceAPIMockRecorder
	isgomock struct{}
}

// MockTidbToolsServiceAPIMockRecorder is the mock recorder for MockTidbToolsServiceAPI.
type MockTidbToolsServiceAPIMockRecorder struct {
	mock *MockTidbToolsServiceAPI
}

// NewMockTidbToolsServiceAPI creates a new mock instance.
func NewMockTidbToolsServiceAPI(ctrl *gomock.Controller) *MockTidbToolsServiceAPI {
	mock := &MockTidbToolsServiceAPI{ctrl: ctrl}
	mock.recorder = &MockTidbToolsServiceAPIMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockTidbToolsServiceAPI) EXPECT() *MockTidbToolsServiceAPIMockRecorder {
	return m.recorder
}

// GetBackupSetting mocks base method.
func (m *MockTidbToolsServiceAPI) GetBackupSetting(ctx context.Context, orgID, projectID, clusterID uint64) (*dataflow.QueryBackupSettingResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetBackupSetting", ctx, orgID, projectID, clusterID)
	ret0, _ := ret[0].(*dataflow.QueryBackupSettingResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetBackupSetting indicates an expected call of GetBackupSetting.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetBackupSetting(ctx, orgID, projectID, clusterID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBackupSetting", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetBackupSetting), ctx, orgID, projectID, clusterID)
}

// GetDMJob mocks base method.
func (m *MockTidbToolsServiceAPI) GetDMJob(ctx context.Context, orgID, projectID, clusterID, jobID uint64) (*dm.GetJobResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetDMJob", ctx, orgID, projectID, clusterID, jobID)
	ret0, _ := ret[0].(*dm.GetJobResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetDMJob indicates an expected call of GetDMJob.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetDMJob(ctx, orgID, projectID, clusterID, jobID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDMJob", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetDMJob), ctx, orgID, projectID, clusterID, jobID)
}

// GetDMPrecheck mocks base method.
func (m *MockTidbToolsServiceAPI) GetDMPrecheck(ctx context.Context, orgID, projectID, clusterID, precheckID uint64) (*dm.GetPrecheckJobResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetDMPrecheck", ctx, orgID, projectID, clusterID, precheckID)
	ret0, _ := ret[0].(*dm.GetPrecheckJobResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetDMPrecheck indicates an expected call of GetDMPrecheck.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetDMPrecheck(ctx, orgID, projectID, clusterID, precheckID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDMPrecheck", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetDMPrecheck), ctx, orgID, projectID, clusterID, precheckID)
}

// GetImportIntoTask mocks base method.
func (m *MockTidbToolsServiceAPI) GetImportIntoTask(ctx context.Context, tidbID, taskID string) (*v1beta2.ImportIntoTask, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetImportIntoTask", ctx, tidbID, taskID)
	ret0, _ := ret[0].(*v1beta2.ImportIntoTask)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetImportIntoTask indicates an expected call of GetImportIntoTask.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetImportIntoTask(ctx, tidbID, taskID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetImportIntoTask", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetImportIntoTask), ctx, tidbID, taskID)
}

// GetPremiumBackupSetting mocks base method.
func (m *MockTidbToolsServiceAPI) GetPremiumBackupSetting(ctx context.Context, orgID, projectID, clusterID uint64) (*nextgenv1.BackupSetting, *nextgenv1.ListBackupPoliciesResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPremiumBackupSetting", ctx, orgID, projectID, clusterID)
	ret0, _ := ret[0].(*nextgenv1.BackupSetting)
	ret1, _ := ret[1].(*nextgenv1.ListBackupPoliciesResponse)
	ret2, _ := ret[2].(error)
	return ret0, ret1, ret2
}

// GetPremiumBackupSetting indicates an expected call of GetPremiumBackupSetting.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetPremiumBackupSetting(ctx, orgID, projectID, clusterID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPremiumBackupSetting", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetPremiumBackupSetting), ctx, orgID, projectID, clusterID)
}

// GetPremiumDMJob mocks base method.
func (m *MockTidbToolsServiceAPI) GetPremiumDMJob(ctx context.Context, tidbID, migrationID string) (*v1beta20.PremiumMigration, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPremiumDMJob", ctx, tidbID, migrationID)
	ret0, _ := ret[0].(*v1beta20.PremiumMigration)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPremiumDMJob indicates an expected call of GetPremiumDMJob.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetPremiumDMJob(ctx, tidbID, migrationID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPremiumDMJob", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetPremiumDMJob), ctx, tidbID, migrationID)
}

// GetPremiumDMPrecheck mocks base method.
func (m *MockTidbToolsServiceAPI) GetPremiumDMPrecheck(ctx context.Context, tidbID, precheckID string) (*v1beta20.PremiumMigrationPrecheck, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPremiumDMPrecheck", ctx, tidbID, precheckID)
	ret0, _ := ret[0].(*v1beta20.PremiumMigrationPrecheck)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPremiumDMPrecheck indicates an expected call of GetPremiumDMPrecheck.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetPremiumDMPrecheck(ctx, tidbID, precheckID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPremiumDMPrecheck", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetPremiumDMPrecheck), ctx, tidbID, precheckID)
}

// GetPremiumRestore mocks base method.
func (m *MockTidbToolsServiceAPI) GetPremiumRestore(ctx context.Context, orgID, projectID, clusterID uint64) (*PremiumRestore, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPremiumRestore", ctx, orgID, projectID, clusterID)
	ret0, _ := ret[0].(*PremiumRestore)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPremiumRestore indicates an expected call of GetPremiumRestore.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetPremiumRestore(ctx, orgID, projectID, clusterID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPremiumRestore", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetPremiumRestore), ctx, orgID, projectID, clusterID)
}

// GetPremiumRestoreStatus mocks base method.
func (m *MockTidbToolsServiceAPI) GetPremiumRestoreStatus(ctx context.Context, orgID, projectID, clusterID uint64) (*nextgenv1.RestoreStatusResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPremiumRestoreStatus", ctx, orgID, projectID, clusterID)
	ret0, _ := ret[0].(*nextgenv1.RestoreStatusResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPremiumRestoreStatus indicates an expected call of GetPremiumRestoreStatus.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetPremiumRestoreStatus(ctx, orgID, projectID, clusterID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPremiumRestoreStatus", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetPremiumRestoreStatus), ctx, orgID, projectID, clusterID)
}

// GetRestore mocks base method.
func (m *MockTidbToolsServiceAPI) GetRestore(ctx context.Context, orgID, projectID, clusterID, restoreID uint64) (*dataflow.GetRestoreResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetRestore", ctx, orgID, projectID, clusterID, restoreID)
	ret0, _ := ret[0].(*dataflow.GetRestoreResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetRestore indicates an expected call of GetRestore.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetRestore(ctx, orgID, projectID, clusterID, restoreID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRestore", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetRestore), ctx, orgID, projectID, clusterID, restoreID)
}

// GetRestoreProgress mocks base method.
func (m *MockTidbToolsServiceAPI) GetRestoreProgress(ctx context.Context, orgID, projectID, clusterID uint64) (*dataflow.RestoreProgressResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetRestoreProgress", ctx, orgID, projectID, clusterID)
	ret0, _ := ret[0].(*dataflow.RestoreProgressResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetRestoreProgress indicates an expected call of GetRestoreProgress.
func (mr *MockTidbToolsServiceAPIMockRecorder) GetRestoreProgress(ctx, orgID, projectID, clusterID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRestoreProgress", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).GetRestoreProgress), ctx, orgID, projectID, clusterID)
}

// ListBackupExports mocks base method.
func (m *MockTidbToolsServiceAPI) ListBackupExports(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize uint32) (*dataflow.ListBackupExportsResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListBackupExports", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dataflow.ListBackupExportsResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListBackupExports indicates an expected call of ListBackupExports.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListBackupExports(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBackupExports", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListBackupExports), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListBackups mocks base method.
func (m *MockTidbToolsServiceAPI) ListBackups(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize uint32) (*dataflow.Backups, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListBackups", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dataflow.Backups)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListBackups indicates an expected call of ListBackups.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListBackups(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBackups", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListBackups), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListDMJobs mocks base method.
func (m *MockTidbToolsServiceAPI) ListDMJobs(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize int32) (*dm.ListJobsResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListDMJobs", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dm.ListJobsResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListDMJobs indicates an expected call of ListDMJobs.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListDMJobs(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDMJobs", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListDMJobs), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListDMPrechecks mocks base method.
func (m *MockTidbToolsServiceAPI) ListDMPrechecks(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize int32) (*dm.ListPrechecksResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListDMPrechecks", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dm.ListPrechecksResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListDMPrechecks indicates an expected call of ListDMPrechecks.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListDMPrechecks(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDMPrechecks", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListDMPrechecks), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListImportIntoTasks mocks base method.
func (m *MockTidbToolsServiceAPI) ListImportIntoTasks(ctx context.Context, tidbID string, pageSize int32) (*v1beta2.ListImportIntoTasksResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListImportIntoTasks", ctx, tidbID, pageSize)
	ret0, _ := ret[0].(*v1beta2.ListImportIntoTasksResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListImportIntoTasks indicates an expected call of ListImportIntoTasks.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListImportIntoTasks(ctx, tidbID, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListImportIntoTasks", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListImportIntoTasks), ctx, tidbID, pageSize)
}

// ListImports mocks base method.
func (m *MockTidbToolsServiceAPI) ListImports(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize int32) (*dataflow.ListImportResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListImports", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dataflow.ListImportResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListImports indicates an expected call of ListImports.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListImports(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListImports", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListImports), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListLogBackups mocks base method.
func (m *MockTidbToolsServiceAPI) ListLogBackups(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize uint32) (*dataflow.LogBackupListResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListLogBackups", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dataflow.LogBackupListResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListLogBackups indicates an expected call of ListLogBackups.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListLogBackups(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListLogBackups", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListLogBackups), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListPremiumBackups mocks base method.
func (m *MockTidbToolsServiceAPI) ListPremiumBackups(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize int32) (*nextgenv1.ListClusterBackupsResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListPremiumBackups", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*nextgenv1.ListClusterBackupsResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListPremiumBackups indicates an expected call of ListPremiumBackups.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListPremiumBackups(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPremiumBackups", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListPremiumBackups), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListPremiumDMJobs mocks base method.
func (m *MockTidbToolsServiceAPI) ListPremiumDMJobs(ctx context.Context, tidbID string, pageSize int32) (*v1beta20.ListPremiumMigrationsResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListPremiumDMJobs", ctx, tidbID, pageSize)
	ret0, _ := ret[0].(*v1beta20.ListPremiumMigrationsResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListPremiumDMJobs indicates an expected call of ListPremiumDMJobs.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListPremiumDMJobs(ctx, tidbID, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPremiumDMJobs", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListPremiumDMJobs), ctx, tidbID, pageSize)
}

// ListPremiumDMPrechecks mocks base method.
func (m *MockTidbToolsServiceAPI) ListPremiumDMPrechecks(ctx context.Context, tidbID string, pageSize int32) (*v1beta20.ListPremiumMigrationPrechecksResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListPremiumDMPrechecks", ctx, tidbID, pageSize)
	ret0, _ := ret[0].(*v1beta20.ListPremiumMigrationPrechecksResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListPremiumDMPrechecks indicates an expected call of ListPremiumDMPrechecks.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListPremiumDMPrechecks(ctx, tidbID, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPremiumDMPrechecks", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListPremiumDMPrechecks), ctx, tidbID, pageSize)
}

// ListPremiumExports mocks base method.
func (m *MockTidbToolsServiceAPI) ListPremiumExports(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize int32) (*nextgenv1.ListExportsResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListPremiumExports", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*nextgenv1.ListExportsResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListPremiumExports indicates an expected call of ListPremiumExports.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListPremiumExports(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPremiumExports", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListPremiumExports), ctx, orgID, projectID, clusterID, page, pageSize)
}

// ListRestores mocks base method.
func (m *MockTidbToolsServiceAPI) ListRestores(ctx context.Context, orgID, projectID, clusterID uint64, page, pageSize uint32) (*dataflow.ListRestoresResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListRestores", ctx, orgID, projectID, clusterID, page, pageSize)
	ret0, _ := ret[0].(*dataflow.ListRestoresResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListRestores indicates an expected call of ListRestores.
func (mr *MockTidbToolsServiceAPIMockRecorder) ListRestores(ctx, orgID, projectID, clusterID, page, pageSize any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRestores", reflect.TypeOf((*MockTidbToolsServiceAPI)(nil).ListRestores), ctx, orgID, projectID, clusterID, page, pageSize)
}