// Code generated by MockGen. DO NOT EDIT.
// Source: ./internal/domain/service/dm.go
//
// Generated by this command:
//
//	mockgen -source=./internal/domain/service/dm.go -destination ./internal/domain/service/mock_service/mock_dm.go -package mock_service
//

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

import (
	context "context"
	reflect "reflect"

	bizerr "github.com/tidbcloud/pkgv2/bizerr"
	entity "github.com/tidbcloud/serverless-service/internal/domain/entity"
	tidb_mgmt_servicev1 "github.com/tidbcloud/tidb-management-service/api/spec/tidb_mgmt_service/v1"
	gomock "go.uber.org/mock/gomock"
)

// MockDMService is a mock of DMService interface.
type MockDMService struct {
	ctrl     *gomock.Controller
	recorder *MockDMServiceMockRecorder
}

// MockDMServiceMockRecorder is the mock recorder for MockDMService.
type MockDMServiceMockRecorder struct {
	mock *MockDMService
}

// NewMockDMService creates a new mock instance.
func NewMockDMService(ctrl *gomock.Controller) *MockDMService {
	mock := &MockDMService{ctrl: ctrl}
	mock.recorder = &MockDMServiceMockRecorder{mock}
	return mock
}

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

// CancelPrecheck mocks base method.
func (m *MockDMService) CancelPrecheck(ctx context.Context, clusterID, id string) bizerr.BizError {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CancelPrecheck", ctx, clusterID, id)
	ret0, _ := ret[0].(bizerr.BizError)
	return ret0
}

// CancelPrecheck indicates an expected call of CancelPrecheck.
func (mr *MockDMServiceMockRecorder) CancelPrecheck(ctx, clusterID, id any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelPrecheck", reflect.TypeOf((*MockDMService)(nil).CancelPrecheck), ctx, clusterID, id)
}

// CreatePrecheck mocks base method.
func (m *MockDMService) CreatePrecheck(ctx context.Context, precheck *entity.DMPrecheck) bizerr.BizError {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreatePrecheck", ctx, precheck)
	ret0, _ := ret[0].(bizerr.BizError)
	return ret0
}

// CreatePrecheck indicates an expected call of CreatePrecheck.
func (mr *MockDMServiceMockRecorder) CreatePrecheck(ctx, precheck any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreatePrecheck", reflect.TypeOf((*MockDMService)(nil).CreatePrecheck), ctx, precheck)
}

// CreateTask mocks base method.
func (m *MockDMService) CreateTask(ctx context.Context, task *entity.DMTask) (*entity.DMTask, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreateTask", ctx, task)
	ret0, _ := ret[0].(*entity.DMTask)
	ret1, _ := ret[1].(bizerr.BizError)
	return ret0, ret1
}

// CreateTask indicates an expected call of CreateTask.
func (mr *MockDMServiceMockRecorder) CreateTask(ctx, task any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockDMService)(nil).CreateTask), ctx, task)
}

// CreateVpcEndpoint mocks base method.
func (m *MockDMService) CreateVpcEndpoint(ctx context.Context, clusterID, serviceName string) (string, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreateVpcEndpoint", ctx, clusterID, serviceName)
	ret0, _ := ret[0].(string)
	ret1, _ := ret[1].(bizerr.BizError)
	return ret0, ret1
}

// CreateVpcEndpoint indicates an expected call of CreateVpcEndpoint.
func (mr *MockDMServiceMockRecorder) CreateVpcEndpoint(ctx, clusterID, serviceName any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateVpcEndpoint", reflect.TypeOf((*MockDMService)(nil).CreateVpcEndpoint), ctx, clusterID, serviceName)
}

// DeleteTask mocks base method.
func (m *MockDMService) DeleteTask(ctx context.Context, clusterID, taskID string) bizerr.BizError {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DeleteTask", ctx, clusterID, taskID)
	ret0, _ := ret[0].(bizerr.BizError)
	return ret0
}

// DeleteTask indicates an expected call of DeleteTask.
func (mr *MockDMServiceMockRecorder) DeleteTask(ctx, clusterID, taskID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockDMService)(nil).DeleteTask), ctx, clusterID, taskID)
}

// DescribeVpcEndpoint mocks base method.
func (m *MockDMService) DescribeVpcEndpoint(ctx context.Context, clusterID, endpointId string) (*entity.DMVpcEndpoint, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DescribeVpcEndpoint", ctx, clusterID, endpointId)
	ret0, _ := ret[0].(*entity.DMVpcEndpoint)
	ret1, _ := ret[1].(bizerr.BizError)
	return ret0, ret1
}

// DescribeVpcEndpoint indicates an expected call of DescribeVpcEndpoint.
func (mr *MockDMServiceMockRecorder) DescribeVpcEndpoint(ctx, clusterID, endpointId any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DescribeVpcEndpoint", reflect.TypeOf((*MockDMService)(nil).DescribeVpcEndpoint), ctx, clusterID, endpointId)
}

// GetPrecheck mocks base method.
func (m *MockDMService) GetPrecheck(ctx context.Context, clusterID, id string) (*entity.DMPrecheck, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPrecheck", ctx, clusterID, id)
	ret0, _ := ret[0].(*entity.DMPrecheck)
	ret1, _ := ret[1].(bizerr.BizError)
	return ret0, ret1
}

// GetPrecheck indicates an expected call of GetPrecheck.
func (mr *MockDMServiceMockRecorder) GetPrecheck(ctx, clusterID, id any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPrecheck", reflect.TypeOf((*MockDMService)(nil).GetPrecheck), ctx, clusterID, id)
}

// GetTask mocks base method.
func (m *MockDMService) GetTask(ctx context.Context, cluster *tidb_mgmt_servicev1.Cluster, ID string) (*entity.DMTask, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetTask", ctx, cluster, ID)
	ret0, _ := ret[0].(*entity.DMTask)
	ret1, _ := ret[1].(bizerr.BizError)
	return ret0, ret1
}

// GetTask indicates an expected call of GetTask.
func (mr *MockDMServiceMockRecorder) GetTask(ctx, cluster, ID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockDMService)(nil).GetTask), ctx, cluster, ID)
}

// ListTasks mocks base method.
func (m *MockDMService) ListTasks(ctx context.Context, filter *entity.DMTaskFilter) (int64, []*entity.DMTask, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListTasks", ctx, filter)
	ret0, _ := ret[0].(int64)
	ret1, _ := ret[1].([]*entity.DMTask)
	ret2, _ := ret[2].(bizerr.BizError)
	return ret0, ret1, ret2
}

// ListTasks indicates an expected call of ListTasks.
func (mr *MockDMServiceMockRecorder) ListTasks(ctx, filter any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTasks", reflect.TypeOf((*MockDMService)(nil).ListTasks), ctx, filter)
}

// PauseTask mocks base method.
func (m *MockDMService) PauseTask(ctx context.Context, cluster *tidb_mgmt_servicev1.Cluster, taskID string) bizerr.BizError {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "PauseTask", ctx, cluster, taskID)
	ret0, _ := ret[0].(bizerr.BizError)
	return ret0
}

// PauseTask indicates an expected call of PauseTask.
func (mr *MockDMServiceMockRecorder) PauseTask(ctx, cluster, taskID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PauseTask", reflect.TypeOf((*MockDMService)(nil).PauseTask), ctx, cluster, taskID)
}

// ResumeTask mocks base method.
func (m *MockDMService) ResumeTask(ctx context.Context, cluster *tidb_mgmt_servicev1.Cluster, taskID string) bizerr.BizError {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ResumeTask", ctx, cluster, taskID)
	ret0, _ := ret[0].(bizerr.BizError)
	return ret0
}

// ResumeTask indicates an expected call of ResumeTask.
func (mr *MockDMServiceMockRecorder) ResumeTask(ctx, cluster, taskID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResumeTask", reflect.TypeOf((*MockDMService)(nil).ResumeTask), ctx, cluster, taskID)
}