mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-26 05:17:22 +00:00
ce725ceec3
* Move state pkg to stateV0 pkg * Build.bazel * Remove unused RootsArrayHashTreeRoot * Revert "Remove unused RootsArrayHashTreeRoot" This reverts commit bf0bda30d1a8eb7a071f6e3ce9ee85041b45aca6. Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package helpers_test
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"testing"
|
|
|
|
types "github.com/prysmaticlabs/eth2-types"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stateV0"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestBlockRootAtSlot_CorrectBlockRoot(t *testing.T) {
|
|
var blockRoots [][]byte
|
|
|
|
for i := uint64(0); i < uint64(params.BeaconConfig().SlotsPerHistoricalRoot); i++ {
|
|
blockRoots = append(blockRoots, []byte{byte(i)})
|
|
}
|
|
s := &pb.BeaconState{
|
|
BlockRoots: blockRoots,
|
|
}
|
|
|
|
tests := []struct {
|
|
slot types.Slot
|
|
stateSlot types.Slot
|
|
expectedRoot [32]byte
|
|
}{
|
|
{
|
|
slot: 0,
|
|
stateSlot: 1,
|
|
expectedRoot: [32]byte{0},
|
|
},
|
|
{
|
|
slot: 2,
|
|
stateSlot: 5,
|
|
expectedRoot: [32]byte{2},
|
|
},
|
|
{
|
|
slot: 64,
|
|
stateSlot: 128,
|
|
expectedRoot: [32]byte{64},
|
|
}, {
|
|
slot: 2999,
|
|
stateSlot: 3000,
|
|
expectedRoot: [32]byte{183},
|
|
}, {
|
|
slot: 2873,
|
|
stateSlot: 3000,
|
|
expectedRoot: [32]byte{57},
|
|
},
|
|
{
|
|
slot: 0,
|
|
stateSlot: params.BeaconConfig().SlotsPerHistoricalRoot,
|
|
expectedRoot: [32]byte{},
|
|
},
|
|
}
|
|
for i, tt := range tests {
|
|
t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
|
|
s.Slot = tt.stateSlot
|
|
state, err := stateV0.InitializeFromProto(s)
|
|
require.NoError(t, err)
|
|
wantedSlot := tt.slot
|
|
result, err := helpers.BlockRootAtSlot(state, wantedSlot)
|
|
require.NoError(t, err, "Failed to get block root at slot %d", wantedSlot)
|
|
assert.DeepEqual(t, tt.expectedRoot[:], result, "Result block root was an unexpected value")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestBlockRootAtSlot_OutOfBounds(t *testing.T) {
|
|
var blockRoots [][]byte
|
|
|
|
for i := uint64(0); i < uint64(params.BeaconConfig().SlotsPerHistoricalRoot); i++ {
|
|
blockRoots = append(blockRoots, []byte{byte(i)})
|
|
}
|
|
state := &pb.BeaconState{
|
|
BlockRoots: blockRoots,
|
|
}
|
|
|
|
tests := []struct {
|
|
slot types.Slot
|
|
stateSlot types.Slot
|
|
expectedErr string
|
|
}{
|
|
{
|
|
slot: 1000,
|
|
stateSlot: 500,
|
|
expectedErr: "slot 1000 out of bounds",
|
|
},
|
|
{
|
|
slot: 3000,
|
|
stateSlot: 3000,
|
|
expectedErr: "slot 3000 out of bounds",
|
|
},
|
|
{
|
|
// Edge case where stateSlot is over slots per historical root and
|
|
// slot is not within (stateSlot - HistoricalRootsLimit, statSlot]
|
|
slot: 1,
|
|
stateSlot: params.BeaconConfig().SlotsPerHistoricalRoot + 2,
|
|
expectedErr: "slot 1 out of bounds",
|
|
},
|
|
{
|
|
slot: math.MaxUint64 - 5,
|
|
stateSlot: 0, // Doesn't matter
|
|
expectedErr: "slot overflows uint64",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
state.Slot = tt.stateSlot
|
|
s, err := stateV0.InitializeFromProto(state)
|
|
require.NoError(t, err)
|
|
_, err = helpers.BlockRootAtSlot(s, tt.slot)
|
|
assert.ErrorContains(t, tt.expectedErr, err)
|
|
}
|
|
}
|