mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
b0128ad894
* bump bitfield dep * add new methods * get it working * add nil check * add check * one more check * add flag * everything works local run * add debug log * more changes * ensuring p2p interface works enough for tests to pass * all tests pass * include proper naming and comments to fix lint * Apply suggestions from code review * discover by peers * cannot figure out why 0 peers * remove keys * fix test * fix it * fix again * remove log * change back * gaz Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
39 lines
1.4 KiB
Go
39 lines
1.4 KiB
Go
package testing
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/libp2p/go-libp2p-core/network"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/p2p/peers"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
)
|
|
|
|
// MockPeersProvider implements PeersProvider for testing.
|
|
type MockPeersProvider struct {
|
|
lock sync.Mutex
|
|
peers *peers.Status
|
|
}
|
|
|
|
// Peers provides access the peer status.
|
|
func (m *MockPeersProvider) Peers() *peers.Status {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
if m.peers == nil {
|
|
m.peers = peers.NewStatus(5 /* maxBadResponses */)
|
|
// Pretend we are connected to two peers
|
|
id0, _ := peer.IDB58Decode("16Uiu2HAkyWZ4Ni1TpvDS8dPxsozmHY85KaiFjodQuV6Tz5tkHVeR")
|
|
ma0, _ := ma.NewMultiaddr("/ip4/213.202.254.180/tcp/13000")
|
|
m.peers.Add(id0, ma0, network.DirInbound, []uint64{})
|
|
m.peers.SetConnectionState(id0, peers.PeerConnected)
|
|
m.peers.SetChainState(id0, &pb.Status{FinalizedEpoch: uint64(10)})
|
|
id1, _ := peer.IDB58Decode("16Uiu2HAm4HgJ9N1o222xK61o7LSgToYWoAy1wNTJRkh9gLZapVAy")
|
|
ma1, _ := ma.NewMultiaddr("/ip4/52.23.23.253/tcp/30000/ipfs/QmfAgkmjiZNZhr2wFN9TwaRgHouMTBT6HELyzE5A3BT2wK/p2p-circuit")
|
|
m.peers.Add(id1, ma1, network.DirOutbound, []uint64{})
|
|
m.peers.SetConnectionState(id1, peers.PeerConnected)
|
|
m.peers.SetChainState(id1, &pb.Status{FinalizedEpoch: uint64(11)})
|
|
}
|
|
return m.peers
|
|
}
|