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  }