203 lines
7.2 KiB
Go
203 lines
7.2 KiB
Go
// 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)
|
|
}
|