nimbus-eth2/tests/mocking/mock_genesis.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