mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
5aac06f04e
* begin move * use same import path * imports * regen protos * regen * no rename * generate ssz * gaz * fmt * edit build file * imports * modify * remove generated files * remove protos * edit imports in prysm * beacon chain all builds * edit script * add generated pbs * add replace rules * license for ethereumapis protos * change visibility * fmt * update build files to gaz ignore * use proper form * edit imports * wrap block * revert scripts * revert go mod
201 lines
7.6 KiB
Go
201 lines
7.6 KiB
Go
package testutil
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
types "github.com/prysmaticlabs/eth2-types"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/state"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/state/stateutils"
|
|
v1 "github.com/prysmaticlabs/prysm/proto/eth/v1"
|
|
eth "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/interfaces"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestGenerateFullBlock_PassesStateTransition(t *testing.T) {
|
|
beaconState, privs := DeterministicGenesisState(t, 128)
|
|
conf := &BlockGenConfig{
|
|
NumAttestations: 1,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
_, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestGenerateFullBlock_ThousandValidators(t *testing.T) {
|
|
params.SetupTestConfigCleanup(t)
|
|
params.OverrideBeaconConfig(params.MinimalSpecConfig())
|
|
beaconState, privs := DeterministicGenesisState(t, 1024)
|
|
conf := &BlockGenConfig{
|
|
NumAttestations: 4,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
_, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestGenerateFullBlock_Passes4Epochs(t *testing.T) {
|
|
// Changing to minimal config as this will process 4 epochs of blocks.
|
|
params.SetupTestConfigCleanup(t)
|
|
params.OverrideBeaconConfig(params.MinimalSpecConfig())
|
|
beaconState, privs := DeterministicGenesisState(t, 64)
|
|
|
|
conf := &BlockGenConfig{
|
|
NumAttestations: 2,
|
|
}
|
|
finalSlot := params.BeaconConfig().SlotsPerEpoch*4 + 3
|
|
for i := 0; i < int(finalSlot); i++ {
|
|
helpers.ClearCache()
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
beaconState, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Blocks are one slot ahead of beacon state.
|
|
if finalSlot != beaconState.Slot() {
|
|
t.Fatalf("expected output slot to be %d, received %d", finalSlot, beaconState.Slot())
|
|
}
|
|
if beaconState.CurrentJustifiedCheckpoint().Epoch != 3 {
|
|
t.Fatalf("expected justified epoch to change to 3, received %d", beaconState.CurrentJustifiedCheckpoint().Epoch)
|
|
}
|
|
if beaconState.FinalizedCheckpointEpoch() != 2 {
|
|
t.Fatalf("expected finalized epoch to change to 2, received %d", beaconState.CurrentJustifiedCheckpoint().Epoch)
|
|
}
|
|
}
|
|
|
|
func TestGenerateFullBlock_ValidProposerSlashings(t *testing.T) {
|
|
params.SetupTestConfigCleanup(t)
|
|
params.OverrideBeaconConfig(params.MinimalSpecConfig())
|
|
beaconState, privs := DeterministicGenesisState(t, 32)
|
|
conf := &BlockGenConfig{
|
|
NumProposerSlashings: 1,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot()+1)
|
|
require.NoError(t, err)
|
|
beaconState, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
|
|
slashableIndice := block.Block.Body.ProposerSlashings[0].Header_1.Header.ProposerIndex
|
|
if val, err := beaconState.ValidatorAtIndexReadOnly(slashableIndice); err != nil || !val.Slashed() {
|
|
require.NoError(t, err)
|
|
t.Fatal("expected validator to be slashed")
|
|
}
|
|
}
|
|
|
|
func TestGenerateFullBlock_ValidAttesterSlashings(t *testing.T) {
|
|
params.SetupTestConfigCleanup(t)
|
|
params.OverrideBeaconConfig(params.MinimalSpecConfig())
|
|
beaconState, privs := DeterministicGenesisState(t, 32)
|
|
conf := &BlockGenConfig{
|
|
NumAttesterSlashings: 1,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
beaconState, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
|
|
slashableIndices := block.Block.Body.AttesterSlashings[0].Attestation_1.AttestingIndices
|
|
if val, err := beaconState.ValidatorAtIndexReadOnly(types.ValidatorIndex(slashableIndices[0])); err != nil || !val.Slashed() {
|
|
require.NoError(t, err)
|
|
t.Fatal("expected validator to be slashed")
|
|
}
|
|
}
|
|
|
|
func TestGenerateFullBlock_ValidAttestations(t *testing.T) {
|
|
params.SetupTestConfigCleanup(t)
|
|
params.OverrideBeaconConfig(params.MinimalSpecConfig())
|
|
|
|
beaconState, privs := DeterministicGenesisState(t, 256)
|
|
conf := &BlockGenConfig{
|
|
NumAttestations: 4,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
beaconState, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
atts, err := beaconState.CurrentEpochAttestations()
|
|
require.NoError(t, err)
|
|
if len(atts) != 4 {
|
|
t.Fatal("expected 4 attestations to be saved to the beacon state")
|
|
}
|
|
}
|
|
|
|
func TestGenerateFullBlock_ValidDeposits(t *testing.T) {
|
|
beaconState, privs := DeterministicGenesisState(t, 256)
|
|
deposits, _, err := DeterministicDepositsAndKeys(257)
|
|
require.NoError(t, err)
|
|
eth1Data, err := DeterministicEth1Data(len(deposits))
|
|
require.NoError(t, err)
|
|
require.NoError(t, beaconState.SetEth1Data(eth1Data))
|
|
conf := &BlockGenConfig{
|
|
NumDeposits: 1,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
beaconState, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
|
|
depositedPubkey := block.Block.Body.Deposits[0].Data.PublicKey
|
|
valIndexMap := stateutils.ValidatorIndexMap(beaconState.Validators())
|
|
index := valIndexMap[bytesutil.ToBytes48(depositedPubkey)]
|
|
val, err := beaconState.ValidatorAtIndexReadOnly(index)
|
|
require.NoError(t, err)
|
|
if val.EffectiveBalance() != params.BeaconConfig().MaxEffectiveBalance {
|
|
t.Fatalf(
|
|
"expected validator balance to be max effective balance, received %d",
|
|
val.EffectiveBalance(),
|
|
)
|
|
}
|
|
}
|
|
|
|
func TestGenerateFullBlock_ValidVoluntaryExits(t *testing.T) {
|
|
beaconState, privs := DeterministicGenesisState(t, 256)
|
|
// Moving the state 2048 epochs forward due to PERSISTENT_COMMITTEE_PERIOD.
|
|
err := beaconState.SetSlot(params.BeaconConfig().SlotsPerEpoch.Mul(uint64(params.BeaconConfig().ShardCommitteePeriod)).Add(3))
|
|
require.NoError(t, err)
|
|
conf := &BlockGenConfig{
|
|
NumVoluntaryExits: 1,
|
|
}
|
|
block, err := GenerateFullBlock(beaconState, privs, conf, beaconState.Slot())
|
|
require.NoError(t, err)
|
|
beaconState, err = state.ExecuteStateTransition(context.Background(), beaconState, interfaces.WrappedPhase0SignedBeaconBlock(block))
|
|
require.NoError(t, err)
|
|
|
|
exitedIndex := block.Block.Body.VoluntaryExits[0].Exit.ValidatorIndex
|
|
|
|
val, err := beaconState.ValidatorAtIndexReadOnly(exitedIndex)
|
|
require.NoError(t, err)
|
|
if val.ExitEpoch() == params.BeaconConfig().FarFutureEpoch {
|
|
t.Fatal("expected exiting validator index to be marked as exiting")
|
|
}
|
|
}
|
|
|
|
func TestHydrateSignedBeaconBlock_NoError(t *testing.T) {
|
|
b := ð.SignedBeaconBlock{}
|
|
b = HydrateSignedBeaconBlock(b)
|
|
_, err := b.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
_, err = b.Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
_, err = b.Block.Body.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestHydrateV1SignedBeaconBlock_NoError(t *testing.T) {
|
|
b := &v1.SignedBeaconBlock{}
|
|
b = HydrateV1SignedBeaconBlock(b)
|
|
_, err := b.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
_, err = b.Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
_, err = b.Block.Body.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
}
|