// Code generated by MockGen. DO NOT EDIT.
// Source: dataflow.go
//
// Generated by this command:
//
//	mockgen -source=dataflow.go -package=mock_dataflow -destination=./mock_dataflow/mock_dataflow.go
//

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

import (
	context "context"
	reflect "reflect"

	dataflow "github.com/tidbcloud/dataflow-service/idl/pbgen/proto/dataflow"
	gomock "go.uber.org/mock/gomock"
)

// MockDataFlow is a mock of DataFlow interface.
type MockDataFlow struct {
	ctrl     *gomock.Controller
	recorder *MockDataFlowMockRecorder
	isgomock struct{}
}

// MockDataFlowMockRecorder is the mock recorder for MockDataFlow.
type MockDataFlowMockRecorder struct {
	mock *MockDataFlow
}

// NewMockDataFlow creates a new mock instance.
func NewMockDataFlow(ctrl *gomock.Controller) *MockDataFlow {
	mock := &MockDataFlow{ctrl: ctrl}
	mock.recorder = &MockDataFlowMockRecorder{mock}
	return mock
}

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

// CreateConnector mocks base method.
func (m *MockDataFlow) CreateConnector(ctx context.Context, req *dataflow.ConnectorCreateReq) (*dataflow.ConnectorID, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreateConnector", ctx, req)
	ret0, _ := ret[0].(*dataflow.ConnectorID)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// CreateConnector indicates an expected call of CreateConnector.
func (mr *MockDataFlowMockRecorder) CreateConnector(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateConnector", reflect.TypeOf((*MockDataFlow)(nil).CreateConnector), ctx, req)
}

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

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

// CreateSinkEndpoint mocks base method.
func (m *MockDataFlow) CreateSinkEndpoint(ctx context.Context, req *dataflow.CreateSinkEndpointRequest) (*dataflow.SinkEndpoint, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreateSinkEndpoint", ctx, req)
	ret0, _ := ret[0].(*dataflow.SinkEndpoint)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// CreateSinkEndpoint indicates an expected call of CreateSinkEndpoint.
func (mr *MockDataFlowMockRecorder) CreateSinkEndpoint(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSinkEndpoint", reflect.TypeOf((*MockDataFlow)(nil).CreateSinkEndpoint), ctx, req)
}

// DeleteConnector mocks base method.
func (m *MockDataFlow) DeleteConnector(ctx context.Context, req *dataflow.ConnectorID) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DeleteConnector", ctx, req)
	ret0, _ := ret[0].(error)
	return ret0
}

// DeleteConnector indicates an expected call of DeleteConnector.
func (mr *MockDataFlowMockRecorder) DeleteConnector(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteConnector", reflect.TypeOf((*MockDataFlow)(nil).DeleteConnector), ctx, req)
}

// DeleteSinkEndpoint mocks base method.
func (m *MockDataFlow) DeleteSinkEndpoint(ctx context.Context, req *dataflow.DeleteSinkEndpointRequest) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DeleteSinkEndpoint", ctx, req)
	ret0, _ := ret[0].(error)
	return ret0
}

// DeleteSinkEndpoint indicates an expected call of DeleteSinkEndpoint.
func (mr *MockDataFlowMockRecorder) DeleteSinkEndpoint(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteSinkEndpoint", reflect.TypeOf((*MockDataFlow)(nil).DeleteSinkEndpoint), ctx, req)
}

// GetBackup mocks base method.
func (m *MockDataFlow) GetBackup(ctx context.Context, req *dataflow.GetBackupReq) (*dataflow.GetBackupResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetBackup", ctx, req)
	ret0, _ := ret[0].(*dataflow.GetBackupResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetBackup indicates an expected call of GetBackup.
func (mr *MockDataFlowMockRecorder) GetBackup(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBackup", reflect.TypeOf((*MockDataFlow)(nil).GetBackup), ctx, req)
}

// GetConnector mocks base method.
func (m *MockDataFlow) GetConnector(ctx context.Context, req *dataflow.ConnectorID) (*dataflow.Connector, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetConnector", ctx, req)
	ret0, _ := ret[0].(*dataflow.Connector)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetConnector indicates an expected call of GetConnector.
func (mr *MockDataFlowMockRecorder) GetConnector(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConnector", reflect.TypeOf((*MockDataFlow)(nil).GetConnector), ctx, req)
}

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

// GetRestore indicates an expected call of GetRestore.
func (mr *MockDataFlowMockRecorder) GetRestore(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRestore", reflect.TypeOf((*MockDataFlow)(nil).GetRestore), ctx, req)
}

// GetSinkEndpoint mocks base method.
func (m *MockDataFlow) GetSinkEndpoint(ctx context.Context, req *dataflow.GetSinkEndpointRequest) (*dataflow.SinkEndpoint, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetSinkEndpoint", ctx, req)
	ret0, _ := ret[0].(*dataflow.SinkEndpoint)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetSinkEndpoint indicates an expected call of GetSinkEndpoint.
func (mr *MockDataFlowMockRecorder) GetSinkEndpoint(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSinkEndpoint", reflect.TypeOf((*MockDataFlow)(nil).GetSinkEndpoint), ctx, req)
}

// ListChangefeedInstanceTypes mocks base method.
func (m *MockDataFlow) ListChangefeedInstanceTypes(ctx context.Context, req *dataflow.ChangefeedInstanceTypeListReq) (*dataflow.ChangefeedInstanceTypes, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListChangefeedInstanceTypes", ctx, req)
	ret0, _ := ret[0].(*dataflow.ChangefeedInstanceTypes)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListChangefeedInstanceTypes indicates an expected call of ListChangefeedInstanceTypes.
func (mr *MockDataFlowMockRecorder) ListChangefeedInstanceTypes(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListChangefeedInstanceTypes", reflect.TypeOf((*MockDataFlow)(nil).ListChangefeedInstanceTypes), ctx, req)
}

// ListConnectors mocks base method.
func (m *MockDataFlow) ListConnectors(ctx context.Context, req *dataflow.ConnectorListReq) (*dataflow.Connectors, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListConnectors", ctx, req)
	ret0, _ := ret[0].(*dataflow.Connectors)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListConnectors indicates an expected call of ListConnectors.
func (mr *MockDataFlowMockRecorder) ListConnectors(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListConnectors", reflect.TypeOf((*MockDataFlow)(nil).ListConnectors), ctx, req)
}

// ListSinkAvailabilityZones mocks base method.
func (m *MockDataFlow) ListSinkAvailabilityZones(ctx context.Context, req *dataflow.ListSinkAvailabilityZonesRequest) (*dataflow.ListSinkAvailabilityZonesResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListSinkAvailabilityZones", ctx, req)
	ret0, _ := ret[0].(*dataflow.ListSinkAvailabilityZonesResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListSinkAvailabilityZones indicates an expected call of ListSinkAvailabilityZones.
func (mr *MockDataFlowMockRecorder) ListSinkAvailabilityZones(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSinkAvailabilityZones", reflect.TypeOf((*MockDataFlow)(nil).ListSinkAvailabilityZones), ctx, req)
}

// Restore mocks base method.
func (m *MockDataFlow) Restore(ctx context.Context, req *dataflow.BackupInternalRestoreReq) (*dataflow.RestoreResp, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Restore", ctx, req)
	ret0, _ := ret[0].(*dataflow.RestoreResp)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// Restore indicates an expected call of Restore.
func (mr *MockDataFlowMockRecorder) Restore(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Restore", reflect.TypeOf((*MockDataFlow)(nil).Restore), ctx, req)
}