368 lines
13 KiB
Go
368 lines
13 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: github.com/pingcap/tidb/pkg/lightning/backend (interfaces: Backend,EngineWriter,TargetInfoGetter,ChunkFlushStatus)
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -package mock github.com/pingcap/tidb/pkg/lightning/backend Backend,EngineWriter,TargetInfoGetter,ChunkFlushStatus
|
|
//
|
|
// Package mock is a generated GoMock package.
|
|
package mock
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
uuid "github.com/google/uuid"
|
|
backend "github.com/pingcap/tidb/pkg/lightning/backend"
|
|
encode "github.com/pingcap/tidb/pkg/lightning/backend/encode"
|
|
model "github.com/pingcap/tidb/pkg/meta/model"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockBackend is a mock of Backend interface.
|
|
type MockBackend struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockBackendMockRecorder
|
|
}
|
|
|
|
// MockBackendMockRecorder is the mock recorder for MockBackend.
|
|
type MockBackendMockRecorder struct {
|
|
mock *MockBackend
|
|
}
|
|
|
|
// NewMockBackend creates a new mock instance.
|
|
func NewMockBackend(ctrl *gomock.Controller) *MockBackend {
|
|
mock := &MockBackend{ctrl: ctrl}
|
|
mock.recorder = &MockBackendMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockBackend) EXPECT() *MockBackendMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CleanupEngine mocks base method.
|
|
func (m *MockBackend) CleanupEngine(arg0 context.Context, arg1 uuid.UUID) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CleanupEngine", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// CleanupEngine indicates an expected call of CleanupEngine.
|
|
func (mr *MockBackendMockRecorder) CleanupEngine(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanupEngine", reflect.TypeOf((*MockBackend)(nil).CleanupEngine), arg0, arg1)
|
|
}
|
|
|
|
// Close mocks base method.
|
|
func (m *MockBackend) Close() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Close")
|
|
}
|
|
|
|
// Close indicates an expected call of Close.
|
|
func (mr *MockBackendMockRecorder) Close() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockBackend)(nil).Close))
|
|
}
|
|
|
|
// CloseEngine mocks base method.
|
|
func (m *MockBackend) CloseEngine(arg0 context.Context, arg1 *backend.EngineConfig, arg2 uuid.UUID) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CloseEngine", arg0, arg1, arg2)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// CloseEngine indicates an expected call of CloseEngine.
|
|
func (mr *MockBackendMockRecorder) CloseEngine(arg0, arg1, arg2 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseEngine", reflect.TypeOf((*MockBackend)(nil).CloseEngine), arg0, arg1, arg2)
|
|
}
|
|
|
|
// FlushAllEngines mocks base method.
|
|
func (m *MockBackend) FlushAllEngines(arg0 context.Context) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FlushAllEngines", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// FlushAllEngines indicates an expected call of FlushAllEngines.
|
|
func (mr *MockBackendMockRecorder) FlushAllEngines(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FlushAllEngines", reflect.TypeOf((*MockBackend)(nil).FlushAllEngines), arg0)
|
|
}
|
|
|
|
// FlushEngine mocks base method.
|
|
func (m *MockBackend) FlushEngine(arg0 context.Context, arg1 uuid.UUID) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FlushEngine", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// FlushEngine indicates an expected call of FlushEngine.
|
|
func (mr *MockBackendMockRecorder) FlushEngine(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FlushEngine", reflect.TypeOf((*MockBackend)(nil).FlushEngine), arg0, arg1)
|
|
}
|
|
|
|
// ImportEngine mocks base method.
|
|
func (m *MockBackend) ImportEngine(arg0 context.Context, arg1 uuid.UUID, arg2, arg3 int64) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ImportEngine", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// ImportEngine indicates an expected call of ImportEngine.
|
|
func (mr *MockBackendMockRecorder) ImportEngine(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportEngine", reflect.TypeOf((*MockBackend)(nil).ImportEngine), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// LocalWriter mocks base method.
|
|
func (m *MockBackend) LocalWriter(arg0 context.Context, arg1 *backend.LocalWriterConfig, arg2 uuid.UUID) (backend.EngineWriter, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "LocalWriter", arg0, arg1, arg2)
|
|
ret0, _ := ret[0].(backend.EngineWriter)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// LocalWriter indicates an expected call of LocalWriter.
|
|
func (mr *MockBackendMockRecorder) LocalWriter(arg0, arg1, arg2 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LocalWriter", reflect.TypeOf((*MockBackend)(nil).LocalWriter), arg0, arg1, arg2)
|
|
}
|
|
|
|
// OpenEngine mocks base method.
|
|
func (m *MockBackend) OpenEngine(arg0 context.Context, arg1 *backend.EngineConfig, arg2 uuid.UUID) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OpenEngine", arg0, arg1, arg2)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// OpenEngine indicates an expected call of OpenEngine.
|
|
func (mr *MockBackendMockRecorder) OpenEngine(arg0, arg1, arg2 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenEngine", reflect.TypeOf((*MockBackend)(nil).OpenEngine), arg0, arg1, arg2)
|
|
}
|
|
|
|
// ResetEngine mocks base method.
|
|
func (m *MockBackend) ResetEngine(arg0 context.Context, arg1 uuid.UUID) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ResetEngine", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// ResetEngine indicates an expected call of ResetEngine.
|
|
func (mr *MockBackendMockRecorder) ResetEngine(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetEngine", reflect.TypeOf((*MockBackend)(nil).ResetEngine), arg0, arg1)
|
|
}
|
|
|
|
// RetryImportDelay mocks base method.
|
|
func (m *MockBackend) RetryImportDelay() time.Duration {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RetryImportDelay")
|
|
ret0, _ := ret[0].(time.Duration)
|
|
return ret0
|
|
}
|
|
|
|
// RetryImportDelay indicates an expected call of RetryImportDelay.
|
|
func (mr *MockBackendMockRecorder) RetryImportDelay() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RetryImportDelay", reflect.TypeOf((*MockBackend)(nil).RetryImportDelay))
|
|
}
|
|
|
|
// ShouldPostProcess mocks base method.
|
|
func (m *MockBackend) ShouldPostProcess() bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ShouldPostProcess")
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// ShouldPostProcess indicates an expected call of ShouldPostProcess.
|
|
func (mr *MockBackendMockRecorder) ShouldPostProcess() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ShouldPostProcess", reflect.TypeOf((*MockBackend)(nil).ShouldPostProcess))
|
|
}
|
|
|
|
// MockEngineWriter is a mock of EngineWriter interface.
|
|
type MockEngineWriter struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockEngineWriterMockRecorder
|
|
}
|
|
|
|
// MockEngineWriterMockRecorder is the mock recorder for MockEngineWriter.
|
|
type MockEngineWriterMockRecorder struct {
|
|
mock *MockEngineWriter
|
|
}
|
|
|
|
// NewMockEngineWriter creates a new mock instance.
|
|
func NewMockEngineWriter(ctrl *gomock.Controller) *MockEngineWriter {
|
|
mock := &MockEngineWriter{ctrl: ctrl}
|
|
mock.recorder = &MockEngineWriterMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockEngineWriter) EXPECT() *MockEngineWriterMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// AppendRows mocks base method.
|
|
func (m *MockEngineWriter) AppendRows(arg0 context.Context, arg1 []string, arg2 encode.Rows) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "AppendRows", arg0, arg1, arg2)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// AppendRows indicates an expected call of AppendRows.
|
|
func (mr *MockEngineWriterMockRecorder) AppendRows(arg0, arg1, arg2 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendRows", reflect.TypeOf((*MockEngineWriter)(nil).AppendRows), arg0, arg1, arg2)
|
|
}
|
|
|
|
// Close mocks base method.
|
|
func (m *MockEngineWriter) Close(arg0 context.Context) (backend.ChunkFlushStatus, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Close", arg0)
|
|
ret0, _ := ret[0].(backend.ChunkFlushStatus)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Close indicates an expected call of Close.
|
|
func (mr *MockEngineWriterMockRecorder) Close(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockEngineWriter)(nil).Close), arg0)
|
|
}
|
|
|
|
// IsSynced mocks base method.
|
|
func (m *MockEngineWriter) IsSynced() bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "IsSynced")
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// IsSynced indicates an expected call of IsSynced.
|
|
func (mr *MockEngineWriterMockRecorder) IsSynced() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsSynced", reflect.TypeOf((*MockEngineWriter)(nil).IsSynced))
|
|
}
|
|
|
|
// MockTargetInfoGetter is a mock of TargetInfoGetter interface.
|
|
type MockTargetInfoGetter struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockTargetInfoGetterMockRecorder
|
|
}
|
|
|
|
// MockTargetInfoGetterMockRecorder is the mock recorder for MockTargetInfoGetter.
|
|
type MockTargetInfoGetterMockRecorder struct {
|
|
mock *MockTargetInfoGetter
|
|
}
|
|
|
|
// NewMockTargetInfoGetter creates a new mock instance.
|
|
func NewMockTargetInfoGetter(ctrl *gomock.Controller) *MockTargetInfoGetter {
|
|
mock := &MockTargetInfoGetter{ctrl: ctrl}
|
|
mock.recorder = &MockTargetInfoGetterMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockTargetInfoGetter) EXPECT() *MockTargetInfoGetterMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CheckRequirements mocks base method.
|
|
func (m *MockTargetInfoGetter) CheckRequirements(arg0 context.Context, arg1 *backend.CheckCtx) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CheckRequirements", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// CheckRequirements indicates an expected call of CheckRequirements.
|
|
func (mr *MockTargetInfoGetterMockRecorder) CheckRequirements(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckRequirements", reflect.TypeOf((*MockTargetInfoGetter)(nil).CheckRequirements), arg0, arg1)
|
|
}
|
|
|
|
// FetchRemoteDBModels mocks base method.
|
|
func (m *MockTargetInfoGetter) FetchRemoteDBModels(arg0 context.Context) ([]*model.DBInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FetchRemoteDBModels", arg0)
|
|
ret0, _ := ret[0].([]*model.DBInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// FetchRemoteDBModels indicates an expected call of FetchRemoteDBModels.
|
|
func (mr *MockTargetInfoGetterMockRecorder) FetchRemoteDBModels(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchRemoteDBModels", reflect.TypeOf((*MockTargetInfoGetter)(nil).FetchRemoteDBModels), arg0)
|
|
}
|
|
|
|
// FetchRemoteTableModels mocks base method.
|
|
func (m *MockTargetInfoGetter) FetchRemoteTableModels(arg0 context.Context, arg1 string) ([]*model.TableInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FetchRemoteTableModels", arg0, arg1)
|
|
ret0, _ := ret[0].([]*model.TableInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// FetchRemoteTableModels indicates an expected call of FetchRemoteTableModels.
|
|
func (mr *MockTargetInfoGetterMockRecorder) FetchRemoteTableModels(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchRemoteTableModels", reflect.TypeOf((*MockTargetInfoGetter)(nil).FetchRemoteTableModels), arg0, arg1)
|
|
}
|
|
|
|
// MockChunkFlushStatus is a mock of ChunkFlushStatus interface.
|
|
type MockChunkFlushStatus struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockChunkFlushStatusMockRecorder
|
|
}
|
|
|
|
// MockChunkFlushStatusMockRecorder is the mock recorder for MockChunkFlushStatus.
|
|
type MockChunkFlushStatusMockRecorder struct {
|
|
mock *MockChunkFlushStatus
|
|
}
|
|
|
|
// NewMockChunkFlushStatus creates a new mock instance.
|
|
func NewMockChunkFlushStatus(ctrl *gomock.Controller) *MockChunkFlushStatus {
|
|
mock := &MockChunkFlushStatus{ctrl: ctrl}
|
|
mock.recorder = &MockChunkFlushStatusMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockChunkFlushStatus) EXPECT() *MockChunkFlushStatusMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Flushed mocks base method.
|
|
func (m *MockChunkFlushStatus) Flushed() bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Flushed")
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// Flushed indicates an expected call of Flushed.
|
|
func (mr *MockChunkFlushStatusMockRecorder) Flushed() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Flushed", reflect.TypeOf((*MockChunkFlushStatus)(nil).Flushed))
|
|
}
|