prysm-pulse/beacon-chain/db/kv/checkpoint_test.go
Preston Van Loon 0481eb4872
Track finalized block root index and filter in blocks by range (#3873)
* checkpoint progress

* feature flag and test

* return true if feature flag is not on

* add lint msg

* rollback tx if failure

* Revert validator/BUILD.bazel
2019-10-29 11:14:17 -04:00

137 lines
3.0 KiB
Go

package kv
import (
"context"
"testing"
"github.com/gogo/protobuf/proto"
"github.com/prysmaticlabs/go-ssz"
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
"github.com/prysmaticlabs/prysm/shared/bytesutil"
)
func TestStore_JustifiedCheckpoint_CanSaveRetrieve(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
ctx := context.Background()
cp := &ethpb.Checkpoint{
Epoch: 10,
Root: []byte{'A'},
}
if err := db.SaveJustifiedCheckpoint(ctx, cp); err != nil {
t.Fatal(err)
}
retrieved, err := db.JustifiedCheckpoint(ctx)
if err != nil {
t.Fatal(err)
}
if !proto.Equal(cp, retrieved) {
t.Errorf("Wanted %v, received %v", cp, retrieved)
}
}
func TestStore_FinalizedCheckpoint_CanSaveRetrieve(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
ctx := context.Background()
genesis := bytesutil.ToBytes32([]byte{'G', 'E', 'N', 'E', 'S', 'I', 'S'})
if err := db.SaveGenesisBlockRoot(ctx, genesis); err != nil {
t.Fatal(err)
}
blk := &ethpb.BeaconBlock{
ParentRoot: genesis[:],
Slot: 40,
}
root, err := ssz.SigningRoot(blk)
if err != nil {
t.Fatal(err)
}
cp := &ethpb.Checkpoint{
Epoch: 5,
Root: root[:],
}
// a valid chain is required to save finalized checkpoint.
if err := db.SaveBlock(ctx, blk); err != nil {
t.Fatal(err)
}
// a state is required to save checkpoint
if err := db.SaveState(ctx, &pb.BeaconState{}, root); err != nil {
t.Fatal(err)
}
if err := db.SaveFinalizedCheckpoint(ctx, cp); err != nil {
t.Fatal(err)
}
retrieved, err := db.FinalizedCheckpoint(ctx)
if err != nil {
t.Fatal(err)
}
if !proto.Equal(cp, retrieved) {
t.Errorf("Wanted %v, received %v", cp, retrieved)
}
}
func TestStore_JustifiedCheckpoint_DefaultCantBeNil(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
ctx := context.Background()
genesisRoot := [32]byte{'A'}
if err := db.SaveGenesisBlockRoot(ctx, genesisRoot); err != nil {
t.Fatal(err)
}
cp := &ethpb.Checkpoint{Root: genesisRoot[:]}
retrieved, err := db.JustifiedCheckpoint(ctx)
if err != nil {
t.Fatal(err)
}
if !proto.Equal(cp, retrieved) {
t.Errorf("Wanted %v, received %v", cp, retrieved)
}
}
func TestStore_FinalizedCheckpoint_DefaultCantBeNil(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
ctx := context.Background()
genesisRoot := [32]byte{'B'}
if err := db.SaveGenesisBlockRoot(ctx, genesisRoot); err != nil {
t.Fatal(err)
}
cp := &ethpb.Checkpoint{Root: genesisRoot[:]}
retrieved, err := db.FinalizedCheckpoint(ctx)
if err != nil {
t.Fatal(err)
}
if !proto.Equal(cp, retrieved) {
t.Errorf("Wanted %v, received %v", cp, retrieved)
}
}
func TestStore_FinalizedCheckpoint_StateMustExist(t *testing.T) {
db := setupDB(t)
defer teardownDB(t, db)
ctx := context.Background()
cp := &ethpb.Checkpoint{
Epoch: 5,
Root: []byte{'B'},
}
if err := db.SaveFinalizedCheckpoint(ctx, cp); err != errMissingStateForFinalizedCheckpoint {
t.Fatalf("wanted err %v, got %v", errMissingStateForFinalizedCheckpoint, err)
}
}