mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
138 lines
2.9 KiB
Go
138 lines
2.9 KiB
Go
package simulator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/types"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/event"
|
|
"github.com/prysmaticlabs/prysm/shared/p2p"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
"github.com/sirupsen/logrus"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func init() {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
logrus.SetOutput(ioutil.Discard)
|
|
}
|
|
|
|
type mockP2P struct{}
|
|
|
|
func (mp *mockP2P) Subscribe(msg proto.Message, channel chan p2p.Message) event.Subscription {
|
|
return new(event.Feed).Subscribe(channel)
|
|
}
|
|
|
|
func (mp *mockP2P) Broadcast(msg proto.Message) {}
|
|
|
|
func (mp *mockP2P) Send(msg proto.Message, peer p2p.Peer) {}
|
|
|
|
type mockPOWChainService struct{}
|
|
|
|
func (mpow *mockPOWChainService) LatestBlockHash() common.Hash {
|
|
return common.BytesToHash([]byte{})
|
|
}
|
|
|
|
func setupSimulator(t *testing.T) *Simulator {
|
|
ctx := context.Background()
|
|
|
|
config := db.Config{Path: "", Name: "", InMemory: true}
|
|
db, err := db.NewDB(config)
|
|
if err != nil {
|
|
t.Fatalf("could not setup beaconDB: %v", err)
|
|
}
|
|
|
|
cfg := &Config{
|
|
Delay: time.Second,
|
|
BlockRequestBuf: 0,
|
|
P2P: &mockP2P{},
|
|
Web3Service: &mockPOWChainService{},
|
|
BeaconDB: db,
|
|
EnablePOWChain: true,
|
|
}
|
|
|
|
return NewSimulator(ctx, cfg)
|
|
}
|
|
|
|
func TestLifecycle(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
sim := setupSimulator(t)
|
|
|
|
sim.Start()
|
|
testutil.AssertLogsContain(t, hook, "Starting service")
|
|
sim.Stop()
|
|
testutil.AssertLogsContain(t, hook, "Stopping service")
|
|
|
|
// The context should have been canceled.
|
|
if sim.ctx.Err() == nil {
|
|
t.Error("context was not canceled")
|
|
}
|
|
}
|
|
|
|
func TestBroadcastBlockHash(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
sim := setupSimulator(t)
|
|
|
|
delayChan := make(chan time.Time)
|
|
exitRoutine := make(chan bool)
|
|
|
|
go func() {
|
|
sim.run(delayChan)
|
|
<-exitRoutine
|
|
}()
|
|
|
|
delayChan <- time.Time{}
|
|
sim.cancel()
|
|
exitRoutine <- true
|
|
|
|
testutil.AssertLogsContain(t, hook, "Announcing block hash")
|
|
|
|
if len(sim.broadcastedBlockHashes) != 1 {
|
|
t.Error("Did not store the broadcasted block hash")
|
|
}
|
|
hook.Reset()
|
|
}
|
|
|
|
func TestBlockRequest(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
sim := setupSimulator(t)
|
|
|
|
delayChan := make(chan time.Time)
|
|
exitRoutine := make(chan bool)
|
|
|
|
go func() {
|
|
sim.run(delayChan)
|
|
<-exitRoutine
|
|
}()
|
|
|
|
block := types.NewBlock(&pb.BeaconBlock{AncestorHashes: make([][]byte, 32)})
|
|
h, err := block.Hash()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
data := &pb.BeaconBlockRequest{
|
|
Hash: h[:],
|
|
}
|
|
|
|
msg := p2p.Message{
|
|
Peer: p2p.Peer{},
|
|
Data: data,
|
|
}
|
|
|
|
sim.broadcastedBlocks[h] = block
|
|
|
|
sim.blockRequestChan <- msg
|
|
sim.cancel()
|
|
exitRoutine <- true
|
|
|
|
testutil.AssertLogsContain(t, hook, fmt.Sprintf("Responding to full block request for hash: 0x%x", h))
|
|
}
|