prysm-pulse/beacon-chain/state/testing/getters_checkpoint.go

100 lines
3.1 KiB
Go
Raw Normal View History

package testing
import (
"github.com/prysmaticlabs/go-bitfield"
"github.com/prysmaticlabs/prysm/v3/beacon-chain/state"
fieldparams "github.com/prysmaticlabs/prysm/v3/config/fieldparams"
types "github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
ethpb "github.com/prysmaticlabs/prysm/v3/proto/prysm/v1alpha1"
"github.com/prysmaticlabs/prysm/v3/testing/require"
"testing"
)
func VerifyBeaconStateJustificationBitsNil(t *testing.T, factory getState) {
s, err := factory()
require.NoError(t, err)
Deduplicate native state (a.k.a. One State to rule them all) (#10483) * v0 * getters/setters * init and copy * hasher * all the nice stuff * make bazel happy * remove tests for smaller PR * remove old states * move files * import fixes * custom MarshalSSZ * fixed deadlock * copy version when copying state * correct issues in state_trie * fix Copy() * better e2e comment * add code to minimal state * spectest test * Revert "Auxiliary commit to revert individual files from 84154423464e8372f7e0a03367403656ac5cd78e" This reverts commit 9602599d183081291dfa0ba4f1036430f63a7822. * native state assert * always error * always log * more native state usage * cleanup * remove empty line * Revert "spectests" This reverts commit 1c49bed5d1cf6224afaf21e18562bf72fae5d2b6. # Conflicts: # beacon-chain/powchain/service.go # beacon-chain/state/v1/state_trie.go # beacon-chain/state/v2/state_trie.go # beacon-chain/state/v3/state_trie.go # testing/spectest/shared/phase0/finality/BUILD.bazel # testing/spectest/shared/phase0/finality/runner.go * dedup field trie * fix test issues * cleanup * use correct field num in FinalizedRootProof * use existing version constant * halfway there * "working" version * some fixes * fix field nums in tests * rename v0types to nativetypes * Revert "Auxiliary commit to revert individual files from dc549b1cf8e724bd08cee1ecc760ff3771d5592d" This reverts commit 7254d3070d8693b283fc686a2e01a822ecbac1b3. * uncomment code * remove map size * Revert "Revert "spectests"" This reverts commit 39c271ae6b57d6a3737e2c202cd8407857475e56. * use reverse map * Revert "Revert "Revert "spectests""" This reverts commit 19ba8cf95cdca689357c8234a262e08cccbafef4. * finally found the bug (cherry picked from commit a5414c4be1bdb61a50b391ea5301895e772cc5e9) * simplify populateFieldIndexes * fix copy (cherry picked from commit 7da4fb8cf51557ef931bb781872ea52fc6731af5) * remove native state from e2e * remove index map * unsupported functions * Use ProtobufBeaconState() from native state * tests * typo * reduce complexity of `SaveStatesEfficient` * remove unused receiver name * update doc.go * fix test assertion * fix test assertion 2 * Phase0 justification bits * bring back state tests * rename fieldIndexRev * versioning of ToProto * remove version check from unexported function * hasher tests * don't return error from JustificationBits * extract fieldConvertersNative * helper error function * use fieldConvertersNative * Introduce RealPosition method on FieldIndex * use RealPosition in hasher * remove unused fields * remove TestAppendBeyondIndicesLimit (cherry picked from commit 3017e700282969c30006b64c95c21ffe6b166f8b) * simplify RealPosition * rename field interface * use helper in proofs.go * Update beacon-chain/core/altair/upgrade.go Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
2022-05-09 13:02:34 +00:00
require.DeepEqual(t, bitfield.Bitvector4{}.Bytes(), s.JustificationBits().Bytes())
}
type getStateWithJustificationBits = func(bitfield.Bitvector4) (state.BeaconState, error)
func VerifyBeaconStateJustificationBits(t *testing.T, factory getStateWithJustificationBits) {
s, err := factory(bitfield.Bitvector4{1, 2, 3, 4})
require.NoError(t, err)
Deduplicate native state (a.k.a. One State to rule them all) (#10483) * v0 * getters/setters * init and copy * hasher * all the nice stuff * make bazel happy * remove tests for smaller PR * remove old states * move files * import fixes * custom MarshalSSZ * fixed deadlock * copy version when copying state * correct issues in state_trie * fix Copy() * better e2e comment * add code to minimal state * spectest test * Revert "Auxiliary commit to revert individual files from 84154423464e8372f7e0a03367403656ac5cd78e" This reverts commit 9602599d183081291dfa0ba4f1036430f63a7822. * native state assert * always error * always log * more native state usage * cleanup * remove empty line * Revert "spectests" This reverts commit 1c49bed5d1cf6224afaf21e18562bf72fae5d2b6. # Conflicts: # beacon-chain/powchain/service.go # beacon-chain/state/v1/state_trie.go # beacon-chain/state/v2/state_trie.go # beacon-chain/state/v3/state_trie.go # testing/spectest/shared/phase0/finality/BUILD.bazel # testing/spectest/shared/phase0/finality/runner.go * dedup field trie * fix test issues * cleanup * use correct field num in FinalizedRootProof * use existing version constant * halfway there * "working" version * some fixes * fix field nums in tests * rename v0types to nativetypes * Revert "Auxiliary commit to revert individual files from dc549b1cf8e724bd08cee1ecc760ff3771d5592d" This reverts commit 7254d3070d8693b283fc686a2e01a822ecbac1b3. * uncomment code * remove map size * Revert "Revert "spectests"" This reverts commit 39c271ae6b57d6a3737e2c202cd8407857475e56. * use reverse map * Revert "Revert "Revert "spectests""" This reverts commit 19ba8cf95cdca689357c8234a262e08cccbafef4. * finally found the bug (cherry picked from commit a5414c4be1bdb61a50b391ea5301895e772cc5e9) * simplify populateFieldIndexes * fix copy (cherry picked from commit 7da4fb8cf51557ef931bb781872ea52fc6731af5) * remove native state from e2e * remove index map * unsupported functions * Use ProtobufBeaconState() from native state * tests * typo * reduce complexity of `SaveStatesEfficient` * remove unused receiver name * update doc.go * fix test assertion * fix test assertion 2 * Phase0 justification bits * bring back state tests * rename fieldIndexRev * versioning of ToProto * remove version check from unexported function * hasher tests * don't return error from JustificationBits * extract fieldConvertersNative * helper error function * use fieldConvertersNative * Introduce RealPosition method on FieldIndex * use RealPosition in hasher * remove unused fields * remove TestAppendBeyondIndicesLimit (cherry picked from commit 3017e700282969c30006b64c95c21ffe6b166f8b) * simplify RealPosition * rename field interface * use helper in proofs.go * Update beacon-chain/core/altair/upgrade.go Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
2022-05-09 13:02:34 +00:00
require.DeepEqual(t, bitfield.Bitvector4{1, 2, 3, 4}.Bytes(), s.JustificationBits().Bytes())
}
func VerifyBeaconStatePreviousJustifiedCheckpointNil(t *testing.T, factory getState) {
s, err := factory()
require.NoError(t, err)
checkpoint := s.PreviousJustifiedCheckpoint()
require.Equal(t, (*ethpb.Checkpoint)(nil), checkpoint)
}
type getStateWithCheckpoint = func(checkpoint *ethpb.Checkpoint) (state.BeaconState, error)
func VerifyBeaconStatePreviousJustifiedCheckpoint(t *testing.T, factory getStateWithCheckpoint) {
orgCheckpoint := &ethpb.Checkpoint{Root: make([]byte, fieldparams.RootLength)}
orgCheckpoint.Root[1] = 1
orgCheckpoint.Root[2] = 2
orgCheckpoint.Root[3] = 3
s, err := factory(orgCheckpoint)
require.NoError(t, err)
checkpoint := s.PreviousJustifiedCheckpoint()
require.DeepEqual(t, orgCheckpoint.Root, checkpoint.Root)
}
func VerifyBeaconStateCurrentJustifiedCheckpointNil(t *testing.T, factory getState) {
s, err := factory()
require.NoError(t, err)
checkpoint := s.CurrentJustifiedCheckpoint()
require.Equal(t, (*ethpb.Checkpoint)(nil), checkpoint)
}
func VerifyBeaconStateCurrentJustifiedCheckpoint(t *testing.T, factory getStateWithCheckpoint) {
orgCheckpoint := &ethpb.Checkpoint{Root: make([]byte, fieldparams.RootLength)}
orgCheckpoint.Root[1] = 1
orgCheckpoint.Root[2] = 2
orgCheckpoint.Root[3] = 3
s, err := factory(orgCheckpoint)
require.NoError(t, err)
checkpoint := s.CurrentJustifiedCheckpoint()
require.DeepEqual(t, orgCheckpoint.Root, checkpoint.Root)
}
func VerifyBeaconStateFinalizedCheckpointNil(t *testing.T, factory getState) {
s, err := factory()
require.NoError(t, err)
checkpoint := s.FinalizedCheckpoint()
require.Equal(t, (*ethpb.Checkpoint)(nil), checkpoint)
epoch := s.FinalizedCheckpointEpoch()
require.Equal(t, types.Epoch(0), epoch)
}
func VerifyBeaconStateFinalizedCheckpoint(t *testing.T, factory getStateWithCheckpoint) {
orgCheckpoint := &ethpb.Checkpoint{Root: make([]byte, fieldparams.RootLength)}
orgCheckpoint.Root[1] = 1
orgCheckpoint.Root[2] = 2
orgCheckpoint.Root[3] = 3
orgCheckpoint.Epoch = 123
s, err := factory(orgCheckpoint)
require.NoError(t, err)
checkpoint := s.FinalizedCheckpoint()
require.DeepEqual(t, orgCheckpoint.Root, checkpoint.Root)
epoch := s.FinalizedCheckpointEpoch()
require.Equal(t, orgCheckpoint.Epoch, epoch)
}