prysm-pulse/beacon-chain/sync/subscriber_beacon_blocks_test.go
2019-12-16 19:53:55 -06:00

52 lines
1.6 KiB
Go

package sync
import (
"context"
"fmt"
"testing"
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
"github.com/prysmaticlabs/go-ssz"
mock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing"
dbtest "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
"github.com/prysmaticlabs/prysm/shared/params"
"github.com/prysmaticlabs/prysm/shared/testutil"
"github.com/sirupsen/logrus"
logTest "github.com/sirupsen/logrus/hooks/test"
)
func init() {
logrus.SetLevel(logrus.DebugLevel)
}
func TestRegularSyncBeaconBlockSubscriber_FilterByFinalizedEpoch(t *testing.T) {
hook := logTest.NewGlobal()
db := dbtest.SetupDB(t)
defer dbtest.TeardownDB(t, db)
s := &pb.BeaconState{FinalizedCheckpoint: &ethpb.Checkpoint{Epoch: 1}}
parent := &ethpb.BeaconBlock{}
if err := db.SaveBlock(context.Background(), parent); err != nil {
t.Fatal(err)
}
parentRoot, _ := ssz.SigningRoot(parent)
r := &Service{
db: db,
chain: &mock.ChainService{State: s},
}
b := &ethpb.BeaconBlock{Slot: 1, ParentRoot: parentRoot[:]}
if err := r.beaconBlockSubscriber(context.Background(), b); err != nil {
t.Fatal(err)
}
testutil.AssertLogsContain(t, hook, fmt.Sprintf("Received a block older than finalized checkpoint, 1 < %d", params.BeaconConfig().SlotsPerEpoch))
hook.Reset()
b.Slot = params.BeaconConfig().SlotsPerEpoch
if err := r.beaconBlockSubscriber(context.Background(), b); err != nil {
t.Fatal(err)
}
testutil.AssertLogsDoNotContain(t, hook, "Received a block older than finalized checkpoint")
}