mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
918129cf36
* refactor initialization to blocking startup method * require genesisSetter in blockchain, fix tests * work-around gazelle weirdness * fix dep gazelle ignores * only call SetGenesis once * fix typo * validator test setup and fix to return right error * move waitForChainStart to Start * wire up sync Service.genesisWaiter * fix p2p genesisWaiter plumbing * remove extra clock type, integrate into genesis and rename * use time.Now when no Nower is specified * remove unused ClockSetter * simplify rpc context checking * fix typo * use clock everywhere in sync; [32]byte val root * don't use DeepEqual to compare [32]byte and []byte * don't use clock in init sync, not wired up yet * use clock waiter in blockchain as well * use cancelable contexts in tests with goroutines * missed a reference to WithClockSetter * Update beacon-chain/startup/genesis.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * Update beacon-chain/blockchain/service_test.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * more clear docs * doc for NewClock * move clock typedef to more logical file name * adding documentation * gaz * fixes for capella * reducing test raciness * fix races in committee cache tests * lint * add tests on Duration slot math helper * startup package test coverage * fix bad merge * set non-zero genesis time in tests that call Start * happy deepsource, happy me-epsource * replace Synced event with channel * remove unused error * remove accidental wip commit * gaz! * remove unused event constants * remove sync statefeed subscription to fix deadlock * remove state notifier * fix build --------- Co-authored-by: Kasey Kirkham <kasey@users.noreply.github.com> Co-authored-by: Radosław Kapka <rkapka@wp.pl> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com> Co-authored-by: nisdas <nishdas93@gmail.com>
225 lines
8.8 KiB
Go
225 lines
8.8 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/cache"
|
|
testDB "github.com/prysmaticlabs/prysm/v4/beacon-chain/db/testing"
|
|
mockExecution "github.com/prysmaticlabs/prysm/v4/beacon-chain/execution/testing"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/blocks"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/util"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func TestService_isNewProposer(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
service := setupBeaconChain(t, beaconDB)
|
|
require.Equal(t, false, service.isNewProposer(service.CurrentSlot()+1))
|
|
|
|
service.cfg.ProposerSlotIndexCache.SetProposerAndPayloadIDs(service.CurrentSlot()+1, 0, [8]byte{}, [32]byte{} /* root */)
|
|
require.Equal(t, true, service.isNewProposer(service.CurrentSlot()+1))
|
|
}
|
|
|
|
func TestService_isNewHead(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
service := setupBeaconChain(t, beaconDB)
|
|
require.Equal(t, true, service.isNewHead([32]byte{}))
|
|
|
|
service.head = &head{root: [32]byte{1}}
|
|
require.Equal(t, true, service.isNewHead([32]byte{2}))
|
|
require.Equal(t, false, service.isNewHead([32]byte{1}))
|
|
|
|
// Nil head should use origin root
|
|
service.head = nil
|
|
service.originBlockRoot = [32]byte{3}
|
|
require.Equal(t, true, service.isNewHead([32]byte{2}))
|
|
require.Equal(t, false, service.isNewHead([32]byte{3}))
|
|
}
|
|
|
|
func TestService_getHeadStateAndBlock(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
service := setupBeaconChain(t, beaconDB)
|
|
_, _, err := service.getStateAndBlock(context.Background(), [32]byte{})
|
|
require.ErrorContains(t, "block does not exist", err)
|
|
|
|
blk, err := blocks.NewSignedBeaconBlock(util.HydrateSignedBeaconBlock(ðpb.SignedBeaconBlock{Signature: []byte{1}}))
|
|
require.NoError(t, err)
|
|
require.NoError(t, service.cfg.BeaconDB.SaveBlock(context.Background(), blk))
|
|
|
|
st, _ := util.DeterministicGenesisState(t, 1)
|
|
r, err := blk.Block().HashTreeRoot()
|
|
require.NoError(t, err)
|
|
require.NoError(t, service.cfg.BeaconDB.SaveState(context.Background(), st, r))
|
|
|
|
gotState, err := service.cfg.BeaconDB.State(context.Background(), r)
|
|
require.NoError(t, err)
|
|
require.DeepEqual(t, st.ToProto(), gotState.ToProto())
|
|
|
|
gotBlk, err := service.cfg.BeaconDB.Block(context.Background(), r)
|
|
require.NoError(t, err)
|
|
require.DeepEqual(t, blk, gotBlk)
|
|
}
|
|
|
|
func TestService_forkchoiceUpdateWithExecution_exceptionalCases(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
ctx := context.Background()
|
|
opts := testServiceOptsWithDB(t)
|
|
|
|
service, err := NewService(ctx, opts...)
|
|
require.NoError(t, err)
|
|
service.cfg.ProposerSlotIndexCache = cache.NewProposerPayloadIDsCache()
|
|
_, err = service.forkchoiceUpdateWithExecution(ctx, service.headRoot(), service.CurrentSlot()+1)
|
|
require.NoError(t, err)
|
|
hookErr := "could not notify forkchoice update"
|
|
invalidStateErr := "could not get state summary: could not find block in DB"
|
|
require.LogsDoNotContain(t, hook, invalidStateErr)
|
|
require.LogsDoNotContain(t, hook, hookErr)
|
|
gb, err := blocks.NewSignedBeaconBlock(util.NewBeaconBlock())
|
|
require.NoError(t, err)
|
|
require.NoError(t, service.saveInitSyncBlock(ctx, [32]byte{'a'}, gb))
|
|
_, err = service.forkchoiceUpdateWithExecution(ctx, [32]byte{'a'}, service.CurrentSlot()+1)
|
|
require.NoError(t, err)
|
|
require.LogsContain(t, hook, invalidStateErr)
|
|
|
|
hook.Reset()
|
|
service.head = &head{
|
|
root: [32]byte{'a'},
|
|
block: nil, /* should not panic if notify head uses correct head */
|
|
}
|
|
|
|
// Block in Cache
|
|
b := util.NewBeaconBlock()
|
|
b.Block.Slot = 2
|
|
wsb, err := blocks.NewSignedBeaconBlock(b)
|
|
require.NoError(t, err)
|
|
r1, err := b.Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
require.NoError(t, service.saveInitSyncBlock(ctx, r1, wsb))
|
|
st, _ := util.DeterministicGenesisState(t, 1)
|
|
service.head = &head{
|
|
root: r1,
|
|
block: wsb,
|
|
state: st,
|
|
}
|
|
service.cfg.ProposerSlotIndexCache.SetProposerAndPayloadIDs(2, 1, [8]byte{1}, [32]byte{2})
|
|
_, err = service.forkchoiceUpdateWithExecution(ctx, r1, service.CurrentSlot())
|
|
require.NoError(t, err)
|
|
require.LogsDoNotContain(t, hook, invalidStateErr)
|
|
require.LogsDoNotContain(t, hook, hookErr)
|
|
|
|
// Block in DB
|
|
b = util.NewBeaconBlock()
|
|
b.Block.Slot = 3
|
|
util.SaveBlock(t, ctx, service.cfg.BeaconDB, b)
|
|
r1, err = b.Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
st, _ = util.DeterministicGenesisState(t, 1)
|
|
service.head = &head{
|
|
root: r1,
|
|
block: wsb,
|
|
state: st,
|
|
}
|
|
service.cfg.ProposerSlotIndexCache.SetProposerAndPayloadIDs(2, 1, [8]byte{1}, [32]byte{2})
|
|
_, err = service.forkchoiceUpdateWithExecution(ctx, r1, service.CurrentSlot()+1)
|
|
require.NoError(t, err)
|
|
require.LogsDoNotContain(t, hook, invalidStateErr)
|
|
require.LogsDoNotContain(t, hook, hookErr)
|
|
vId, payloadID, has := service.cfg.ProposerSlotIndexCache.GetProposerPayloadIDs(2, [32]byte{2})
|
|
require.Equal(t, true, has)
|
|
require.Equal(t, primitives.ValidatorIndex(1), vId)
|
|
require.Equal(t, [8]byte{1}, payloadID)
|
|
|
|
// Test zero headRoot returns immediately.
|
|
headRoot := service.headRoot()
|
|
_, err = service.forkchoiceUpdateWithExecution(ctx, [32]byte{}, service.CurrentSlot()+1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, service.headRoot(), headRoot)
|
|
}
|
|
|
|
func TestService_forkchoiceUpdateWithExecution_SameHeadRootNewProposer(t *testing.T) {
|
|
service, tr := minimalTestService(t)
|
|
ctx, beaconDB, fcs := tr.ctx, tr.db, tr.fcs
|
|
|
|
altairBlk := util.SaveBlock(t, ctx, beaconDB, util.NewBeaconBlockAltair())
|
|
altairBlkRoot, err := altairBlk.Block().HashTreeRoot()
|
|
require.NoError(t, err)
|
|
bellatrixBlk := util.SaveBlock(t, ctx, beaconDB, util.NewBeaconBlockBellatrix())
|
|
bellatrixBlkRoot, err := bellatrixBlk.Block().HashTreeRoot()
|
|
require.NoError(t, err)
|
|
st, _ := util.DeterministicGenesisState(t, 10)
|
|
service.head = &head{
|
|
state: st,
|
|
}
|
|
|
|
ojc := ðpb.Checkpoint{Root: params.BeaconConfig().ZeroHash[:]}
|
|
ofc := ðpb.Checkpoint{Root: params.BeaconConfig().ZeroHash[:]}
|
|
state, blkRoot, err := prepareForkchoiceState(ctx, 0, [32]byte{}, [32]byte{}, params.BeaconConfig().ZeroHash, ojc, ofc)
|
|
require.NoError(t, err)
|
|
require.NoError(t, fcs.InsertNode(ctx, state, blkRoot))
|
|
state, blkRoot, err = prepareForkchoiceState(ctx, 1, altairBlkRoot, [32]byte{}, params.BeaconConfig().ZeroHash, ojc, ofc)
|
|
require.NoError(t, err)
|
|
require.NoError(t, fcs.InsertNode(ctx, state, blkRoot))
|
|
state, blkRoot, err = prepareForkchoiceState(ctx, 2, bellatrixBlkRoot, altairBlkRoot, params.BeaconConfig().ZeroHash, ojc, ofc)
|
|
require.NoError(t, err)
|
|
require.NoError(t, fcs.InsertNode(ctx, state, blkRoot))
|
|
|
|
service.cfg.ExecutionEngineCaller = &mockExecution.EngineClient{}
|
|
require.NoError(t, beaconDB.SaveState(ctx, st, bellatrixBlkRoot))
|
|
require.NoError(t, beaconDB.SaveGenesisBlockRoot(ctx, bellatrixBlkRoot))
|
|
sb, err := blocks.NewSignedBeaconBlock(util.HydrateSignedBeaconBlockBellatrix(ðpb.SignedBeaconBlockBellatrix{}))
|
|
require.NoError(t, err)
|
|
require.NoError(t, beaconDB.SaveBlock(ctx, sb))
|
|
r, err := sb.Block().HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
// Set head to be the same but proposing next slot
|
|
service.head.root = r
|
|
service.head.block = sb
|
|
service.head.state = st
|
|
service.cfg.ProposerSlotIndexCache.SetProposerAndPayloadIDs(service.CurrentSlot()+1, 0, [8]byte{}, [32]byte{} /* root */)
|
|
_, err = service.forkchoiceUpdateWithExecution(ctx, r, service.CurrentSlot()+1)
|
|
require.NoError(t, err)
|
|
|
|
}
|
|
|
|
func TestShouldOverrideFCU(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
service, tr := minimalTestService(t)
|
|
ctx, fcs := tr.ctx, tr.fcs
|
|
|
|
service.SetGenesisTime(time.Now().Add(-time.Duration(2*params.BeaconConfig().SecondsPerSlot) * time.Second))
|
|
headRoot := [32]byte{'b'}
|
|
parentRoot := [32]byte{'a'}
|
|
ojc := ðpb.Checkpoint{}
|
|
st, root, err := prepareForkchoiceState(ctx, 1, parentRoot, [32]byte{}, [32]byte{}, ojc, ojc)
|
|
require.NoError(t, err)
|
|
require.NoError(t, fcs.InsertNode(ctx, st, root))
|
|
st, root, err = prepareForkchoiceState(ctx, 2, headRoot, parentRoot, [32]byte{}, ojc, ojc)
|
|
require.NoError(t, err)
|
|
require.NoError(t, fcs.InsertNode(ctx, st, root))
|
|
|
|
require.Equal(t, primitives.Slot(2), service.CurrentSlot())
|
|
require.Equal(t, true, service.shouldOverrideFCU(headRoot, 2))
|
|
require.LogsDoNotContain(t, hook, "12 seconds")
|
|
require.Equal(t, false, service.shouldOverrideFCU(parentRoot, 2))
|
|
require.LogsContain(t, hook, "12 seconds")
|
|
|
|
head, err := fcs.Head(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, headRoot, head)
|
|
|
|
fcs.SetGenesisTime(uint64(time.Now().Unix()) - 29)
|
|
require.Equal(t, true, service.shouldOverrideFCU(parentRoot, 3))
|
|
require.LogsDoNotContain(t, hook, "10 seconds")
|
|
fcs.SetGenesisTime(uint64(time.Now().Unix()) - 24)
|
|
service.SetGenesisTime(time.Now().Add(-time.Duration(2*params.BeaconConfig().SecondsPerSlot+10) * time.Second))
|
|
require.Equal(t, false, service.shouldOverrideFCU(parentRoot, 3))
|
|
require.LogsContain(t, hook, "10 seconds")
|
|
}
|