github.com/uber/kraken@v0.1.4/mocks/tracker/peerstore/store.go (about) 1 // Code generated by MockGen. DO NOT EDIT. 2 // Source: github.com/uber/kraken/tracker/peerstore (interfaces: Store) 3 4 // Package mockpeerstore is a generated GoMock package. 5 package mockpeerstore 6 7 import ( 8 gomock "github.com/golang/mock/gomock" 9 core "github.com/uber/kraken/core" 10 reflect "reflect" 11 ) 12 13 // MockStore is a mock of Store interface 14 type MockStore struct { 15 ctrl *gomock.Controller 16 recorder *MockStoreMockRecorder 17 } 18 19 // MockStoreMockRecorder is the mock recorder for MockStore 20 type MockStoreMockRecorder struct { 21 mock *MockStore 22 } 23 24 // NewMockStore creates a new mock instance 25 func NewMockStore(ctrl *gomock.Controller) *MockStore { 26 mock := &MockStore{ctrl: ctrl} 27 mock.recorder = &MockStoreMockRecorder{mock} 28 return mock 29 } 30 31 // EXPECT returns an object that allows the caller to indicate expected use 32 func (m *MockStore) EXPECT() *MockStoreMockRecorder { 33 return m.recorder 34 } 35 36 // GetPeers mocks base method 37 func (m *MockStore) GetPeers(arg0 core.InfoHash, arg1 int) ([]*core.PeerInfo, error) { 38 m.ctrl.T.Helper() 39 ret := m.ctrl.Call(m, "GetPeers", arg0, arg1) 40 ret0, _ := ret[0].([]*core.PeerInfo) 41 ret1, _ := ret[1].(error) 42 return ret0, ret1 43 } 44 45 // GetPeers indicates an expected call of GetPeers 46 func (mr *MockStoreMockRecorder) GetPeers(arg0, arg1 interface{}) *gomock.Call { 47 mr.mock.ctrl.T.Helper() 48 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPeers", reflect.TypeOf((*MockStore)(nil).GetPeers), arg0, arg1) 49 } 50 51 // UpdatePeer mocks base method 52 func (m *MockStore) UpdatePeer(arg0 core.InfoHash, arg1 *core.PeerInfo) error { 53 m.ctrl.T.Helper() 54 ret := m.ctrl.Call(m, "UpdatePeer", arg0, arg1) 55 ret0, _ := ret[0].(error) 56 return ret0 57 } 58 59 // UpdatePeer indicates an expected call of UpdatePeer 60 func (mr *MockStoreMockRecorder) UpdatePeer(arg0, arg1 interface{}) *gomock.Call { 61 mr.mock.ctrl.T.Helper() 62 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdatePeer", reflect.TypeOf((*MockStore)(nil).UpdatePeer), arg0, arg1) 63 }