// 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"
	dataflow0 "github.com/tidbcloud/tidb-management-service/nextgen-global-service/domain/adapter/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
}

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

// CleanupClassicRestoreAccess indicates an expected call of CleanupClassicRestoreAccess.
func (mr *MockDataFlowMockRecorder) CleanupClassicRestoreAccess(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanupClassicRestoreAccess", reflect.TypeOf((*MockDataFlow)(nil).CleanupClassicRestoreAccess), 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)
}

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

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

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

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

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

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