mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
7ba500c417
* applies assertion funcs to beacon-chain/state tests * applies assertion funcs to beacon-chain/state/stategen tests * applies assertion funcs to beacon-chain/state/stateutil tests * gazelle * Merge refs/heads/master into state-apply-testutils-assertions * Merge refs/heads/master into state-apply-testutils-assertions * Merge refs/heads/master into state-apply-testutils-assertions
114 lines
3.1 KiB
Go
114 lines
3.1 KiB
Go
package stateutil_test
|
|
|
|
import (
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stateutil"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/interop"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestState_FieldCount(t *testing.T) {
|
|
count := 21
|
|
typ := reflect.TypeOf(pb.BeaconState{})
|
|
numFields := 0
|
|
for i := 0; i < typ.NumField(); i++ {
|
|
if strings.HasPrefix(typ.Field(i).Name, "XXX_") {
|
|
continue
|
|
}
|
|
numFields++
|
|
}
|
|
assert.Equal(t, count, numFields)
|
|
}
|
|
|
|
func BenchmarkHashTreeRootState_Custom_512(b *testing.B) {
|
|
b.StopTimer()
|
|
genesisState := setupGenesisState(b, 512)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := stateutil.HashTreeRootState(genesisState)
|
|
require.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashTreeRootState_Custom_16384(b *testing.B) {
|
|
b.StopTimer()
|
|
genesisState := setupGenesisState(b, 16384)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := stateutil.HashTreeRootState(genesisState)
|
|
require.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashTreeRootState_Custom_300000(b *testing.B) {
|
|
b.StopTimer()
|
|
genesisState := setupGenesisState(b, 300000)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := stateutil.HashTreeRootState(genesisState)
|
|
require.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashTreeRoot_Generic_512(b *testing.B) {
|
|
b.StopTimer()
|
|
genesisState := setupGenesisState(b, 512)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := ssz.HashTreeRoot(genesisState)
|
|
require.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashTreeRoot_Generic_16384(b *testing.B) {
|
|
b.StopTimer()
|
|
genesisState := setupGenesisState(b, 16384)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := ssz.HashTreeRoot(genesisState)
|
|
require.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashTreeRoot_Generic_300000(b *testing.B) {
|
|
b.StopTimer()
|
|
genesisState := setupGenesisState(b, 300000)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := ssz.HashTreeRoot(genesisState)
|
|
require.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func setupGenesisState(tb testing.TB, count uint64) *pb.BeaconState {
|
|
genesisState, _, err := interop.GenerateGenesisState(0, 1)
|
|
require.NoError(tb, err, "Could not generate genesis beacon state")
|
|
for i := uint64(1); i < count; i++ {
|
|
someRoot := [32]byte{}
|
|
someKey := [48]byte{}
|
|
copy(someRoot[:], strconv.Itoa(int(i)))
|
|
copy(someKey[:], strconv.Itoa(int(i)))
|
|
genesisState.Validators = append(genesisState.Validators, ðpb.Validator{
|
|
PublicKey: someKey[:],
|
|
WithdrawalCredentials: someRoot[:],
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
Slashed: false,
|
|
ActivationEligibilityEpoch: 1,
|
|
ActivationEpoch: 1,
|
|
ExitEpoch: 1,
|
|
WithdrawableEpoch: 1,
|
|
})
|
|
genesisState.Balances = append(genesisState.Balances, params.BeaconConfig().MaxEffectiveBalance)
|
|
}
|
|
return genesisState
|
|
}
|