mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
1f21e196b6
* Merge variable declaration and assignment * Use result of type assertion to simplify cases * Replace call to bytes.Compare with bytes.Equal * Drop unnecessary use of the blank identifier * Replace x.Sub(time.Now()) with time.Until(x) * Function literal can be simplified * Use a single append to concatenate two slices * Replace time.Now().Sub(x) with time.Since(x) * Omit comparison with boolean constant * Omit redundant nil check on slices * Nested if can be replaced with else-if * Function call can be replaced with helper function * Omit redundant control flow * Use plain channel send or receive * Simplify returning boolean expression * Merge branch 'origin-master' into fix-antipatterns * Merge branch 'master' into fix-antipatterns
48 lines
925 B
Go
48 lines
925 B
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"
|
|
)
|
|
|
|
// 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
|
|
}
|
|
|
|
// RefreshENR .
|
|
func (m MockPeerManager) RefreshENR() {}
|
|
|
|
// FindPeersWithSubnet .
|
|
func (m MockPeerManager) FindPeersWithSubnet(ctx context.Context, index uint64) (bool, error) {
|
|
return true, nil
|
|
}
|
|
|
|
// AddPingMethod .
|
|
func (m MockPeerManager) AddPingMethod(reqFunc func(ctx context.Context, id peer.ID) error) {}
|