mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
379 lines
11 KiB
Go
379 lines
11 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io/ioutil"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
gethTypes "github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/params"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/powchain"
|
|
"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/testutil"
|
|
"github.com/sirupsen/logrus"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func init() {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
logrus.SetOutput(ioutil.Discard)
|
|
}
|
|
|
|
type mockClient struct{}
|
|
|
|
func (f *mockClient) SubscribeNewHead(ctx context.Context, ch chan<- *gethTypes.Header) (ethereum.Subscription, error) {
|
|
return new(event.Feed).Subscribe(ch), nil
|
|
}
|
|
|
|
func (f *mockClient) BlockByHash(ctx context.Context, hash common.Hash) (*gethTypes.Block, error) {
|
|
head := &gethTypes.Header{Number: big.NewInt(0), Difficulty: big.NewInt(100)}
|
|
return gethTypes.NewBlockWithHeader(head), nil
|
|
}
|
|
|
|
func (f *mockClient) SubscribeFilterLogs(ctx context.Context, q ethereum.FilterQuery, ch chan<- gethTypes.Log) (ethereum.Subscription, error) {
|
|
return new(event.Feed).Subscribe(ch), nil
|
|
}
|
|
|
|
func (f *mockClient) LatestBlockHash() common.Hash {
|
|
return common.BytesToHash([]byte{'A'})
|
|
}
|
|
|
|
type faultyClient struct{}
|
|
|
|
func (f *faultyClient) SubscribeNewHead(ctx context.Context, ch chan<- *gethTypes.Header) (ethereum.Subscription, error) {
|
|
return new(event.Feed).Subscribe(ch), nil
|
|
}
|
|
|
|
func (f *faultyClient) BlockByHash(ctx context.Context, hash common.Hash) (*gethTypes.Block, error) {
|
|
return nil, errors.New("failed")
|
|
}
|
|
|
|
func (f *faultyClient) SubscribeFilterLogs(ctx context.Context, q ethereum.FilterQuery, ch chan<- gethTypes.Log) (ethereum.Subscription, error) {
|
|
return new(event.Feed).Subscribe(ch), nil
|
|
}
|
|
|
|
func (f *faultyClient) LatestBlockHash() common.Hash {
|
|
return common.BytesToHash([]byte{'A'})
|
|
}
|
|
|
|
func init() {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
}
|
|
|
|
func setupBeaconChain(t *testing.T, faultyPoWClient bool) *ChainService {
|
|
config := db.Config{Path: "", Name: "", InMemory: true}
|
|
db, err := db.NewDB(config)
|
|
if err != nil {
|
|
t.Fatalf("could not setup beaconDB: %v", err)
|
|
}
|
|
|
|
endpoint := "ws://127.0.0.1"
|
|
ctx := context.Background()
|
|
var web3Service *powchain.Web3Service
|
|
if faultyPoWClient {
|
|
client := &faultyClient{}
|
|
web3Service, err = powchain.NewWeb3Service(ctx, &powchain.Web3ServiceConfig{Endpoint: endpoint, Pubkey: "", VrcAddr: common.Address{}}, client, client, client)
|
|
} else {
|
|
client := &mockClient{}
|
|
web3Service, err = powchain.NewWeb3Service(ctx, &powchain.Web3ServiceConfig{Endpoint: endpoint, Pubkey: "", VrcAddr: common.Address{}}, client, client, client)
|
|
}
|
|
if err != nil {
|
|
t.Fatalf("unable to set up web3 service: %v", err)
|
|
}
|
|
cfg := &Config{
|
|
BeaconBlockBuf: 0,
|
|
BeaconDB: db,
|
|
Web3Service: web3Service,
|
|
EnablePOWChain: true,
|
|
}
|
|
if err != nil {
|
|
t.Fatalf("could not register blockchain service: %v", err)
|
|
}
|
|
chainService, err := NewChainService(ctx, cfg)
|
|
if err != nil {
|
|
t.Fatalf("unable to setup chain service: %v", err)
|
|
}
|
|
|
|
return chainService
|
|
}
|
|
|
|
func TestStartStop(t *testing.T) {
|
|
chainService := setupBeaconChain(t, false)
|
|
defer chainService.beaconDB.Close()
|
|
|
|
chainService.IncomingBlockFeed()
|
|
|
|
// Test the start function.
|
|
chainService.Start()
|
|
|
|
if err := chainService.Stop(); err != nil {
|
|
t.Fatalf("unable to stop chain service: %v", err)
|
|
}
|
|
|
|
// The context should have been canceled.
|
|
if chainService.ctx.Err() == nil {
|
|
t.Error("context was not canceled")
|
|
}
|
|
}
|
|
|
|
func TestRunningChainServiceFaultyPOWChain(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
chainService := setupBeaconChain(t, true)
|
|
defer chainService.beaconDB.Close()
|
|
|
|
block := types.NewBlock(&pb.BeaconBlock{
|
|
Slot: 1,
|
|
PowChainRef: []byte("a"),
|
|
})
|
|
|
|
blockChan := make(chan *types.Block)
|
|
exitRoutine := make(chan bool)
|
|
go func() {
|
|
chainService.blockProcessing(blockChan)
|
|
<-exitRoutine
|
|
}()
|
|
|
|
if err := chainService.beaconDB.SaveBlock(block); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
chainService.incomingBlockChan <- block
|
|
chainService.cancel()
|
|
exitRoutine <- true
|
|
|
|
testutil.AssertLogsContain(t, hook, "Proof-of-Work chain reference in block does not exist")
|
|
}
|
|
|
|
func TestRunningChainService(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
|
|
chainService := setupBeaconChain(t, false)
|
|
defer chainService.beaconDB.Close()
|
|
|
|
active := types.NewGenesisActiveState()
|
|
crystallized, err := types.NewGenesisCrystallizedState(nil)
|
|
if err != nil {
|
|
t.Fatalf("Can't generate genesis state: %v", err)
|
|
}
|
|
|
|
activeStateRoot, _ := active.Hash()
|
|
crystallizedStateRoot, _ := crystallized.Hash()
|
|
|
|
genesis := types.NewGenesisBlock([32]byte{}, [32]byte{})
|
|
chainService.beaconDB.SaveBlock(genesis)
|
|
parentHash, err := genesis.Hash()
|
|
if err != nil {
|
|
t.Fatalf("unable to get hash of canonical head: %v", err)
|
|
}
|
|
|
|
currentSlot := uint64(1)
|
|
slotsStart := crystallized.LastStateRecalculationSlot() - params.GetConfig().CycleLength
|
|
slotIndex := (currentSlot - slotsStart) % params.GetConfig().CycleLength
|
|
Shard := crystallized.ShardAndCommitteesForSlots()[slotIndex].ArrayShardAndCommittee[0].Shard
|
|
|
|
block := types.NewBlock(&pb.BeaconBlock{
|
|
Slot: currentSlot,
|
|
ActiveStateRoot: activeStateRoot[:],
|
|
CrystallizedStateRoot: crystallizedStateRoot[:],
|
|
AncestorHashes: [][]byte{parentHash[:]},
|
|
PowChainRef: []byte("a"),
|
|
Attestations: []*pb.AggregatedAttestation{{
|
|
Slot: currentSlot,
|
|
AttesterBitfield: []byte{128, 0},
|
|
Shard: Shard,
|
|
JustifiedBlockHash: parentHash[:],
|
|
}},
|
|
})
|
|
|
|
blockNoParent := types.NewBlock(&pb.BeaconBlock{
|
|
Slot: currentSlot,
|
|
PowChainRef: []byte("a"),
|
|
AncestorHashes: [][]byte{{}},
|
|
})
|
|
|
|
blockChan := make(chan *types.Block)
|
|
exitRoutine := make(chan bool)
|
|
go func() {
|
|
chainService.blockProcessing(blockChan)
|
|
<-exitRoutine
|
|
}()
|
|
|
|
if err := chainService.beaconDB.SaveBlock(block); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
chainService.incomingBlockChan <- blockNoParent
|
|
chainService.incomingBlockChan <- block
|
|
<-blockChan
|
|
chainService.cancel()
|
|
exitRoutine <- true
|
|
testutil.AssertLogsContain(t, hook, "Chain service context closed, exiting goroutine")
|
|
testutil.AssertLogsContain(t, hook, "Block points to nil parent")
|
|
testutil.AssertLogsContain(t, hook, "Finished processing received block")
|
|
}
|
|
|
|
func TestDoesPOWBlockExist(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
chainService := setupBeaconChain(t, true)
|
|
defer chainService.beaconDB.Close()
|
|
|
|
block := types.NewBlock(&pb.BeaconBlock{
|
|
Slot: 10,
|
|
})
|
|
|
|
// Using a faulty client should throw error.
|
|
exists := chainService.doesPoWBlockExist(block)
|
|
if exists {
|
|
t.Error("Block corresponding to nil powchain reference should not exist")
|
|
}
|
|
testutil.AssertLogsContain(t, hook, "fetching PoW block corresponding to mainchain reference failed")
|
|
}
|
|
|
|
func TestUpdateHead(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
chainService := setupBeaconChain(t, false)
|
|
defer chainService.beaconDB.Close()
|
|
|
|
active := types.NewGenesisActiveState()
|
|
crystallized, err := types.NewGenesisCrystallizedState(nil)
|
|
if err != nil {
|
|
t.Fatalf("Can't generate genesis state: %v", err)
|
|
}
|
|
ActiveStateRoot, _ := active.Hash()
|
|
CrystallizedStateRoot, _ := crystallized.Hash()
|
|
|
|
genesis := types.NewGenesisBlock(ActiveStateRoot, CrystallizedStateRoot)
|
|
genesisHash, err := genesis.Hash()
|
|
if err != nil {
|
|
t.Fatalf("Could not get genesis block hash: %v", err)
|
|
}
|
|
|
|
block := types.NewBlock(&pb.BeaconBlock{
|
|
Slot: 64,
|
|
ActiveStateRoot: ActiveStateRoot[:],
|
|
CrystallizedStateRoot: CrystallizedStateRoot[:],
|
|
AncestorHashes: [][]byte{genesisHash[:]},
|
|
PowChainRef: []byte("a"),
|
|
})
|
|
|
|
exitRoutine := make(chan bool)
|
|
blockChan := make(chan *types.Block)
|
|
go func() {
|
|
chainService.updateHead(blockChan)
|
|
<-exitRoutine
|
|
}()
|
|
|
|
if err := chainService.beaconDB.SaveBlock(block); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// If blocks pending processing is empty, the updateHead routine does nothing.
|
|
blockChan <- block
|
|
chainService.cancel()
|
|
exitRoutine <- true
|
|
|
|
testutil.AssertLogsContain(t, hook, "Canonical block determined")
|
|
}
|
|
|
|
func TestProcessBlocksWithCorrectAttestations(t *testing.T) {
|
|
chainService := setupBeaconChain(t, false)
|
|
defer chainService.beaconDB.Close()
|
|
|
|
active := types.NewGenesisActiveState()
|
|
crystallized, err := types.NewGenesisCrystallizedState(nil)
|
|
if err != nil {
|
|
t.Fatalf("Can't generate genesis state: %v", err)
|
|
}
|
|
|
|
ActiveStateRoot, _ := active.Hash()
|
|
CrystallizedStateRoot, _ := crystallized.Hash()
|
|
|
|
block0 := types.NewBlock(&pb.BeaconBlock{
|
|
Slot: 0,
|
|
})
|
|
if saveErr := chainService.beaconDB.SaveBlock(block0); saveErr != nil {
|
|
t.Fatalf("Cannot save block: %v", saveErr)
|
|
}
|
|
block0Hash, err := block0.Hash()
|
|
if err != nil {
|
|
t.Fatalf("Failed to compute block's hash: %v", err)
|
|
}
|
|
|
|
currentSlot := uint64(1)
|
|
|
|
block1 := types.NewBlock(&pb.BeaconBlock{
|
|
AncestorHashes: [][]byte{block0Hash[:]},
|
|
Slot: currentSlot,
|
|
ActiveStateRoot: ActiveStateRoot[:],
|
|
CrystallizedStateRoot: CrystallizedStateRoot[:],
|
|
Attestations: []*pb.AggregatedAttestation{{
|
|
Slot: currentSlot,
|
|
AttesterBitfield: []byte{16, 0},
|
|
Shard: 0,
|
|
}},
|
|
})
|
|
|
|
blockChan := make(chan *types.Block)
|
|
exitRoutine := make(chan bool)
|
|
go func() {
|
|
chainService.blockProcessing(blockChan)
|
|
<-exitRoutine
|
|
}()
|
|
|
|
chainService.incomingBlockChan <- block1
|
|
block1Returned := <-blockChan
|
|
|
|
if block1 != block1Returned {
|
|
t.Fatalf("expected %v and %v to be the same", block1, block1Returned)
|
|
}
|
|
|
|
block1Hash, err := block1.Hash()
|
|
if err != nil {
|
|
t.Fatalf("unable to get hash of block 1: %v", err)
|
|
}
|
|
|
|
currentSlot++
|
|
|
|
// Add 1 more attestation field for slot2
|
|
block2 := types.NewBlock(&pb.BeaconBlock{
|
|
AncestorHashes: [][]byte{block1Hash[:]},
|
|
Slot: currentSlot,
|
|
Attestations: []*pb.AggregatedAttestation{
|
|
{Slot: currentSlot - 1, AttesterBitfield: []byte{8, 0}, Shard: 0},
|
|
{Slot: currentSlot, AttesterBitfield: []byte{8, 0}, Shard: 0},
|
|
}})
|
|
block2Hash, err := block2.Hash()
|
|
if err != nil {
|
|
t.Fatalf("unable to get hash of block 1: %v", err)
|
|
}
|
|
|
|
currentSlot++
|
|
|
|
// Add 1 more attestation field for slot3
|
|
block3 := types.NewBlock(&pb.BeaconBlock{
|
|
AncestorHashes: [][]byte{block2Hash[:]},
|
|
Slot: currentSlot,
|
|
Attestations: []*pb.AggregatedAttestation{
|
|
{Slot: currentSlot - 2, AttesterBitfield: []byte{4, 0}, Shard: 0},
|
|
{Slot: currentSlot - 1, AttesterBitfield: []byte{4, 0}, Shard: 0},
|
|
{Slot: currentSlot, AttesterBitfield: []byte{4, 0}, Shard: 0},
|
|
}})
|
|
|
|
chainService.incomingBlockChan <- block1
|
|
<-blockChan
|
|
chainService.incomingBlockChan <- block2
|
|
<-blockChan
|
|
chainService.incomingBlockChan <- block3
|
|
<-blockChan
|
|
|
|
chainService.cancel()
|
|
exitRoutine <- true
|
|
}
|