mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-09 20:41:20 +00:00
95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ledgerwatch/erigon/cmd/lightclient/rpc/lightrpc"
|
|
"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/communication/p2p"
|
|
"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/communication/ssz_snappy"
|
|
"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/peers"
|
|
"github.com/ledgerwatch/erigon/common"
|
|
"github.com/libp2p/go-libp2p/core/host"
|
|
"github.com/libp2p/go-libp2p/core/protocol"
|
|
basichost "github.com/libp2p/go-libp2p/p2p/host/basic"
|
|
swarmt "github.com/libp2p/go-libp2p/p2p/net/swarm/testing"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func initializeNetwork(t *testing.T, ctx context.Context) (*ConsensusHandlers, host.Host, host.Host) {
|
|
h1, err := basichost.NewHost(swarmt.GenSwarm(t), nil)
|
|
require.NoError(t, err)
|
|
|
|
h2, err := basichost.NewHost(swarmt.GenSwarm(t), nil)
|
|
require.NoError(t, err)
|
|
h2pi := h2.Peerstore().PeerInfo(h2.ID())
|
|
require.NoError(t, h1.Connect(ctx, h2pi))
|
|
|
|
return NewConsensusHandlers(h2, &peers.Peers{}, &lightrpc.MetadataV1{}), h1, h2
|
|
}
|
|
|
|
func TestPingHandler(t *testing.T) {
|
|
ctx := context.TODO()
|
|
|
|
handlers, h1, h2 := initializeNetwork(t, ctx)
|
|
defer h1.Close()
|
|
defer h2.Close()
|
|
handlers.Start()
|
|
|
|
stream, err := h1.NewStream(ctx, h2.ID(), protocol.ID(PingProtocolV1))
|
|
require.NoError(t, err)
|
|
packet := &p2p.Ping{
|
|
Id: 32,
|
|
}
|
|
codec := ssz_snappy.NewStreamCodec(stream)
|
|
_, err = codec.WritePacket(packet)
|
|
require.NoError(t, err)
|
|
require.NoError(t, codec.CloseWriter())
|
|
time.Sleep(100 * time.Millisecond)
|
|
r := &p2p.Ping{}
|
|
|
|
code := make([]byte, 1)
|
|
stream.Read(code)
|
|
assert.Equal(t, code, []byte{SuccessfullResponsePrefix})
|
|
|
|
_, err = codec.Decode(r)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, r, packet)
|
|
}
|
|
|
|
func TestStatusHandler(t *testing.T) {
|
|
ctx := context.TODO()
|
|
|
|
handlers, h1, h2 := initializeNetwork(t, ctx)
|
|
defer h1.Close()
|
|
defer h2.Close()
|
|
handlers.Start()
|
|
|
|
stream, err := h1.NewStream(ctx, h2.ID(), protocol.ID(StatusProtocolV1))
|
|
require.NoError(t, err)
|
|
packet := &p2p.Status{
|
|
ForkDigest: common.Hex2Bytes("69696969"),
|
|
HeadRoot: make([]byte, 32),
|
|
FinalizedRoot: make([]byte, 32),
|
|
HeadSlot: 666999,
|
|
}
|
|
codec := ssz_snappy.NewStreamCodec(stream)
|
|
_, err = codec.WritePacket(packet)
|
|
require.NoError(t, err)
|
|
require.NoError(t, codec.CloseWriter())
|
|
time.Sleep(100 * time.Millisecond)
|
|
r := &p2p.Status{}
|
|
|
|
code := make([]byte, 1)
|
|
stream.Read(code)
|
|
assert.Equal(t, code, []byte{SuccessfullResponsePrefix})
|
|
|
|
_, err = codec.Decode(r)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, r, packet)
|
|
}
|