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

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

import (
	context "context"
	reflect "reflect"

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

// MockImportService is a mock of ImportService interface.
type MockImportService struct {
	ctrl     *gomock.Controller
	recorder *MockImportServiceMockRecorder
}

// MockImportServiceMockRecorder is the mock recorder for MockImportService.
type MockImportServiceMockRecorder struct {
	mock *MockImportService
}

// NewMockImportService creates a new mock instance.
func NewMockImportService(ctrl *gomock.Controller) *MockImportService {
	mock := &MockImportService{ctrl: ctrl}
	mock.recorder = &MockImportServiceMockRecorder{mock}
	return mock
}

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

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

// CancelImport indicates an expected call of CancelImport.
func (mr *MockImportServiceMockRecorder) CancelImport(ctx, clusterID, importID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelImport", reflect.TypeOf((*MockImportService)(nil).CancelImport), ctx, clusterID, importID)
}

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

// CancelImportByIDCompat indicates an expected call of CancelImportByIDCompat.
func (mr *MockImportServiceMockRecorder) CancelImportByIDCompat(ctx, clusterID, importIDCompat any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelImportByIDCompat", reflect.TypeOf((*MockImportService)(nil).CancelImportByIDCompat), ctx, clusterID, importIDCompat)
}

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

// CreateImport indicates an expected call of CreateImport.
func (mr *MockImportServiceMockRecorder) CreateImport(ctx, imp any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateImport", reflect.TypeOf((*MockImportService)(nil).CreateImport), ctx, imp)
}

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

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

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

// GetImport indicates an expected call of GetImport.
func (mr *MockImportServiceMockRecorder) GetImport(ctx, clusterID, importID any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetImport", reflect.TypeOf((*MockImportService)(nil).GetImport), ctx, clusterID, importID)
}

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

// GetImportByIDCompat indicates an expected call of GetImportByIDCompat.
func (mr *MockImportServiceMockRecorder) GetImportByIDCompat(ctx, clusterID, importIDCompat any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetImportByIDCompat", reflect.TypeOf((*MockImportService)(nil).GetImportByIDCompat), ctx, clusterID, importIDCompat)
}

// ListImports mocks base method.
func (m *MockImportService) ListImports(ctx context.Context, opts *vo.ListImportOpt) (int64, []*entity.Import, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListImports", ctx, opts)
	ret0, _ := ret[0].(int64)
	ret1, _ := ret[1].([]*entity.Import)
	ret2, _ := ret[2].(bizerr.BizError)
	return ret0, ret1, ret2
}

// ListImports indicates an expected call of ListImports.
func (mr *MockImportServiceMockRecorder) ListImports(ctx, opts any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListImports", reflect.TypeOf((*MockImportService)(nil).ListImports), ctx, opts)
}

// NewInfraAPIImportObject mocks base method.
func (m *MockImportService) NewInfraAPIImportObject(ctx context.Context, imp *entity.ImportTask, cloudAdminPassword string, isSuperVip bool) (*v1.Import, bizerr.BizError) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "NewInfraAPIImportObject", ctx, imp, cloudAdminPassword, isSuperVip)
	ret0, _ := ret[0].(*v1.Import)
	ret1, _ := ret[1].(bizerr.BizError)
	return ret0, ret1
}

// NewInfraAPIImportObject indicates an expected call of NewInfraAPIImportObject.
func (mr *MockImportServiceMockRecorder) NewInfraAPIImportObject(ctx, imp, cloudAdminPassword, isSuperVip any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewInfraAPIImportObject", reflect.TypeOf((*MockImportService)(nil).NewInfraAPIImportObject), ctx, imp, cloudAdminPassword, isSuperVip)
}

// PublishImportEvent mocks base method.
func (m *MockImportService) PublishImportEvent(ctx context.Context, imp *entity.ImportTask, status event.EventStatus, eventType string) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "PublishImportEvent", ctx, imp, status, eventType)
}

// PublishImportEvent indicates an expected call of PublishImportEvent.
func (mr *MockImportServiceMockRecorder) PublishImportEvent(ctx, imp, status, eventType any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishImportEvent", reflect.TypeOf((*MockImportService)(nil).PublishImportEvent), ctx, imp, status, eventType)
}