146 lines
4.9 KiB
Go
146 lines
4.9 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: github.com/pingcap/tidb/disttask/framework/scheduler/execute (interfaces: SubtaskExecutor,MiniTaskExecutor)
|
|
|
|
// Package execute is a generated GoMock package.
|
|
package execute
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
proto "github.com/pingcap/tidb/disttask/framework/proto"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockSubtaskExecutor is a mock of SubtaskExecutor interface.
|
|
type MockSubtaskExecutor struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockSubtaskExecutorMockRecorder
|
|
}
|
|
|
|
// MockSubtaskExecutorMockRecorder is the mock recorder for MockSubtaskExecutor.
|
|
type MockSubtaskExecutorMockRecorder struct {
|
|
mock *MockSubtaskExecutor
|
|
}
|
|
|
|
// NewMockSubtaskExecutor creates a new mock instance.
|
|
func NewMockSubtaskExecutor(ctrl *gomock.Controller) *MockSubtaskExecutor {
|
|
mock := &MockSubtaskExecutor{ctrl: ctrl}
|
|
mock.recorder = &MockSubtaskExecutorMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockSubtaskExecutor) EXPECT() *MockSubtaskExecutorMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Cleanup mocks base method.
|
|
func (m *MockSubtaskExecutor) Cleanup(arg0 context.Context) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Cleanup", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Cleanup indicates an expected call of Cleanup.
|
|
func (mr *MockSubtaskExecutorMockRecorder) Cleanup(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Cleanup", reflect.TypeOf((*MockSubtaskExecutor)(nil).Cleanup), arg0)
|
|
}
|
|
|
|
// Init mocks base method.
|
|
func (m *MockSubtaskExecutor) Init(arg0 context.Context) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Init", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Init indicates an expected call of Init.
|
|
func (mr *MockSubtaskExecutorMockRecorder) Init(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Init", reflect.TypeOf((*MockSubtaskExecutor)(nil).Init), arg0)
|
|
}
|
|
|
|
// OnFinished mocks base method.
|
|
func (m *MockSubtaskExecutor) OnFinished(arg0 context.Context, arg1 []byte) ([]byte, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnFinished", arg0, arg1)
|
|
ret0, _ := ret[0].([]byte)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// OnFinished indicates an expected call of OnFinished.
|
|
func (mr *MockSubtaskExecutorMockRecorder) OnFinished(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnFinished", reflect.TypeOf((*MockSubtaskExecutor)(nil).OnFinished), arg0, arg1)
|
|
}
|
|
|
|
// Rollback mocks base method.
|
|
func (m *MockSubtaskExecutor) Rollback(arg0 context.Context) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Rollback", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Rollback indicates an expected call of Rollback.
|
|
func (mr *MockSubtaskExecutorMockRecorder) Rollback(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rollback", reflect.TypeOf((*MockSubtaskExecutor)(nil).Rollback), arg0)
|
|
}
|
|
|
|
// SplitSubtask mocks base method.
|
|
func (m *MockSubtaskExecutor) SplitSubtask(arg0 context.Context, arg1 *proto.Subtask) ([]proto.MinimalTask, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SplitSubtask", arg0, arg1)
|
|
ret0, _ := ret[0].([]proto.MinimalTask)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SplitSubtask indicates an expected call of SplitSubtask.
|
|
func (mr *MockSubtaskExecutorMockRecorder) SplitSubtask(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SplitSubtask", reflect.TypeOf((*MockSubtaskExecutor)(nil).SplitSubtask), arg0, arg1)
|
|
}
|
|
|
|
// MockMiniTaskExecutor is a mock of MiniTaskExecutor interface.
|
|
type MockMiniTaskExecutor struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockMiniTaskExecutorMockRecorder
|
|
}
|
|
|
|
// MockMiniTaskExecutorMockRecorder is the mock recorder for MockMiniTaskExecutor.
|
|
type MockMiniTaskExecutorMockRecorder struct {
|
|
mock *MockMiniTaskExecutor
|
|
}
|
|
|
|
// NewMockMiniTaskExecutor creates a new mock instance.
|
|
func NewMockMiniTaskExecutor(ctrl *gomock.Controller) *MockMiniTaskExecutor {
|
|
mock := &MockMiniTaskExecutor{ctrl: ctrl}
|
|
mock.recorder = &MockMiniTaskExecutorMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockMiniTaskExecutor) EXPECT() *MockMiniTaskExecutorMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Run mocks base method.
|
|
func (m *MockMiniTaskExecutor) Run(arg0 context.Context) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Run", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Run indicates an expected call of Run.
|
|
func (mr *MockMiniTaskExecutorMockRecorder) Run(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Run", reflect.TypeOf((*MockMiniTaskExecutor)(nil).Run), arg0)
|
|
}
|