45 lines
1.5 KiB
Nim
45 lines
1.5 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018-2020 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
# Mocking a genesis state
|
|
# ---------------------------------------------------------------
|
|
|
|
import
|
|
# Specs
|
|
../../beacon_chain/spec/[beaconstate, forks, state_transition],
|
|
# Internals
|
|
../../beacon_chain/interop,
|
|
# Mocking procs
|
|
./mock_deposits
|
|
|
|
proc initGenesisState*(
|
|
num_validators: uint64 = 8'u64 * SLOTS_PER_EPOCH,
|
|
beaconStateFork: BeaconStateFork = forkPhase0): ref ForkedHashedBeaconState =
|
|
let deposits = mockGenesisBalancedDeposits(
|
|
validatorCount = num_validators,
|
|
amountInEth = 32, # We create canonical validators with 32 Eth
|
|
flags = {}
|
|
)
|
|
|
|
var cfg = defaultRuntimeConfig
|
|
if beaconStateFork >= forkAltair:
|
|
cfg.ALTAIR_FORK_EPOCH = GENESIS_EPOCH
|
|
|
|
result = (ref ForkedHashedBeaconState)(
|
|
beaconStateFork: forkPhase0,
|
|
hbsPhase0: initialize_hashed_beacon_state_from_eth1(
|
|
cfg, eth1BlockHash, 0, deposits, {}))
|
|
|
|
maybeUpgradeStateToAltair(cfg, result[])
|
|
|
|
doAssert result.beaconStateFork == beaconStateFork
|
|
|
|
when isMainModule:
|
|
# Smoke test
|
|
let state = initGenesisState(num_validators = SLOTS_PER_EPOCH)
|
|
doAssert state.validators.len == SLOTS_PER_EPOCH
|