erigon-pulse/cmd/ef-tests-cl/consensus_tests/finality.go
a 9644e6d220
Implement SpecTests in native go, add fork_choice handler (#7422)
a few TODO: remain to make this not a draft

---------

Co-authored-by: Giulio <giulio.rebuffo@gmail.com>
2023-05-02 16:19:22 +02:00

42 lines
1.1 KiB
Go

package consensus_tests
import (
"fmt"
"io/fs"
"testing"
"github.com/ledgerwatch/erigon/cmd/ef-tests-cl/spectest"
"github.com/ledgerwatch/erigon/cmd/erigon-cl/core/transition"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
var FinalityFinality = spectest.HandlerFunc(func(t *testing.T, root fs.FS, c spectest.TestCase) (err error) {
testState, err := spectest.ReadBeaconState(root, c.Version(), spectest.PreSsz)
require.NoError(t, err)
expectedState, err := spectest.ReadBeaconState(root, c.Version(), spectest.PostSsz)
require.NoError(t, err)
blocks, err := spectest.ReadBlocks(root, c.Version())
if err != nil {
return err
}
startSlot := testState.Slot()
for _, block := range blocks {
if err := transition.TransitionState(testState, block, true); err != nil {
require.NoError(t, fmt.Errorf("cannot transition state: %w. slot=%d. start_slot=%d", err, block.Block.Slot, startSlot))
}
}
expectedRoot, err := testState.HashSSZ()
assert.NoError(t, err)
haveRoot, err := expectedState.HashSSZ()
assert.NoError(t, err)
assert.EqualValues(t, haveRoot, expectedRoot, "state root")
return nil
})