// Code generated by MockGen. DO NOT EDIT. // Source: peer_manager.go // // Generated by this command: // // mockgen -destination peer_manager_mock.go -source peer_manager.go -package persistentcache // // Package persistentcache is a generated GoMock package. package persistentcache import ( context "context" reflect "reflect" gomock "go.uber.org/mock/gomock" ) // MockPeerManager is a mock of PeerManager interface. type MockPeerManager struct { ctrl *gomock.Controller recorder *MockPeerManagerMockRecorder isgomock struct{} } // MockPeerManagerMockRecorder is the mock recorder for MockPeerManager. type MockPeerManagerMockRecorder struct { mock *MockPeerManager } // NewMockPeerManager creates a new mock instance. func NewMockPeerManager(ctrl *gomock.Controller) *MockPeerManager { mock := &MockPeerManager{ctrl: ctrl} mock.recorder = &MockPeerManagerMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockPeerManager) EXPECT() *MockPeerManagerMockRecorder { return m.recorder } // Delete mocks base method. func (m *MockPeerManager) Delete(arg0 context.Context, arg1 string) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Delete", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // Delete indicates an expected call of Delete. func (mr *MockPeerManagerMockRecorder) Delete(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockPeerManager)(nil).Delete), arg0, arg1) } // DeleteAllByHostID mocks base method. func (m *MockPeerManager) DeleteAllByHostID(arg0 context.Context, arg1 string) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteAllByHostID", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // DeleteAllByHostID indicates an expected call of DeleteAllByHostID. func (mr *MockPeerManagerMockRecorder) DeleteAllByHostID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAllByHostID", reflect.TypeOf((*MockPeerManager)(nil).DeleteAllByHostID), arg0, arg1) } // DeleteAllByTaskID mocks base method. func (m *MockPeerManager) DeleteAllByTaskID(arg0 context.Context, arg1 string) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteAllByTaskID", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // DeleteAllByTaskID indicates an expected call of DeleteAllByTaskID. func (mr *MockPeerManagerMockRecorder) DeleteAllByTaskID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAllByTaskID", reflect.TypeOf((*MockPeerManager)(nil).DeleteAllByTaskID), arg0, arg1) } // Load mocks base method. func (m *MockPeerManager) Load(arg0 context.Context, arg1 string) (*Peer, bool) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Load", arg0, arg1) ret0, _ := ret[0].(*Peer) ret1, _ := ret[1].(bool) return ret0, ret1 } // Load indicates an expected call of Load. func (mr *MockPeerManagerMockRecorder) Load(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Load", reflect.TypeOf((*MockPeerManager)(nil).Load), arg0, arg1) } // LoadAll mocks base method. func (m *MockPeerManager) LoadAll(arg0 context.Context) ([]*Peer, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "LoadAll", arg0) ret0, _ := ret[0].([]*Peer) ret1, _ := ret[1].(error) return ret0, ret1 } // LoadAll indicates an expected call of LoadAll. func (mr *MockPeerManagerMockRecorder) LoadAll(arg0 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadAll", reflect.TypeOf((*MockPeerManager)(nil).LoadAll), arg0) } // LoadAllByHostID mocks base method. func (m *MockPeerManager) LoadAllByHostID(arg0 context.Context, arg1 string) ([]*Peer, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "LoadAllByHostID", arg0, arg1) ret0, _ := ret[0].([]*Peer) ret1, _ := ret[1].(error) return ret0, ret1 } // LoadAllByHostID indicates an expected call of LoadAllByHostID. func (mr *MockPeerManagerMockRecorder) LoadAllByHostID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadAllByHostID", reflect.TypeOf((*MockPeerManager)(nil).LoadAllByHostID), arg0, arg1) } // LoadAllByTaskID mocks base method. func (m *MockPeerManager) LoadAllByTaskID(arg0 context.Context, arg1 string) ([]*Peer, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "LoadAllByTaskID", arg0, arg1) ret0, _ := ret[0].([]*Peer) ret1, _ := ret[1].(error) return ret0, ret1 } // LoadAllByTaskID indicates an expected call of LoadAllByTaskID. func (mr *MockPeerManagerMockRecorder) LoadAllByTaskID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadAllByTaskID", reflect.TypeOf((*MockPeerManager)(nil).LoadAllByTaskID), arg0, arg1) } // LoadAllIDsByHostID mocks base method. func (m *MockPeerManager) LoadAllIDsByHostID(arg0 context.Context, arg1 string) ([]string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "LoadAllIDsByHostID", arg0, arg1) ret0, _ := ret[0].([]string) ret1, _ := ret[1].(error) return ret0, ret1 } // LoadAllIDsByHostID indicates an expected call of LoadAllIDsByHostID. func (mr *MockPeerManagerMockRecorder) LoadAllIDsByHostID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadAllIDsByHostID", reflect.TypeOf((*MockPeerManager)(nil).LoadAllIDsByHostID), arg0, arg1) } // LoadAllIDsByTaskID mocks base method. func (m *MockPeerManager) LoadAllIDsByTaskID(arg0 context.Context, arg1 string) ([]string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "LoadAllIDsByTaskID", arg0, arg1) ret0, _ := ret[0].([]string) ret1, _ := ret[1].(error) return ret0, ret1 } // LoadAllIDsByTaskID indicates an expected call of LoadAllIDsByTaskID. func (mr *MockPeerManagerMockRecorder) LoadAllIDsByTaskID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadAllIDsByTaskID", reflect.TypeOf((*MockPeerManager)(nil).LoadAllIDsByTaskID), arg0, arg1) } // LoadPersistentAllByTaskID mocks base method. func (m *MockPeerManager) LoadPersistentAllByTaskID(arg0 context.Context, arg1 string) ([]*Peer, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "LoadPersistentAllByTaskID", arg0, arg1) ret0, _ := ret[0].([]*Peer) ret1, _ := ret[1].(error) return ret0, ret1 } // LoadPersistentAllByTaskID indicates an expected call of LoadPersistentAllByTaskID. func (mr *MockPeerManagerMockRecorder) LoadPersistentAllByTaskID(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadPersistentAllByTaskID", reflect.TypeOf((*MockPeerManager)(nil).LoadPersistentAllByTaskID), arg0, arg1) } // Store mocks base method. func (m *MockPeerManager) Store(arg0 context.Context, arg1 *Peer) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Store", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // Store indicates an expected call of Store. func (mr *MockPeerManagerMockRecorder) Store(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Store", reflect.TypeOf((*MockPeerManager)(nil).Store), arg0, arg1) }