You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.2 KiB
106 lines
3.2 KiB
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: ./p2p/host/host.go
|
|
|
|
// Package mock is a generated GoMock package.
|
|
package mock
|
|
|
|
import (
|
|
reflect "reflect"
|
|
|
|
gomock "github.com/golang/mock/gomock"
|
|
p2p "github.com/harmony-one/harmony/p2p"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
)
|
|
|
|
// MockHost is a mock of Host interface
|
|
type MockHost struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockHostMockRecorder
|
|
}
|
|
|
|
// MockHostMockRecorder is the mock recorder for MockHost
|
|
type MockHostMockRecorder struct {
|
|
mock *MockHost
|
|
}
|
|
|
|
// NewMockHost creates a new mock instance
|
|
func NewMockHost(ctrl *gomock.Controller) *MockHost {
|
|
mock := &MockHost{ctrl: ctrl}
|
|
mock.recorder = &MockHostMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use
|
|
func (m *MockHost) EXPECT() *MockHostMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// GetSelfPeer mocks base method
|
|
func (m *MockHost) GetSelfPeer() p2p.Peer {
|
|
ret := m.ctrl.Call(m, "GetSelfPeer")
|
|
ret0, _ := ret[0].(p2p.Peer)
|
|
return ret0
|
|
}
|
|
|
|
// GetSelfPeer indicates an expected call of GetSelfPeer
|
|
func (mr *MockHostMockRecorder) GetSelfPeer() *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSelfPeer", reflect.TypeOf((*MockHost)(nil).GetSelfPeer))
|
|
}
|
|
|
|
// SendMessage mocks base method
|
|
func (m *MockHost) SendMessage(arg0 p2p.Peer, arg1 []byte) error {
|
|
ret := m.ctrl.Call(m, "SendMessage", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// SendMessage indicates an expected call of SendMessage
|
|
func (mr *MockHostMockRecorder) SendMessage(arg0, arg1 interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMessage", reflect.TypeOf((*MockHost)(nil).SendMessage), arg0, arg1)
|
|
}
|
|
|
|
// BindHandlerAndServe mocks base method
|
|
func (m *MockHost) BindHandlerAndServe(handler p2p.StreamHandler) {
|
|
m.ctrl.Call(m, "BindHandlerAndServe", handler)
|
|
}
|
|
|
|
// BindHandlerAndServe indicates an expected call of BindHandlerAndServe
|
|
func (mr *MockHostMockRecorder) BindHandlerAndServe(handler interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindHandlerAndServe", reflect.TypeOf((*MockHost)(nil).BindHandlerAndServe), handler)
|
|
}
|
|
|
|
// Close mocks base method
|
|
func (m *MockHost) Close() error {
|
|
ret := m.ctrl.Call(m, "Close")
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Close indicates an expected call of Close
|
|
func (mr *MockHostMockRecorder) Close() *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockHost)(nil).Close))
|
|
}
|
|
|
|
// AddPeer mocks base method
|
|
func (m *MockHost) AddPeer(arg0 *p2p.Peer) error {
|
|
ret := m.ctrl.Call(m, "AddPeer", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// AddPeer indicates an expected call of AddPeer
|
|
func (mr *MockHostMockRecorder) AddPeer(arg0 interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddPeer", reflect.TypeOf((*MockHost)(nil).AddPeer), arg0)
|
|
}
|
|
|
|
// GetID mocks base method
|
|
func (m *MockHost) GetID() peer.ID {
|
|
ret := m.ctrl.Call(m, "GetID")
|
|
ret0, _ := ret[0].(peer.ID)
|
|
return ret0
|
|
}
|
|
|
|
// GetID indicates an expected call of GetID
|
|
func (mr *MockHostMockRecorder) GetID() *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetID", reflect.TypeOf((*MockHost)(nil).GetID))
|
|
}
|
|
|