github.com/influxdata/influxdb/v2@v2.7.6/replications/mock/queue_management.go (about) 1 // Code generated by MockGen. DO NOT EDIT. 2 // Source: github.com/influxdata/influxdb/v2/replications (interfaces: DurableQueueManager) 3 4 // Package mock is a generated GoMock package. 5 package mock 6 7 import ( 8 reflect "reflect" 9 10 gomock "github.com/golang/mock/gomock" 11 influxdb "github.com/influxdata/influxdb/v2" 12 platform "github.com/influxdata/influxdb/v2/kit/platform" 13 ) 14 15 // MockDurableQueueManager is a mock of DurableQueueManager interface. 16 type MockDurableQueueManager struct { 17 ctrl *gomock.Controller 18 recorder *MockDurableQueueManagerMockRecorder 19 } 20 21 // MockDurableQueueManagerMockRecorder is the mock recorder for MockDurableQueueManager. 22 type MockDurableQueueManagerMockRecorder struct { 23 mock *MockDurableQueueManager 24 } 25 26 // NewMockDurableQueueManager creates a new mock instance. 27 func NewMockDurableQueueManager(ctrl *gomock.Controller) *MockDurableQueueManager { 28 mock := &MockDurableQueueManager{ctrl: ctrl} 29 mock.recorder = &MockDurableQueueManagerMockRecorder{mock} 30 return mock 31 } 32 33 // EXPECT returns an object that allows the caller to indicate expected use. 34 func (m *MockDurableQueueManager) EXPECT() *MockDurableQueueManagerMockRecorder { 35 return m.recorder 36 } 37 38 // CloseAll mocks base method. 39 func (m *MockDurableQueueManager) CloseAll() error { 40 m.ctrl.T.Helper() 41 ret := m.ctrl.Call(m, "CloseAll") 42 ret0, _ := ret[0].(error) 43 return ret0 44 } 45 46 // CloseAll indicates an expected call of CloseAll. 47 func (mr *MockDurableQueueManagerMockRecorder) CloseAll() *gomock.Call { 48 mr.mock.ctrl.T.Helper() 49 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseAll", reflect.TypeOf((*MockDurableQueueManager)(nil).CloseAll)) 50 } 51 52 // CurrentQueueSizes mocks base method. 53 func (m *MockDurableQueueManager) CurrentQueueSizes(arg0 []platform.ID) (map[platform.ID]int64, error) { 54 m.ctrl.T.Helper() 55 ret := m.ctrl.Call(m, "CurrentQueueSizes", arg0) 56 ret0, _ := ret[0].(map[platform.ID]int64) 57 ret1, _ := ret[1].(error) 58 return ret0, ret1 59 } 60 61 // CurrentQueueSizes indicates an expected call of CurrentQueueSizes. 62 func (mr *MockDurableQueueManagerMockRecorder) CurrentQueueSizes(arg0 interface{}) *gomock.Call { 63 mr.mock.ctrl.T.Helper() 64 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CurrentQueueSizes", reflect.TypeOf((*MockDurableQueueManager)(nil).CurrentQueueSizes), arg0) 65 } 66 67 // DeleteQueue mocks base method. 68 func (m *MockDurableQueueManager) DeleteQueue(arg0 platform.ID) error { 69 m.ctrl.T.Helper() 70 ret := m.ctrl.Call(m, "DeleteQueue", arg0) 71 ret0, _ := ret[0].(error) 72 return ret0 73 } 74 75 // DeleteQueue indicates an expected call of DeleteQueue. 76 func (mr *MockDurableQueueManagerMockRecorder) DeleteQueue(arg0 interface{}) *gomock.Call { 77 mr.mock.ctrl.T.Helper() 78 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteQueue", reflect.TypeOf((*MockDurableQueueManager)(nil).DeleteQueue), arg0) 79 } 80 81 // EnqueueData mocks base method. 82 func (m *MockDurableQueueManager) EnqueueData(arg0 platform.ID, arg1 []byte, arg2 int) error { 83 m.ctrl.T.Helper() 84 ret := m.ctrl.Call(m, "EnqueueData", arg0, arg1, arg2) 85 ret0, _ := ret[0].(error) 86 return ret0 87 } 88 89 // EnqueueData indicates an expected call of EnqueueData. 90 func (mr *MockDurableQueueManagerMockRecorder) EnqueueData(arg0, arg1, arg2 interface{}) *gomock.Call { 91 mr.mock.ctrl.T.Helper() 92 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnqueueData", reflect.TypeOf((*MockDurableQueueManager)(nil).EnqueueData), arg0, arg1, arg2) 93 } 94 95 // GetReplications mocks base method. 96 func (m *MockDurableQueueManager) GetReplications(arg0, arg1 platform.ID) []platform.ID { 97 m.ctrl.T.Helper() 98 ret := m.ctrl.Call(m, "GetReplications", arg0, arg1) 99 ret0, _ := ret[0].([]platform.ID) 100 return ret0 101 } 102 103 // GetReplications indicates an expected call of GetReplications. 104 func (mr *MockDurableQueueManagerMockRecorder) GetReplications(arg0, arg1 interface{}) *gomock.Call { 105 mr.mock.ctrl.T.Helper() 106 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplications", reflect.TypeOf((*MockDurableQueueManager)(nil).GetReplications), arg0, arg1) 107 } 108 109 // InitializeQueue mocks base method. 110 func (m *MockDurableQueueManager) InitializeQueue(arg0 platform.ID, arg1 int64, arg2, arg3 platform.ID, arg4 int64) error { 111 m.ctrl.T.Helper() 112 ret := m.ctrl.Call(m, "InitializeQueue", arg0, arg1, arg2, arg3, arg4) 113 ret0, _ := ret[0].(error) 114 return ret0 115 } 116 117 // InitializeQueue indicates an expected call of InitializeQueue. 118 func (mr *MockDurableQueueManagerMockRecorder) InitializeQueue(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { 119 mr.mock.ctrl.T.Helper() 120 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InitializeQueue", reflect.TypeOf((*MockDurableQueueManager)(nil).InitializeQueue), arg0, arg1, arg2, arg3, arg4) 121 } 122 123 // RemainingQueueSizes mocks base method. 124 func (m *MockDurableQueueManager) RemainingQueueSizes(arg0 []platform.ID) (map[platform.ID]int64, error) { 125 m.ctrl.T.Helper() 126 ret := m.ctrl.Call(m, "RemainingQueueSizes", arg0) 127 ret0, _ := ret[0].(map[platform.ID]int64) 128 ret1, _ := ret[1].(error) 129 return ret0, ret1 130 } 131 132 // RemainingQueueSizes indicates an expected call of RemainingQueueSizes. 133 func (mr *MockDurableQueueManagerMockRecorder) RemainingQueueSizes(arg0 interface{}) *gomock.Call { 134 mr.mock.ctrl.T.Helper() 135 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemainingQueueSizes", reflect.TypeOf((*MockDurableQueueManager)(nil).RemainingQueueSizes), arg0) 136 } 137 138 // StartReplicationQueues mocks base method. 139 func (m *MockDurableQueueManager) StartReplicationQueues(arg0 map[platform.ID]*influxdb.TrackedReplication) error { 140 m.ctrl.T.Helper() 141 ret := m.ctrl.Call(m, "StartReplicationQueues", arg0) 142 ret0, _ := ret[0].(error) 143 return ret0 144 } 145 146 // StartReplicationQueues indicates an expected call of StartReplicationQueues. 147 func (mr *MockDurableQueueManagerMockRecorder) StartReplicationQueues(arg0 interface{}) *gomock.Call { 148 mr.mock.ctrl.T.Helper() 149 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartReplicationQueues", reflect.TypeOf((*MockDurableQueueManager)(nil).StartReplicationQueues), arg0) 150 } 151 152 // UpdateMaxQueueSize mocks base method. 153 func (m *MockDurableQueueManager) UpdateMaxQueueSize(arg0 platform.ID, arg1 int64) error { 154 m.ctrl.T.Helper() 155 ret := m.ctrl.Call(m, "UpdateMaxQueueSize", arg0, arg1) 156 ret0, _ := ret[0].(error) 157 return ret0 158 } 159 160 // UpdateMaxQueueSize indicates an expected call of UpdateMaxQueueSize. 161 func (mr *MockDurableQueueManagerMockRecorder) UpdateMaxQueueSize(arg0, arg1 interface{}) *gomock.Call { 162 mr.mock.ctrl.T.Helper() 163 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateMaxQueueSize", reflect.TypeOf((*MockDurableQueueManager)(nil).UpdateMaxQueueSize), arg0, arg1) 164 }