dragonfly/scheduler/resource/peer_manager_mock.go

103 lines
3.1 KiB
Go

// Code generated by MockGen. DO NOT EDIT.
// Source: peer_manager.go
// Package resource is a generated GoMock package.
package resource
import (
reflect "reflect"
gomock "github.com/golang/mock/gomock"
)
// MockPeerManager is a mock of PeerManager interface.
type MockPeerManager struct {
ctrl *gomock.Controller
recorder *MockPeerManagerMockRecorder
}
// 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 string) {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Delete", arg0)
}
// Delete indicates an expected call of Delete.
func (mr *MockPeerManagerMockRecorder) Delete(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockPeerManager)(nil).Delete), arg0)
}
// Load mocks base method.
func (m *MockPeerManager) Load(arg0 string) (*Peer, bool) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Load", arg0)
ret0, _ := ret[0].(*Peer)
ret1, _ := ret[1].(bool)
return ret0, ret1
}
// Load indicates an expected call of Load.
func (mr *MockPeerManagerMockRecorder) Load(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Load", reflect.TypeOf((*MockPeerManager)(nil).Load), arg0)
}
// LoadOrStore mocks base method.
func (m *MockPeerManager) LoadOrStore(arg0 *Peer) (*Peer, bool) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "LoadOrStore", arg0)
ret0, _ := ret[0].(*Peer)
ret1, _ := ret[1].(bool)
return ret0, ret1
}
// LoadOrStore indicates an expected call of LoadOrStore.
func (mr *MockPeerManagerMockRecorder) LoadOrStore(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadOrStore", reflect.TypeOf((*MockPeerManager)(nil).LoadOrStore), arg0)
}
// RunGC mocks base method.
func (m *MockPeerManager) RunGC() error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "RunGC")
ret0, _ := ret[0].(error)
return ret0
}
// RunGC indicates an expected call of RunGC.
func (mr *MockPeerManagerMockRecorder) RunGC() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunGC", reflect.TypeOf((*MockPeerManager)(nil).RunGC))
}
// Store mocks base method.
func (m *MockPeerManager) Store(arg0 *Peer) {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Store", arg0)
}
// Store indicates an expected call of Store.
func (mr *MockPeerManagerMockRecorder) Store(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Store", reflect.TypeOf((*MockPeerManager)(nil).Store), arg0)
}