mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
cc741ed8af
* begin state service * begin on the state trie idea * created beacon state structure * add in the full clone getter * return by value instead * add all setters * new state setters are being completed * arrays roots exposed * close to finishing all these headerssss * functionality complete * added in proto benchmark test * test for compatibility * add test for compat * comments fixed * add clone * add clone * remove underlying copies * make it immutable * integrate it into chainservice * revert * wrap up comments for package * address all comments and godocs * address all comments * clone the pending attestation properly * properly clone remaining items * tests pass fixed bug * begin using it instead of head state * prevent nil pointer exceptions * begin using new struct in db * integrated new type into db package * add proper nil checks * using new state in archiver * refactored much of core * editing all the precompute functions * done with most core refactor * fixed up some bugs in the clone comparisons * append current epoch atts * add missing setters * add new setters * fix other core methods * fix up transition * main service and forkchoice * fix rpc * integrated to powchain * some more changes * fix build * improve processing of deposits * fix error * prevent panic * comment * fix process att * gaz * fix up att process * resolve existing review comments * resolve another batch of gh comments * resolve broken cpt state * revise testutil to use the new state * begin updating the state transition func to pass in more compartmentalized args * finish editing transition function to return errors * block operations pretty much done with refactor * state transition fully refactored * got epoch processing completed * fix build in fork choice * fixing more of the build * fix up broken sync package * it builds nowww it buildssss * revert registry changes * Recompute on Read (#4627) * compute on read * fix up eth1 data votes * looking into slashings bug introduced in core/ * able to advance more slots * add logging * can now sync with testnet yay * remove the leaves algorithm and other merkle imports * expose initialize unsafe funcs * Update beacon-chain/db/kv/state.go * lint Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> * More Optimizations for New State (#4641) * map optimization * more optimizations * use a custom hasher * comment * block operations optimizations * Update beacon-chain/state/types.go Co-Authored-By: Raul Jordan <raul@prysmaticlabs.com> * fixed up various operations to use the validator index map access Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> * archiver tests pass * fixing cache tests * cache tests passing * edited validator tests * powchain tests passing * halfway thru sync tests * more sync test fixes * add in tests for state/ * working through rpc tests * assignments tests passed * almost done with rpc/beacon tests * resolved painful validator test * fixed up even more tests * resolve tests * fix build * reduce a randao mixes copy * fixes under //beacon-chain/blockchain/... * build //beacon-chain/core/... * fixes * Runtime Optimizations (#4648) * parallelize shuffling * clean up * lint * fix build * use callback to read from registry * fix array roots and size map * new improvements * reduce hash allocs * improved shuffling * terence's review * use different method * raul's comment * new array roots * remove clone in pre-compute * Update beacon-chain/state/types.go Co-Authored-By: Raul Jordan <raul@prysmaticlabs.com> * raul's review * lint * fix build issues * fix visibility Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> * fix visibility * build works for all * fix blockchain test * fix a few tests * fix more tests * update validator in slashing * archiver passing * fixed rpc/validator * progress on core tests * resolve broken rpc tests * blockchain tests passed * fix up some tests in core * fix message diff * remove unnecessary save * Save validator after slashing * Update validators one by one * another update * fix everything * fix more precompute tests * fix blocks tests * more elegant fix * more helper fixes * change back ? * fix test * fix skip slot * fix test * reset caches * fix testutil * raceoff fixed * passing * Retrieve cached state in the beginning * lint * Fixed tests part 1 * Fixed rest of the tests * Minor changes to avoid copying, small refactor to reduce deplicated code * Handle att req for slot 0 * New beacon state: Only populate merkle layers as needed, copy merkle layers on copy/clone. (#4689) * Only populate merkle layers as needed, copy merkle layers on copy/clone. * use custom copy * Make maps of correct size * slightly fast, doesn't wait for lock Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com> * Target root can't be 0x00 * Don't use cache for current slot (may not be the right fix) * fixed up tests * Remove some copy for init sync. Not sure if it is safe enough for runtime though... testing... * Align with prev logic for process slots cachedState.Slot() < slot * Fix Initial Sync Flag (#4692) * fixes * fix up some test failures due to lack of nil checks * fix up some test failures due to lack of nil checks * fix up imports * revert some changes * imports Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> * resolving further conflicts * Better skip slot cache (#4694) * Return copy of skip slot cache state, disable skip slot cache on sync * fix * Fix pruning * fix up issues with broken tests Co-authored-by: Nishant Das <nish1993@hotmail.com> Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> Co-authored-by: shayzluf <thezluf@gmail.com> Co-authored-by: terence tsao <terence@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
200 lines
6.1 KiB
Go
200 lines
6.1 KiB
Go
package testing
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/epoch/precompute"
|
|
opfeed "github.com/prysmaticlabs/prysm/beacon-chain/core/feed/operation"
|
|
statefeed "github.com/prysmaticlabs/prysm/beacon-chain/core/feed/state"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db"
|
|
stateTrie "github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/event"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// ChainService defines the mock interface for testing
|
|
type ChainService struct {
|
|
State *stateTrie.BeaconState
|
|
Root []byte
|
|
Block *ethpb.SignedBeaconBlock
|
|
FinalizedCheckPoint *ethpb.Checkpoint
|
|
CurrentJustifiedCheckPoint *ethpb.Checkpoint
|
|
PreviousJustifiedCheckPoint *ethpb.Checkpoint
|
|
BlocksReceived []*ethpb.SignedBeaconBlock
|
|
Balance *precompute.Balance
|
|
Genesis time.Time
|
|
Fork *pb.Fork
|
|
DB db.Database
|
|
stateNotifier statefeed.Notifier
|
|
opNotifier opfeed.Notifier
|
|
}
|
|
|
|
// StateNotifier mocks the same method in the chain service.
|
|
func (ms *ChainService) StateNotifier() statefeed.Notifier {
|
|
if ms.stateNotifier == nil {
|
|
ms.stateNotifier = &MockStateNotifier{}
|
|
}
|
|
return ms.stateNotifier
|
|
}
|
|
|
|
// MockStateNotifier mocks the state notifier.
|
|
type MockStateNotifier struct {
|
|
feed *event.Feed
|
|
}
|
|
|
|
// StateFeed returns a state feed.
|
|
func (msn *MockStateNotifier) StateFeed() *event.Feed {
|
|
if msn.feed == nil {
|
|
msn.feed = new(event.Feed)
|
|
}
|
|
return msn.feed
|
|
}
|
|
|
|
// OperationNotifier mocks the same method in the chain service.
|
|
func (ms *ChainService) OperationNotifier() opfeed.Notifier {
|
|
if ms.opNotifier == nil {
|
|
ms.opNotifier = &MockOperationNotifier{}
|
|
}
|
|
return ms.opNotifier
|
|
}
|
|
|
|
// MockOperationNotifier mocks the operation notifier.
|
|
type MockOperationNotifier struct {
|
|
feed *event.Feed
|
|
}
|
|
|
|
// OperationFeed returns an operation feed.
|
|
func (mon *MockOperationNotifier) OperationFeed() *event.Feed {
|
|
if mon.feed == nil {
|
|
mon.feed = new(event.Feed)
|
|
}
|
|
return mon.feed
|
|
}
|
|
|
|
// ReceiveBlock mocks ReceiveBlock method in chain service.
|
|
func (ms *ChainService) ReceiveBlock(ctx context.Context, block *ethpb.SignedBeaconBlock) error {
|
|
return nil
|
|
}
|
|
|
|
// ReceiveBlockNoVerify mocks ReceiveBlockNoVerify method in chain service.
|
|
func (ms *ChainService) ReceiveBlockNoVerify(ctx context.Context, block *ethpb.SignedBeaconBlock) error {
|
|
return nil
|
|
}
|
|
|
|
// ReceiveBlockNoPubsub mocks ReceiveBlockNoPubsub method in chain service.
|
|
func (ms *ChainService) ReceiveBlockNoPubsub(ctx context.Context, block *ethpb.SignedBeaconBlock) error {
|
|
return nil
|
|
}
|
|
|
|
// ReceiveBlockNoPubsubForkchoice mocks ReceiveBlockNoPubsubForkchoice method in chain service.
|
|
func (ms *ChainService) ReceiveBlockNoPubsubForkchoice(ctx context.Context, block *ethpb.SignedBeaconBlock) error {
|
|
if ms.State == nil {
|
|
ms.State = &stateTrie.BeaconState{}
|
|
}
|
|
if !bytes.Equal(ms.Root, block.Block.ParentRoot) {
|
|
return errors.Errorf("wanted %#x but got %#x", ms.Root, block.Block.ParentRoot)
|
|
}
|
|
if err := ms.State.SetSlot(block.Block.Slot); err != nil {
|
|
return err
|
|
}
|
|
ms.BlocksReceived = append(ms.BlocksReceived, block)
|
|
signingRoot, err := ssz.HashTreeRoot(block.Block)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if ms.DB != nil {
|
|
if err := ms.DB.SaveBlock(ctx, block); err != nil {
|
|
return err
|
|
}
|
|
logrus.Infof("Saved block with root: %#x at slot %d", signingRoot, block.Block.Slot)
|
|
}
|
|
ms.Root = signingRoot[:]
|
|
ms.Block = block
|
|
return nil
|
|
}
|
|
|
|
// HeadSlot mocks HeadSlot method in chain service.
|
|
func (ms *ChainService) HeadSlot() uint64 {
|
|
if ms.State == nil {
|
|
return 0
|
|
}
|
|
return ms.State.Slot()
|
|
}
|
|
|
|
// HeadRoot mocks HeadRoot method in chain service.
|
|
func (ms *ChainService) HeadRoot(ctx context.Context) ([]byte, error) {
|
|
return ms.Root, nil
|
|
|
|
}
|
|
|
|
// HeadBlock mocks HeadBlock method in chain service.
|
|
func (ms *ChainService) HeadBlock() *ethpb.SignedBeaconBlock {
|
|
return ms.Block
|
|
}
|
|
|
|
// HeadState mocks HeadState method in chain service.
|
|
func (ms *ChainService) HeadState(context.Context) (*stateTrie.BeaconState, error) {
|
|
return ms.State, nil
|
|
}
|
|
|
|
// CurrentFork mocks HeadState method in chain service.
|
|
func (ms *ChainService) CurrentFork() *pb.Fork {
|
|
return ms.Fork
|
|
}
|
|
|
|
// FinalizedCheckpt mocks FinalizedCheckpt method in chain service.
|
|
func (ms *ChainService) FinalizedCheckpt() *ethpb.Checkpoint {
|
|
return ms.FinalizedCheckPoint
|
|
}
|
|
|
|
// CurrentJustifiedCheckpt mocks CurrentJustifiedCheckpt method in chain service.
|
|
func (ms *ChainService) CurrentJustifiedCheckpt() *ethpb.Checkpoint {
|
|
return ms.CurrentJustifiedCheckPoint
|
|
}
|
|
|
|
// PreviousJustifiedCheckpt mocks PreviousJustifiedCheckpt method in chain service.
|
|
func (ms *ChainService) PreviousJustifiedCheckpt() *ethpb.Checkpoint {
|
|
return ms.PreviousJustifiedCheckPoint
|
|
}
|
|
|
|
// ReceiveAttestation mocks ReceiveAttestation method in chain service.
|
|
func (ms *ChainService) ReceiveAttestation(context.Context, *ethpb.Attestation) error {
|
|
return nil
|
|
}
|
|
|
|
// ReceiveAttestationNoPubsub mocks ReceiveAttestationNoPubsub method in chain service.
|
|
func (ms *ChainService) ReceiveAttestationNoPubsub(context.Context, *ethpb.Attestation) error {
|
|
return nil
|
|
}
|
|
|
|
// HeadValidatorsIndices mocks the same method in the chain service.
|
|
func (ms *ChainService) HeadValidatorsIndices(epoch uint64) ([]uint64, error) {
|
|
if ms.State == nil {
|
|
return []uint64{}, nil
|
|
}
|
|
return helpers.ActiveValidatorIndices(ms.State, epoch)
|
|
}
|
|
|
|
// HeadSeed mocks the same method in the chain service.
|
|
func (ms *ChainService) HeadSeed(epoch uint64) ([32]byte, error) {
|
|
return helpers.Seed(ms.State, epoch, params.BeaconConfig().DomainBeaconAttester)
|
|
}
|
|
|
|
// GenesisTime mocks the same method in the chain service.
|
|
func (ms *ChainService) GenesisTime() time.Time {
|
|
return ms.Genesis
|
|
}
|
|
|
|
// Participation mocks the same method in the chain service.
|
|
func (ms *ChainService) Participation(epoch uint64) *precompute.Balance {
|
|
return ms.Balance
|
|
}
|