prysm-pulse/beacon-chain/db/block_test.go
Raul Jordan 088507af66
Initialize State from ChainStart Deposits in POWChainService (#1474)
* tests passing once again in db package

* fixed all tests

* revert some weird p2p changes made accidentally

* fix repeated code

* build fixed

* fix p2p

* fix all tests
2019-02-03 16:44:48 -06:00

237 lines
5.8 KiB
Go

package db
import (
"testing"
"time"
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
"github.com/prysmaticlabs/prysm/shared/hashutil"
"github.com/prysmaticlabs/prysm/shared/params"
)
func TestNilDB(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
block := &pb.BeaconBlock{}
h, _ := hashutil.HashBeaconBlock(block)
hasBlock := db.HasBlock(h)
if hasBlock {
t.Fatal("HashBlock should return false")
}
bPrime, err := db.Block(h)
if err != nil {
t.Fatalf("failed to get block: %v", err)
}
if bPrime != nil {
t.Fatalf("get should return nil for a non existent key")
}
}
func TestSave(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
block1 := &pb.BeaconBlock{}
h1, _ := hashutil.HashBeaconBlock(block1)
err := db.SaveBlock(block1)
if err != nil {
t.Fatalf("save block failed: %v", err)
}
b1Prime, err := db.Block(h1)
if err != nil {
t.Fatalf("failed to get block: %v", err)
}
h1Prime, _ := hashutil.HashBeaconBlock(b1Prime)
if b1Prime == nil || h1 != h1Prime {
t.Fatalf("get should return b1: %x", h1)
}
block2 := &pb.BeaconBlock{
Slot: 0,
}
h2, _ := hashutil.HashBeaconBlock(block2)
err = db.SaveBlock(block2)
if err != nil {
t.Fatalf("save block failed: %v", err)
}
b2Prime, err := db.Block(h2)
if err != nil {
t.Fatalf("failed to get block: %v", err)
}
h2Prime, _ := hashutil.HashBeaconBlock(b2Prime)
if b2Prime == nil || h2 != h2Prime {
t.Fatalf("get should return b2: %x", h2)
}
}
func TestBlockBySlotEmptyChain(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
block, err := db.BlockBySlot(0)
if err != nil {
t.Errorf("failure when fetching block by slot: %v", err)
}
if block != nil {
t.Error("BlockBySlot should return nil for an empty chain")
}
}
func TestUpdateChainHeadNoBlock(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
genesisTime := uint64(time.Now().Unix())
deposits := setupInitialDeposits(t)
err := db.InitializeState(genesisTime, deposits)
if err != nil {
t.Fatalf("failed to initialize state: %v", err)
}
beaconState, err := db.State()
if err != nil {
t.Fatalf("failed to get beacon state: %v", err)
}
block := &pb.BeaconBlock{Slot: 1}
if err := db.UpdateChainHead(block, beaconState); err == nil {
t.Fatalf("expected UpdateChainHead to fail if the block does not exist: %v", err)
}
}
func TestUpdateChainHead(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
genesisTime := uint64(time.Now().Unix())
deposits := setupInitialDeposits(t)
err := db.InitializeState(genesisTime, deposits)
if err != nil {
t.Fatalf("failed to initialize state: %v", err)
}
block, err := db.BlockBySlot(0)
if err != nil {
t.Fatalf("failed to get genesis block: %v", err)
}
bHash, err := hashutil.HashBeaconBlock(block)
if err != nil {
t.Fatalf("failed to get hash of b: %v", err)
}
beaconState, err := db.State()
if err != nil {
t.Fatalf("failed to get beacon state: %v", err)
}
block2 := &pb.BeaconBlock{
Slot: 1,
ParentRootHash32: bHash[:],
}
b2Hash, err := hashutil.HashBeaconBlock(block2)
if err != nil {
t.Fatalf("failed to hash b2: %v", err)
}
if err := db.SaveBlock(block2); err != nil {
t.Fatalf("failed to save block: %v", err)
}
if err := db.UpdateChainHead(block2, beaconState); err != nil {
t.Fatalf("failed to record the new head of the main chain: %v", err)
}
b2Prime, err := db.BlockBySlot(1)
if err != nil {
t.Fatalf("failed to retrieve slot 1: %v", err)
}
b2Sigma, err := db.ChainHead()
if err != nil {
t.Fatalf("failed to retrieve head: %v", err)
}
b2PrimeHash, err := hashutil.HashBeaconBlock(b2Prime)
if err != nil {
t.Fatalf("failed to hash b2Prime: %v", err)
}
b2SigmaHash, err := hashutil.HashBeaconBlock(b2Sigma)
if err != nil {
t.Fatalf("failed to hash b2Sigma: %v", err)
}
if b2Hash != b2PrimeHash {
t.Fatalf("expected %x and %x to be equal", b2Hash, b2PrimeHash)
}
if b2Hash != b2SigmaHash {
t.Fatalf("expected %x and %x to be equal", b2Hash, b2SigmaHash)
}
}
func TestChainProgress(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
genesisTime := uint64(time.Now().Unix())
deposits := setupInitialDeposits(t)
err := db.InitializeState(genesisTime, deposits)
if err != nil {
t.Fatalf("failed to initialize state: %v", err)
}
beaconState, err := db.State()
if err != nil {
t.Fatalf("Failed to get beacon state: %v", err)
}
cycleLength := params.BeaconConfig().EpochLength
block1 := &pb.BeaconBlock{Slot: 1}
if err := db.SaveBlock(block1); err != nil {
t.Fatalf("failed to save block: %v", err)
}
if err := db.UpdateChainHead(block1, beaconState); err != nil {
t.Fatalf("failed to record the new head: %v", err)
}
heighestBlock, err := db.ChainHead()
if err != nil {
t.Fatalf("failed to get chain head: %v", err)
}
if heighestBlock.Slot != block1.Slot {
t.Fatalf("expected height to equal %d, got %d", block1.Slot, heighestBlock.Slot)
}
block2 := &pb.BeaconBlock{Slot: cycleLength}
if err := db.SaveBlock(block2); err != nil {
t.Fatalf("failed to save block: %v", err)
}
if err := db.UpdateChainHead(block2, beaconState); err != nil {
t.Fatalf("failed to record the new head: %v", err)
}
heighestBlock, err = db.ChainHead()
if err != nil {
t.Fatalf("failed to get block: %v", err)
}
if heighestBlock.Slot != block2.Slot {
t.Fatalf("expected height to equal %d, got %d", block2.Slot, heighestBlock.Slot)
}
block3 := &pb.BeaconBlock{Slot: 3}
if err := db.SaveBlock(block3); err != nil {
t.Fatalf("failed to save block: %v", err)
}
if err := db.UpdateChainHead(block3, beaconState); err != nil {
t.Fatalf("failed to update head: %v", err)
}
heighestBlock, err = db.ChainHead()
if err != nil {
t.Fatalf("failed to get chain head: %v", err)
}
if heighestBlock.Slot != block3.Slot {
t.Fatalf("expected height to equal %d, got %d", block3.Slot, heighestBlock.Slot)
}
}