mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
44c3adb367
* add method * fix Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/ethereum/go-ethereum/p2p/enr"
|
|
"github.com/libp2p/go-libp2p-core/host"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
"github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
// MockPeerManager is mock of the PeerManager interface.
|
|
type MockPeerManager struct {
|
|
Enr *enr.Record
|
|
PID peer.ID
|
|
BHost host.Host
|
|
}
|
|
|
|
// Disconnect .
|
|
func (m *MockPeerManager) Disconnect(peer.ID) error {
|
|
return nil
|
|
}
|
|
|
|
// PeerID .
|
|
func (m *MockPeerManager) PeerID() peer.ID {
|
|
return m.PID
|
|
}
|
|
|
|
// Host .
|
|
func (m *MockPeerManager) Host() host.Host {
|
|
return m.BHost
|
|
}
|
|
|
|
// ENR .
|
|
func (m MockPeerManager) ENR() *enr.Record {
|
|
return m.Enr
|
|
}
|
|
|
|
// DiscoveryAddress .
|
|
func (m MockPeerManager) DiscoveryAddress() (multiaddr.Multiaddr, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
// RefreshENR .
|
|
func (m MockPeerManager) RefreshENR() {}
|
|
|
|
// FindPeersWithSubnet .
|
|
func (m MockPeerManager) FindPeersWithSubnet(_ context.Context, _ string, _, _ uint64) (bool, error) {
|
|
return true, nil
|
|
}
|
|
|
|
// AddPingMethod .
|
|
func (m MockPeerManager) AddPingMethod(_ func(ctx context.Context, id peer.ID) error) {}
|