// Code generated by mockery v2.53.3. DO NOT EDIT.

package mock

import (
	bizerr "github.com/tidbcloud/pkgv2/bizerr"

	context "context"

	entity "github.com/tidbcloud/dataflow-service/app/repo/entity"

	mock "github.com/stretchr/testify/mock"
)

// Model is an autogenerated mock type for the Model type
type Model struct {
	mock.Mock
}

// DeregisterBackupClusterJob provides a mock function with given fields: ctx, clusterID, backupID
func (_m *Model) DeregisterBackupClusterJob(ctx context.Context, clusterID uint64, backupID uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, backupID)

	if len(ret) == 0 {
		panic("no return value specified for DeregisterBackupClusterJob")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, backupID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// DeregisterCDCClusterJob provides a mock function with given fields: ctx, clusterID, taskID
func (_m *Model) DeregisterCDCClusterJob(ctx context.Context, clusterID uint64, taskID uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, taskID)

	if len(ret) == 0 {
		panic("no return value specified for DeregisterCDCClusterJob")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, taskID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// DeregisterPiTRPauseLock provides a mock function with given fields: ctx, clusterID, turn
func (_m *Model) DeregisterPiTRPauseLock(ctx context.Context, clusterID uint64, turn uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, turn)

	if len(ret) == 0 {
		panic("no return value specified for DeregisterPiTRPauseLock")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, turn)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// DeregisterRestoreClusterJob provides a mock function with given fields: ctx, clusterID, restoreMode
func (_m *Model) DeregisterRestoreClusterJob(ctx context.Context, clusterID uint64, restoreMode entity.RestoreMode) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, restoreMode)

	if len(ret) == 0 {
		panic("no return value specified for DeregisterRestoreClusterJob")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, entity.RestoreMode) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, restoreMode)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// DeregisterRestoreClusterJobV2 provides a mock function with given fields: ctx, clusterID, taskID
func (_m *Model) DeregisterRestoreClusterJobV2(ctx context.Context, clusterID uint64, taskID uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, taskID)

	if len(ret) == 0 {
		panic("no return value specified for DeregisterRestoreClusterJobV2")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, taskID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// IsBackupClusterJobRegistered provides a mock function with given fields: ctx, clusterID, taskID
func (_m *Model) IsBackupClusterJobRegistered(ctx context.Context, clusterID uint64, taskID uint64) bool {
	ret := _m.Called(ctx, clusterID, taskID)

	if len(ret) == 0 {
		panic("no return value specified for IsBackupClusterJobRegistered")
	}

	var r0 bool
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bool); ok {
		r0 = rf(ctx, clusterID, taskID)
	} else {
		r0 = ret.Get(0).(bool)
	}

	return r0
}

// IsPiTRPauseLockRegistered provides a mock function with given fields: ctx, clusterID, turn
func (_m *Model) IsPiTRPauseLockRegistered(ctx context.Context, clusterID uint64, turn uint64) (bool, bizerr.BizError) {
	ret := _m.Called(ctx, clusterID, turn)

	if len(ret) == 0 {
		panic("no return value specified for IsPiTRPauseLockRegistered")
	}

	var r0 bool
	var r1 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) (bool, bizerr.BizError)); ok {
		return rf(ctx, clusterID, turn)
	}
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bool); ok {
		r0 = rf(ctx, clusterID, turn)
	} else {
		r0 = ret.Get(0).(bool)
	}

	if rf, ok := ret.Get(1).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r1 = rf(ctx, clusterID, turn)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(bizerr.BizError)
		}
	}

	return r0, r1
}

// RegisterBackupClusterJob provides a mock function with given fields: ctx, clusterID, taskID
func (_m *Model) RegisterBackupClusterJob(ctx context.Context, clusterID uint64, taskID uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, taskID)

	if len(ret) == 0 {
		panic("no return value specified for RegisterBackupClusterJob")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, taskID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// RegisterCDCClusterJob provides a mock function with given fields: ctx, clusterID, taskID
func (_m *Model) RegisterCDCClusterJob(ctx context.Context, clusterID uint64, taskID uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, taskID)

	if len(ret) == 0 {
		panic("no return value specified for RegisterCDCClusterJob")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, taskID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// RegisterPiTRPauseLock provides a mock function with given fields: ctx, clusterID, turn
func (_m *Model) RegisterPiTRPauseLock(ctx context.Context, clusterID uint64, turn uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, turn)

	if len(ret) == 0 {
		panic("no return value specified for RegisterPiTRPauseLock")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, turn)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// RegisterRestoreClusterV2 provides a mock function with given fields: ctx, clusterID, taskID
func (_m *Model) RegisterRestoreClusterV2(ctx context.Context, clusterID uint64, taskID uint64) bizerr.BizError {
	ret := _m.Called(ctx, clusterID, taskID)

	if len(ret) == 0 {
		panic("no return value specified for RegisterRestoreClusterV2")
	}

	var r0 bizerr.BizError
	if rf, ok := ret.Get(0).(func(context.Context, uint64, uint64) bizerr.BizError); ok {
		r0 = rf(ctx, clusterID, taskID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(bizerr.BizError)
		}
	}

	return r0
}

// NewModel creates a new instance of Model. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func NewModel(t interface {
	mock.TestingT
	Cleanup(func())
}) *Model {
	mock := &Model{}
	mock.Mock.Test(t)

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}