mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-09 11:11:20 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
60 lines
2.0 KiB
Go
60 lines
2.0 KiB
Go
package debug
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/golang/protobuf/ptypes/empty"
|
|
mockP2p "github.com/prysmaticlabs/prysm/v4/beacon-chain/p2p/testing"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
)
|
|
|
|
func TestDebugServer_GetPeer(t *testing.T) {
|
|
peersProvider := &mockP2p.MockPeersProvider{}
|
|
mP2P := mockP2p.NewTestP2P(t)
|
|
ds := &Server{
|
|
PeersFetcher: peersProvider,
|
|
PeerManager: &mockP2p.MockPeerManager{BHost: mP2P.BHost},
|
|
}
|
|
firstPeer := peersProvider.Peers().All()[0]
|
|
|
|
res, err := ds.GetPeer(context.Background(), ðpb.PeerRequest{PeerId: firstPeer.String()})
|
|
require.NoError(t, err)
|
|
require.Equal(t, firstPeer.String(), res.PeerId, "Unexpected peer ID")
|
|
|
|
assert.Equal(t, int(ethpb.PeerDirection_INBOUND), int(res.Direction), "Expected 1st peer to be an inbound connection")
|
|
assert.Equal(t, ethpb.ConnectionState_CONNECTED, res.ConnectionState, "Expected peer to be connected")
|
|
}
|
|
|
|
func TestDebugServer_ListPeers(t *testing.T) {
|
|
peersProvider := &mockP2p.MockPeersProvider{}
|
|
mP2P := mockP2p.NewTestP2P(t)
|
|
ds := &Server{
|
|
PeersFetcher: peersProvider,
|
|
PeerManager: &mockP2p.MockPeerManager{BHost: mP2P.BHost},
|
|
}
|
|
|
|
res, err := ds.ListPeers(context.Background(), &empty.Empty{})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 2, len(res.Responses))
|
|
|
|
direction1 := res.Responses[0].Direction
|
|
direction2 := res.Responses[1].Direction
|
|
assert.Equal(t,
|
|
true,
|
|
direction1 == ethpb.PeerDirection_INBOUND || direction2 == ethpb.PeerDirection_INBOUND,
|
|
"Expected an inbound peer")
|
|
assert.Equal(t,
|
|
true,
|
|
direction1 == ethpb.PeerDirection_OUTBOUND || direction2 == ethpb.PeerDirection_OUTBOUND,
|
|
"Expected an outbound peer")
|
|
if len(res.Responses[0].ListeningAddresses) == 0 {
|
|
t.Errorf("Expected 1st peer to have a multiaddress, instead they have no addresses")
|
|
}
|
|
if len(res.Responses[1].ListeningAddresses) == 0 {
|
|
t.Errorf("Expected 2nd peer to have a multiaddress, instead they have no addresses")
|
|
}
|
|
}
|