common/persistence/data_store_interfaces_mock.go (346 lines of code) (raw):

// The MIT License (MIT) // Copyright (c) 2017-2020 Uber Technologies Inc. // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to deal // in the Software without restriction, including without limitation the rights // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell // copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // // The above copyright notice and this permission notice shall be included in all // copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE // SOFTWARE. // Code generated by MockGen. DO NOT EDIT. // Source: github.com/uber/cadence/common/persistence (interfaces: ExecutionStore) // Package persistence is a generated GoMock package. package persistence import ( context "context" reflect "reflect" gomock "github.com/golang/mock/gomock" ) // MockExecutionStore is a mock of ExecutionStore interface. type MockExecutionStore struct { ctrl *gomock.Controller recorder *MockExecutionStoreMockRecorder } // MockExecutionStoreMockRecorder is the mock recorder for MockExecutionStore. type MockExecutionStoreMockRecorder struct { mock *MockExecutionStore } // NewMockExecutionStore creates a new mock instance. func NewMockExecutionStore(ctrl *gomock.Controller) *MockExecutionStore { mock := &MockExecutionStore{ctrl: ctrl} mock.recorder = &MockExecutionStoreMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockExecutionStore) EXPECT() *MockExecutionStoreMockRecorder { return m.recorder } // Close mocks base method. func (m *MockExecutionStore) Close() { m.ctrl.T.Helper() m.ctrl.Call(m, "Close") } // Close indicates an expected call of Close. func (mr *MockExecutionStoreMockRecorder) Close() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockExecutionStore)(nil).Close)) } // CompleteCrossClusterTask mocks base method. func (m *MockExecutionStore) CompleteCrossClusterTask(arg0 context.Context, arg1 *CompleteCrossClusterTaskRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CompleteCrossClusterTask", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // CompleteCrossClusterTask indicates an expected call of CompleteCrossClusterTask. func (mr *MockExecutionStoreMockRecorder) CompleteCrossClusterTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CompleteCrossClusterTask", reflect.TypeOf((*MockExecutionStore)(nil).CompleteCrossClusterTask), arg0, arg1) } // CompleteReplicationTask mocks base method. func (m *MockExecutionStore) CompleteReplicationTask(arg0 context.Context, arg1 *CompleteReplicationTaskRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CompleteReplicationTask", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // CompleteReplicationTask indicates an expected call of CompleteReplicationTask. func (mr *MockExecutionStoreMockRecorder) CompleteReplicationTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CompleteReplicationTask", reflect.TypeOf((*MockExecutionStore)(nil).CompleteReplicationTask), arg0, arg1) } // CompleteTimerTask mocks base method. func (m *MockExecutionStore) CompleteTimerTask(arg0 context.Context, arg1 *CompleteTimerTaskRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CompleteTimerTask", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // CompleteTimerTask indicates an expected call of CompleteTimerTask. func (mr *MockExecutionStoreMockRecorder) CompleteTimerTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CompleteTimerTask", reflect.TypeOf((*MockExecutionStore)(nil).CompleteTimerTask), arg0, arg1) } // CompleteTransferTask mocks base method. func (m *MockExecutionStore) CompleteTransferTask(arg0 context.Context, arg1 *CompleteTransferTaskRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CompleteTransferTask", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // CompleteTransferTask indicates an expected call of CompleteTransferTask. func (mr *MockExecutionStoreMockRecorder) CompleteTransferTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CompleteTransferTask", reflect.TypeOf((*MockExecutionStore)(nil).CompleteTransferTask), arg0, arg1) } // ConflictResolveWorkflowExecution mocks base method. func (m *MockExecutionStore) ConflictResolveWorkflowExecution(arg0 context.Context, arg1 *InternalConflictResolveWorkflowExecutionRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ConflictResolveWorkflowExecution", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // ConflictResolveWorkflowExecution indicates an expected call of ConflictResolveWorkflowExecution. func (mr *MockExecutionStoreMockRecorder) ConflictResolveWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConflictResolveWorkflowExecution", reflect.TypeOf((*MockExecutionStore)(nil).ConflictResolveWorkflowExecution), arg0, arg1) } // CreateFailoverMarkerTasks mocks base method. func (m *MockExecutionStore) CreateFailoverMarkerTasks(arg0 context.Context, arg1 *CreateFailoverMarkersRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CreateFailoverMarkerTasks", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // CreateFailoverMarkerTasks indicates an expected call of CreateFailoverMarkerTasks. func (mr *MockExecutionStoreMockRecorder) CreateFailoverMarkerTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateFailoverMarkerTasks", reflect.TypeOf((*MockExecutionStore)(nil).CreateFailoverMarkerTasks), arg0, arg1) } // CreateWorkflowExecution mocks base method. func (m *MockExecutionStore) CreateWorkflowExecution(arg0 context.Context, arg1 *InternalCreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CreateWorkflowExecution", arg0, arg1) ret0, _ := ret[0].(*CreateWorkflowExecutionResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateWorkflowExecution indicates an expected call of CreateWorkflowExecution. func (mr *MockExecutionStoreMockRecorder) CreateWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateWorkflowExecution", reflect.TypeOf((*MockExecutionStore)(nil).CreateWorkflowExecution), arg0, arg1) } // DeleteCurrentWorkflowExecution mocks base method. func (m *MockExecutionStore) DeleteCurrentWorkflowExecution(arg0 context.Context, arg1 *DeleteCurrentWorkflowExecutionRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteCurrentWorkflowExecution", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // DeleteCurrentWorkflowExecution indicates an expected call of DeleteCurrentWorkflowExecution. func (mr *MockExecutionStoreMockRecorder) DeleteCurrentWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentWorkflowExecution", reflect.TypeOf((*MockExecutionStore)(nil).DeleteCurrentWorkflowExecution), arg0, arg1) } // DeleteReplicationTaskFromDLQ mocks base method. func (m *MockExecutionStore) DeleteReplicationTaskFromDLQ(arg0 context.Context, arg1 *DeleteReplicationTaskFromDLQRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteReplicationTaskFromDLQ", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // DeleteReplicationTaskFromDLQ indicates an expected call of DeleteReplicationTaskFromDLQ. func (mr *MockExecutionStoreMockRecorder) DeleteReplicationTaskFromDLQ(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteReplicationTaskFromDLQ", reflect.TypeOf((*MockExecutionStore)(nil).DeleteReplicationTaskFromDLQ), arg0, arg1) } // DeleteWorkflowExecution mocks base method. func (m *MockExecutionStore) DeleteWorkflowExecution(arg0 context.Context, arg1 *DeleteWorkflowExecutionRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteWorkflowExecution", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution. func (mr *MockExecutionStoreMockRecorder) DeleteWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorkflowExecution", reflect.TypeOf((*MockExecutionStore)(nil).DeleteWorkflowExecution), arg0, arg1) } // GetCrossClusterTasks mocks base method. func (m *MockExecutionStore) GetCrossClusterTasks(arg0 context.Context, arg1 *GetCrossClusterTasksRequest) (*GetCrossClusterTasksResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetCrossClusterTasks", arg0, arg1) ret0, _ := ret[0].(*GetCrossClusterTasksResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetCrossClusterTasks indicates an expected call of GetCrossClusterTasks. func (mr *MockExecutionStoreMockRecorder) GetCrossClusterTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCrossClusterTasks", reflect.TypeOf((*MockExecutionStore)(nil).GetCrossClusterTasks), arg0, arg1) } // GetCurrentExecution mocks base method. func (m *MockExecutionStore) GetCurrentExecution(arg0 context.Context, arg1 *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetCurrentExecution", arg0, arg1) ret0, _ := ret[0].(*GetCurrentExecutionResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetCurrentExecution indicates an expected call of GetCurrentExecution. func (mr *MockExecutionStoreMockRecorder) GetCurrentExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentExecution", reflect.TypeOf((*MockExecutionStore)(nil).GetCurrentExecution), arg0, arg1) } // GetName mocks base method. func (m *MockExecutionStore) GetName() string { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetName") ret0, _ := ret[0].(string) return ret0 } // GetName indicates an expected call of GetName. func (mr *MockExecutionStoreMockRecorder) GetName() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetName", reflect.TypeOf((*MockExecutionStore)(nil).GetName)) } // GetReplicationDLQSize mocks base method. func (m *MockExecutionStore) GetReplicationDLQSize(arg0 context.Context, arg1 *GetReplicationDLQSizeRequest) (*GetReplicationDLQSizeResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetReplicationDLQSize", arg0, arg1) ret0, _ := ret[0].(*GetReplicationDLQSizeResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetReplicationDLQSize indicates an expected call of GetReplicationDLQSize. func (mr *MockExecutionStoreMockRecorder) GetReplicationDLQSize(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicationDLQSize", reflect.TypeOf((*MockExecutionStore)(nil).GetReplicationDLQSize), arg0, arg1) } // GetReplicationTasks mocks base method. func (m *MockExecutionStore) GetReplicationTasks(arg0 context.Context, arg1 *GetReplicationTasksRequest) (*InternalGetReplicationTasksResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetReplicationTasks", arg0, arg1) ret0, _ := ret[0].(*InternalGetReplicationTasksResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetReplicationTasks indicates an expected call of GetReplicationTasks. func (mr *MockExecutionStoreMockRecorder) GetReplicationTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicationTasks", reflect.TypeOf((*MockExecutionStore)(nil).GetReplicationTasks), arg0, arg1) } // GetReplicationTasksFromDLQ mocks base method. func (m *MockExecutionStore) GetReplicationTasksFromDLQ(arg0 context.Context, arg1 *GetReplicationTasksFromDLQRequest) (*InternalGetReplicationTasksResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetReplicationTasksFromDLQ", arg0, arg1) ret0, _ := ret[0].(*InternalGetReplicationTasksResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetReplicationTasksFromDLQ indicates an expected call of GetReplicationTasksFromDLQ. func (mr *MockExecutionStoreMockRecorder) GetReplicationTasksFromDLQ(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicationTasksFromDLQ", reflect.TypeOf((*MockExecutionStore)(nil).GetReplicationTasksFromDLQ), arg0, arg1) } // GetShardID mocks base method. func (m *MockExecutionStore) GetShardID() int { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetShardID") ret0, _ := ret[0].(int) return ret0 } // GetShardID indicates an expected call of GetShardID. func (mr *MockExecutionStoreMockRecorder) GetShardID() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetShardID", reflect.TypeOf((*MockExecutionStore)(nil).GetShardID)) } // GetTimerIndexTasks mocks base method. func (m *MockExecutionStore) GetTimerIndexTasks(arg0 context.Context, arg1 *GetTimerIndexTasksRequest) (*GetTimerIndexTasksResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetTimerIndexTasks", arg0, arg1) ret0, _ := ret[0].(*GetTimerIndexTasksResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTimerIndexTasks indicates an expected call of GetTimerIndexTasks. func (mr *MockExecutionStoreMockRecorder) GetTimerIndexTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimerIndexTasks", reflect.TypeOf((*MockExecutionStore)(nil).GetTimerIndexTasks), arg0, arg1) } // GetTransferTasks mocks base method. func (m *MockExecutionStore) GetTransferTasks(arg0 context.Context, arg1 *GetTransferTasksRequest) (*GetTransferTasksResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetTransferTasks", arg0, arg1) ret0, _ := ret[0].(*GetTransferTasksResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTransferTasks indicates an expected call of GetTransferTasks. func (mr *MockExecutionStoreMockRecorder) GetTransferTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTransferTasks", reflect.TypeOf((*MockExecutionStore)(nil).GetTransferTasks), arg0, arg1) } // GetWorkflowExecution mocks base method. func (m *MockExecutionStore) GetWorkflowExecution(arg0 context.Context, arg1 *InternalGetWorkflowExecutionRequest) (*InternalGetWorkflowExecutionResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetWorkflowExecution", arg0, arg1) ret0, _ := ret[0].(*InternalGetWorkflowExecutionResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkflowExecution indicates an expected call of GetWorkflowExecution. func (mr *MockExecutionStoreMockRecorder) GetWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkflowExecution", reflect.TypeOf((*MockExecutionStore)(nil).GetWorkflowExecution), arg0, arg1) } // IsWorkflowExecutionExists mocks base method. func (m *MockExecutionStore) IsWorkflowExecutionExists(arg0 context.Context, arg1 *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "IsWorkflowExecutionExists", arg0, arg1) ret0, _ := ret[0].(*IsWorkflowExecutionExistsResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists. func (mr *MockExecutionStoreMockRecorder) IsWorkflowExecutionExists(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsWorkflowExecutionExists", reflect.TypeOf((*MockExecutionStore)(nil).IsWorkflowExecutionExists), arg0, arg1) } // ListConcreteExecutions mocks base method. func (m *MockExecutionStore) ListConcreteExecutions(arg0 context.Context, arg1 *ListConcreteExecutionsRequest) (*InternalListConcreteExecutionsResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ListConcreteExecutions", arg0, arg1) ret0, _ := ret[0].(*InternalListConcreteExecutionsResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListConcreteExecutions indicates an expected call of ListConcreteExecutions. func (mr *MockExecutionStoreMockRecorder) ListConcreteExecutions(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListConcreteExecutions", reflect.TypeOf((*MockExecutionStore)(nil).ListConcreteExecutions), arg0, arg1) } // ListCurrentExecutions mocks base method. func (m *MockExecutionStore) ListCurrentExecutions(arg0 context.Context, arg1 *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ListCurrentExecutions", arg0, arg1) ret0, _ := ret[0].(*ListCurrentExecutionsResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListCurrentExecutions indicates an expected call of ListCurrentExecutions. func (mr *MockExecutionStoreMockRecorder) ListCurrentExecutions(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListCurrentExecutions", reflect.TypeOf((*MockExecutionStore)(nil).ListCurrentExecutions), arg0, arg1) } // PutReplicationTaskToDLQ mocks base method. func (m *MockExecutionStore) PutReplicationTaskToDLQ(arg0 context.Context, arg1 *InternalPutReplicationTaskToDLQRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "PutReplicationTaskToDLQ", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // PutReplicationTaskToDLQ indicates an expected call of PutReplicationTaskToDLQ. func (mr *MockExecutionStoreMockRecorder) PutReplicationTaskToDLQ(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutReplicationTaskToDLQ", reflect.TypeOf((*MockExecutionStore)(nil).PutReplicationTaskToDLQ), arg0, arg1) } // RangeCompleteCrossClusterTask mocks base method. func (m *MockExecutionStore) RangeCompleteCrossClusterTask(arg0 context.Context, arg1 *RangeCompleteCrossClusterTaskRequest) (*RangeCompleteCrossClusterTaskResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RangeCompleteCrossClusterTask", arg0, arg1) ret0, _ := ret[0].(*RangeCompleteCrossClusterTaskResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // RangeCompleteCrossClusterTask indicates an expected call of RangeCompleteCrossClusterTask. func (mr *MockExecutionStoreMockRecorder) RangeCompleteCrossClusterTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RangeCompleteCrossClusterTask", reflect.TypeOf((*MockExecutionStore)(nil).RangeCompleteCrossClusterTask), arg0, arg1) } // RangeCompleteReplicationTask mocks base method. func (m *MockExecutionStore) RangeCompleteReplicationTask(arg0 context.Context, arg1 *RangeCompleteReplicationTaskRequest) (*RangeCompleteReplicationTaskResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RangeCompleteReplicationTask", arg0, arg1) ret0, _ := ret[0].(*RangeCompleteReplicationTaskResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // RangeCompleteReplicationTask indicates an expected call of RangeCompleteReplicationTask. func (mr *MockExecutionStoreMockRecorder) RangeCompleteReplicationTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RangeCompleteReplicationTask", reflect.TypeOf((*MockExecutionStore)(nil).RangeCompleteReplicationTask), arg0, arg1) } // RangeCompleteTimerTask mocks base method. func (m *MockExecutionStore) RangeCompleteTimerTask(arg0 context.Context, arg1 *RangeCompleteTimerTaskRequest) (*RangeCompleteTimerTaskResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RangeCompleteTimerTask", arg0, arg1) ret0, _ := ret[0].(*RangeCompleteTimerTaskResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // RangeCompleteTimerTask indicates an expected call of RangeCompleteTimerTask. func (mr *MockExecutionStoreMockRecorder) RangeCompleteTimerTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RangeCompleteTimerTask", reflect.TypeOf((*MockExecutionStore)(nil).RangeCompleteTimerTask), arg0, arg1) } // RangeCompleteTransferTask mocks base method. func (m *MockExecutionStore) RangeCompleteTransferTask(arg0 context.Context, arg1 *RangeCompleteTransferTaskRequest) (*RangeCompleteTransferTaskResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RangeCompleteTransferTask", arg0, arg1) ret0, _ := ret[0].(*RangeCompleteTransferTaskResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // RangeCompleteTransferTask indicates an expected call of RangeCompleteTransferTask. func (mr *MockExecutionStoreMockRecorder) RangeCompleteTransferTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RangeCompleteTransferTask", reflect.TypeOf((*MockExecutionStore)(nil).RangeCompleteTransferTask), arg0, arg1) } // RangeDeleteReplicationTaskFromDLQ mocks base method. func (m *MockExecutionStore) RangeDeleteReplicationTaskFromDLQ(arg0 context.Context, arg1 *RangeDeleteReplicationTaskFromDLQRequest) (*RangeDeleteReplicationTaskFromDLQResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RangeDeleteReplicationTaskFromDLQ", arg0, arg1) ret0, _ := ret[0].(*RangeDeleteReplicationTaskFromDLQResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // RangeDeleteReplicationTaskFromDLQ indicates an expected call of RangeDeleteReplicationTaskFromDLQ. func (mr *MockExecutionStoreMockRecorder) RangeDeleteReplicationTaskFromDLQ(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RangeDeleteReplicationTaskFromDLQ", reflect.TypeOf((*MockExecutionStore)(nil).RangeDeleteReplicationTaskFromDLQ), arg0, arg1) } // UpdateWorkflowExecution mocks base method. func (m *MockExecutionStore) UpdateWorkflowExecution(arg0 context.Context, arg1 *InternalUpdateWorkflowExecutionRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateWorkflowExecution", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkflowExecution indicates an expected call of UpdateWorkflowExecution. func (mr *MockExecutionStoreMockRecorder) UpdateWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkflowExecution", reflect.TypeOf((*MockExecutionStore)(nil).UpdateWorkflowExecution), arg0, arg1) }