github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/engine/enginepb/mock/executor_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/pingcap/tiflow/engine/enginepb (interfaces: ExecutorServiceClient)
     3  
     4  // Package mock is a generated GoMock package.
     5  package mock
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	enginepb "github.com/pingcap/tiflow/engine/enginepb"
    13  	grpc "google.golang.org/grpc"
    14  )
    15  
    16  // MockExecutorServiceClient is a mock of ExecutorServiceClient interface.
    17  type MockExecutorServiceClient struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockExecutorServiceClientMockRecorder
    20  }
    21  
    22  // MockExecutorServiceClientMockRecorder is the mock recorder for MockExecutorServiceClient.
    23  type MockExecutorServiceClientMockRecorder struct {
    24  	mock *MockExecutorServiceClient
    25  }
    26  
    27  // NewMockExecutorServiceClient creates a new mock instance.
    28  func NewMockExecutorServiceClient(ctrl *gomock.Controller) *MockExecutorServiceClient {
    29  	mock := &MockExecutorServiceClient{ctrl: ctrl}
    30  	mock.recorder = &MockExecutorServiceClientMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use.
    35  func (m *MockExecutorServiceClient) EXPECT() *MockExecutorServiceClientMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // ConfirmDispatchTask mocks base method.
    40  func (m *MockExecutorServiceClient) ConfirmDispatchTask(arg0 context.Context, arg1 *enginepb.ConfirmDispatchTaskRequest, arg2 ...grpc.CallOption) (*enginepb.ConfirmDispatchTaskResponse, error) {
    41  	m.ctrl.T.Helper()
    42  	varargs := []interface{}{arg0, arg1}
    43  	for _, a := range arg2 {
    44  		varargs = append(varargs, a)
    45  	}
    46  	ret := m.ctrl.Call(m, "ConfirmDispatchTask", varargs...)
    47  	ret0, _ := ret[0].(*enginepb.ConfirmDispatchTaskResponse)
    48  	ret1, _ := ret[1].(error)
    49  	return ret0, ret1
    50  }
    51  
    52  // ConfirmDispatchTask indicates an expected call of ConfirmDispatchTask.
    53  func (mr *MockExecutorServiceClientMockRecorder) ConfirmDispatchTask(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    54  	mr.mock.ctrl.T.Helper()
    55  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    56  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConfirmDispatchTask", reflect.TypeOf((*MockExecutorServiceClient)(nil).ConfirmDispatchTask), varargs...)
    57  }
    58  
    59  // PreDispatchTask mocks base method.
    60  func (m *MockExecutorServiceClient) PreDispatchTask(arg0 context.Context, arg1 *enginepb.PreDispatchTaskRequest, arg2 ...grpc.CallOption) (*enginepb.PreDispatchTaskResponse, error) {
    61  	m.ctrl.T.Helper()
    62  	varargs := []interface{}{arg0, arg1}
    63  	for _, a := range arg2 {
    64  		varargs = append(varargs, a)
    65  	}
    66  	ret := m.ctrl.Call(m, "PreDispatchTask", varargs...)
    67  	ret0, _ := ret[0].(*enginepb.PreDispatchTaskResponse)
    68  	ret1, _ := ret[1].(error)
    69  	return ret0, ret1
    70  }
    71  
    72  // PreDispatchTask indicates an expected call of PreDispatchTask.
    73  func (mr *MockExecutorServiceClientMockRecorder) PreDispatchTask(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    74  	mr.mock.ctrl.T.Helper()
    75  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    76  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreDispatchTask", reflect.TypeOf((*MockExecutorServiceClient)(nil).PreDispatchTask), varargs...)
    77  }