service/history/queue/interface_mock.go (360 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: interface.go // Package queue is a generated GoMock package. package queue import ( context "context" reflect "reflect" gomock "github.com/golang/mock/gomock" common "github.com/uber/cadence/service/history/common" task "github.com/uber/cadence/service/history/task" ) // MockProcessingQueueState is a mock of ProcessingQueueState interface. type MockProcessingQueueState struct { ctrl *gomock.Controller recorder *MockProcessingQueueStateMockRecorder } // MockProcessingQueueStateMockRecorder is the mock recorder for MockProcessingQueueState. type MockProcessingQueueStateMockRecorder struct { mock *MockProcessingQueueState } // NewMockProcessingQueueState creates a new mock instance. func NewMockProcessingQueueState(ctrl *gomock.Controller) *MockProcessingQueueState { mock := &MockProcessingQueueState{ctrl: ctrl} mock.recorder = &MockProcessingQueueStateMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockProcessingQueueState) EXPECT() *MockProcessingQueueStateMockRecorder { return m.recorder } // AckLevel mocks base method. func (m *MockProcessingQueueState) AckLevel() task.Key { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "AckLevel") ret0, _ := ret[0].(task.Key) return ret0 } // AckLevel indicates an expected call of AckLevel. func (mr *MockProcessingQueueStateMockRecorder) AckLevel() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AckLevel", reflect.TypeOf((*MockProcessingQueueState)(nil).AckLevel)) } // DomainFilter mocks base method. func (m *MockProcessingQueueState) DomainFilter() DomainFilter { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DomainFilter") ret0, _ := ret[0].(DomainFilter) return ret0 } // DomainFilter indicates an expected call of DomainFilter. func (mr *MockProcessingQueueStateMockRecorder) DomainFilter() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DomainFilter", reflect.TypeOf((*MockProcessingQueueState)(nil).DomainFilter)) } // Level mocks base method. func (m *MockProcessingQueueState) Level() int { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Level") ret0, _ := ret[0].(int) return ret0 } // Level indicates an expected call of Level. func (mr *MockProcessingQueueStateMockRecorder) Level() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Level", reflect.TypeOf((*MockProcessingQueueState)(nil).Level)) } // MaxLevel mocks base method. func (m *MockProcessingQueueState) MaxLevel() task.Key { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "MaxLevel") ret0, _ := ret[0].(task.Key) return ret0 } // MaxLevel indicates an expected call of MaxLevel. func (mr *MockProcessingQueueStateMockRecorder) MaxLevel() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MaxLevel", reflect.TypeOf((*MockProcessingQueueState)(nil).MaxLevel)) } // ReadLevel mocks base method. func (m *MockProcessingQueueState) ReadLevel() task.Key { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadLevel") ret0, _ := ret[0].(task.Key) return ret0 } // ReadLevel indicates an expected call of ReadLevel. func (mr *MockProcessingQueueStateMockRecorder) ReadLevel() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadLevel", reflect.TypeOf((*MockProcessingQueueState)(nil).ReadLevel)) } // MockProcessingQueue is a mock of ProcessingQueue interface. type MockProcessingQueue struct { ctrl *gomock.Controller recorder *MockProcessingQueueMockRecorder } // MockProcessingQueueMockRecorder is the mock recorder for MockProcessingQueue. type MockProcessingQueueMockRecorder struct { mock *MockProcessingQueue } // NewMockProcessingQueue creates a new mock instance. func NewMockProcessingQueue(ctrl *gomock.Controller) *MockProcessingQueue { mock := &MockProcessingQueue{ctrl: ctrl} mock.recorder = &MockProcessingQueueMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockProcessingQueue) EXPECT() *MockProcessingQueueMockRecorder { return m.recorder } // AddTasks mocks base method. func (m *MockProcessingQueue) AddTasks(arg0 map[task.Key]task.Task, arg1 task.Key) { m.ctrl.T.Helper() m.ctrl.Call(m, "AddTasks", arg0, arg1) } // AddTasks indicates an expected call of AddTasks. func (mr *MockProcessingQueueMockRecorder) AddTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTasks", reflect.TypeOf((*MockProcessingQueue)(nil).AddTasks), arg0, arg1) } // GetTask mocks base method. func (m *MockProcessingQueue) GetTask(arg0 task.Key) (task.Task, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetTask", arg0) ret0, _ := ret[0].(task.Task) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTask indicates an expected call of GetTask. func (mr *MockProcessingQueueMockRecorder) GetTask(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockProcessingQueue)(nil).GetTask), arg0) } // GetTasks mocks base method. func (m *MockProcessingQueue) GetTasks() []task.Task { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetTasks") ret0, _ := ret[0].([]task.Task) return ret0 } // GetTasks indicates an expected call of GetTasks. func (mr *MockProcessingQueueMockRecorder) GetTasks() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTasks", reflect.TypeOf((*MockProcessingQueue)(nil).GetTasks)) } // Merge mocks base method. func (m *MockProcessingQueue) Merge(arg0 ProcessingQueue) []ProcessingQueue { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Merge", arg0) ret0, _ := ret[0].([]ProcessingQueue) return ret0 } // Merge indicates an expected call of Merge. func (mr *MockProcessingQueueMockRecorder) Merge(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Merge", reflect.TypeOf((*MockProcessingQueue)(nil).Merge), arg0) } // Split mocks base method. func (m *MockProcessingQueue) Split(arg0 ProcessingQueueSplitPolicy) []ProcessingQueue { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Split", arg0) ret0, _ := ret[0].([]ProcessingQueue) return ret0 } // Split indicates an expected call of Split. func (mr *MockProcessingQueueMockRecorder) Split(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Split", reflect.TypeOf((*MockProcessingQueue)(nil).Split), arg0) } // State mocks base method. func (m *MockProcessingQueue) State() ProcessingQueueState { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "State") ret0, _ := ret[0].(ProcessingQueueState) return ret0 } // State indicates an expected call of State. func (mr *MockProcessingQueueMockRecorder) State() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "State", reflect.TypeOf((*MockProcessingQueue)(nil).State)) } // UpdateAckLevel mocks base method. func (m *MockProcessingQueue) UpdateAckLevel() (task.Key, int) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateAckLevel") ret0, _ := ret[0].(task.Key) ret1, _ := ret[1].(int) return ret0, ret1 } // UpdateAckLevel indicates an expected call of UpdateAckLevel. func (mr *MockProcessingQueueMockRecorder) UpdateAckLevel() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAckLevel", reflect.TypeOf((*MockProcessingQueue)(nil).UpdateAckLevel)) } // MockProcessingQueueSplitPolicy is a mock of ProcessingQueueSplitPolicy interface. type MockProcessingQueueSplitPolicy struct { ctrl *gomock.Controller recorder *MockProcessingQueueSplitPolicyMockRecorder } // MockProcessingQueueSplitPolicyMockRecorder is the mock recorder for MockProcessingQueueSplitPolicy. type MockProcessingQueueSplitPolicyMockRecorder struct { mock *MockProcessingQueueSplitPolicy } // NewMockProcessingQueueSplitPolicy creates a new mock instance. func NewMockProcessingQueueSplitPolicy(ctrl *gomock.Controller) *MockProcessingQueueSplitPolicy { mock := &MockProcessingQueueSplitPolicy{ctrl: ctrl} mock.recorder = &MockProcessingQueueSplitPolicyMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockProcessingQueueSplitPolicy) EXPECT() *MockProcessingQueueSplitPolicyMockRecorder { return m.recorder } // Evaluate mocks base method. func (m *MockProcessingQueueSplitPolicy) Evaluate(arg0 ProcessingQueue) []ProcessingQueueState { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Evaluate", arg0) ret0, _ := ret[0].([]ProcessingQueueState) return ret0 } // Evaluate indicates an expected call of Evaluate. func (mr *MockProcessingQueueSplitPolicyMockRecorder) Evaluate(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Evaluate", reflect.TypeOf((*MockProcessingQueueSplitPolicy)(nil).Evaluate), arg0) } // MockProcessingQueueCollection is a mock of ProcessingQueueCollection interface. type MockProcessingQueueCollection struct { ctrl *gomock.Controller recorder *MockProcessingQueueCollectionMockRecorder } // MockProcessingQueueCollectionMockRecorder is the mock recorder for MockProcessingQueueCollection. type MockProcessingQueueCollectionMockRecorder struct { mock *MockProcessingQueueCollection } // NewMockProcessingQueueCollection creates a new mock instance. func NewMockProcessingQueueCollection(ctrl *gomock.Controller) *MockProcessingQueueCollection { mock := &MockProcessingQueueCollection{ctrl: ctrl} mock.recorder = &MockProcessingQueueCollectionMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockProcessingQueueCollection) EXPECT() *MockProcessingQueueCollectionMockRecorder { return m.recorder } // ActiveQueue mocks base method. func (m *MockProcessingQueueCollection) ActiveQueue() ProcessingQueue { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ActiveQueue") ret0, _ := ret[0].(ProcessingQueue) return ret0 } // ActiveQueue indicates an expected call of ActiveQueue. func (mr *MockProcessingQueueCollectionMockRecorder) ActiveQueue() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActiveQueue", reflect.TypeOf((*MockProcessingQueueCollection)(nil).ActiveQueue)) } // AddTasks mocks base method. func (m *MockProcessingQueueCollection) AddTasks(arg0 map[task.Key]task.Task, arg1 task.Key) { m.ctrl.T.Helper() m.ctrl.Call(m, "AddTasks", arg0, arg1) } // AddTasks indicates an expected call of AddTasks. func (mr *MockProcessingQueueCollectionMockRecorder) AddTasks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTasks", reflect.TypeOf((*MockProcessingQueueCollection)(nil).AddTasks), arg0, arg1) } // GetTask mocks base method. func (m *MockProcessingQueueCollection) GetTask(arg0 task.Key) (task.Task, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetTask", arg0) ret0, _ := ret[0].(task.Task) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTask indicates an expected call of GetTask. func (mr *MockProcessingQueueCollectionMockRecorder) GetTask(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockProcessingQueueCollection)(nil).GetTask), arg0) } // GetTasks mocks base method. func (m *MockProcessingQueueCollection) GetTasks() []task.Task { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetTasks") ret0, _ := ret[0].([]task.Task) return ret0 } // GetTasks indicates an expected call of GetTasks. func (mr *MockProcessingQueueCollectionMockRecorder) GetTasks() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTasks", reflect.TypeOf((*MockProcessingQueueCollection)(nil).GetTasks)) } // Level mocks base method. func (m *MockProcessingQueueCollection) Level() int { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Level") ret0, _ := ret[0].(int) return ret0 } // Level indicates an expected call of Level. func (mr *MockProcessingQueueCollectionMockRecorder) Level() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Level", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Level)) } // Merge mocks base method. func (m *MockProcessingQueueCollection) Merge(arg0 []ProcessingQueue) { m.ctrl.T.Helper() m.ctrl.Call(m, "Merge", arg0) } // Merge indicates an expected call of Merge. func (mr *MockProcessingQueueCollectionMockRecorder) Merge(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Merge", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Merge), arg0) } // Queues mocks base method. func (m *MockProcessingQueueCollection) Queues() []ProcessingQueue { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Queues") ret0, _ := ret[0].([]ProcessingQueue) return ret0 } // Queues indicates an expected call of Queues. func (mr *MockProcessingQueueCollectionMockRecorder) Queues() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Queues", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Queues)) } // Split mocks base method. func (m *MockProcessingQueueCollection) Split(arg0 ProcessingQueueSplitPolicy) []ProcessingQueue { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Split", arg0) ret0, _ := ret[0].([]ProcessingQueue) return ret0 } // Split indicates an expected call of Split. func (mr *MockProcessingQueueCollectionMockRecorder) Split(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Split", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Split), arg0) } // UpdateAckLevels mocks base method. func (m *MockProcessingQueueCollection) UpdateAckLevels() (task.Key, int) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateAckLevels") ret0, _ := ret[0].(task.Key) ret1, _ := ret[1].(int) return ret0, ret1 } // UpdateAckLevels indicates an expected call of UpdateAckLevels. func (mr *MockProcessingQueueCollectionMockRecorder) UpdateAckLevels() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAckLevels", reflect.TypeOf((*MockProcessingQueueCollection)(nil).UpdateAckLevels)) } // MockProcessor is a mock of Processor interface. type MockProcessor struct { ctrl *gomock.Controller recorder *MockProcessorMockRecorder } // MockProcessorMockRecorder is the mock recorder for MockProcessor. type MockProcessorMockRecorder struct { mock *MockProcessor } // NewMockProcessor creates a new mock instance. func NewMockProcessor(ctrl *gomock.Controller) *MockProcessor { mock := &MockProcessor{ctrl: ctrl} mock.recorder = &MockProcessorMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockProcessor) EXPECT() *MockProcessorMockRecorder { return m.recorder } // FailoverDomain mocks base method. func (m *MockProcessor) FailoverDomain(domainIDs map[string]struct{}) { m.ctrl.T.Helper() m.ctrl.Call(m, "FailoverDomain", domainIDs) } // FailoverDomain indicates an expected call of FailoverDomain. func (mr *MockProcessorMockRecorder) FailoverDomain(domainIDs interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FailoverDomain", reflect.TypeOf((*MockProcessor)(nil).FailoverDomain), domainIDs) } // HandleAction mocks base method. func (m *MockProcessor) HandleAction(ctx context.Context, clusterName string, action *Action) (*ActionResult, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "HandleAction", ctx, clusterName, action) ret0, _ := ret[0].(*ActionResult) ret1, _ := ret[1].(error) return ret0, ret1 } // HandleAction indicates an expected call of HandleAction. func (mr *MockProcessorMockRecorder) HandleAction(ctx, clusterName, action interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleAction", reflect.TypeOf((*MockProcessor)(nil).HandleAction), ctx, clusterName, action) } // LockTaskProcessing mocks base method. func (m *MockProcessor) LockTaskProcessing() { m.ctrl.T.Helper() m.ctrl.Call(m, "LockTaskProcessing") } // LockTaskProcessing indicates an expected call of LockTaskProcessing. func (mr *MockProcessorMockRecorder) LockTaskProcessing() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LockTaskProcessing", reflect.TypeOf((*MockProcessor)(nil).LockTaskProcessing)) } // NotifyNewTask mocks base method. func (m *MockProcessor) NotifyNewTask(clusterName string, info *common.NotifyTaskInfo) { m.ctrl.T.Helper() m.ctrl.Call(m, "NotifyNewTask", clusterName, info) } // NotifyNewTask indicates an expected call of NotifyNewTask. func (mr *MockProcessorMockRecorder) NotifyNewTask(clusterName, info interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NotifyNewTask", reflect.TypeOf((*MockProcessor)(nil).NotifyNewTask), clusterName, info) } // Start mocks base method. func (m *MockProcessor) Start() { m.ctrl.T.Helper() m.ctrl.Call(m, "Start") } // Start indicates an expected call of Start. func (mr *MockProcessorMockRecorder) Start() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockProcessor)(nil).Start)) } // Stop mocks base method. func (m *MockProcessor) Stop() { m.ctrl.T.Helper() m.ctrl.Call(m, "Stop") } // Stop indicates an expected call of Stop. func (mr *MockProcessorMockRecorder) Stop() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockProcessor)(nil).Stop)) } // UnlockTaskProcessing mocks base method. func (m *MockProcessor) UnlockTaskProcessing() { m.ctrl.T.Helper() m.ctrl.Call(m, "UnlockTaskProcessing") } // UnlockTaskProcessing indicates an expected call of UnlockTaskProcessing. func (mr *MockProcessorMockRecorder) UnlockTaskProcessing() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnlockTaskProcessing", reflect.TypeOf((*MockProcessor)(nil).UnlockTaskProcessing)) }