mirror of
https://github.com/status-im/nimbus-eth2.git
synced 2025-01-21 20:10:36 +00:00
9ee6c3e7ef
There are multiple ways how states can be processed during tests. One is through tests/mocking, the other through tests/xxxutil. Both methods can create genesis states with a set of deposits from mock validators. The private keys for those validators are created differently, though. While tests/mocking generates random keys, testblockutil generates fixed keys. This is a problem when mixing both tests/mocking and testblockutil in a single test case, as they assume that the deposits were generated using the same method. This patch aligns the key generation across tests.
138 lines
4.1 KiB
Nim
138 lines
4.1 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018-2021 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 deposits and genesis deposits
|
|
# ---------------------------------------------------------------
|
|
|
|
import
|
|
# Standard library
|
|
math,
|
|
|
|
# Specs
|
|
../../beacon_chain/spec/[eth2_merkleization, keystore, signatures],
|
|
../../beacon_chain/spec/datatypes/base,
|
|
|
|
# Internals
|
|
../../beacon_chain/extras,
|
|
../../beacon_chain/eth1/merkle_minimal,
|
|
|
|
# Test utilities
|
|
../testblockutil
|
|
|
|
func mockDepositData(
|
|
pubkey: ValidatorPubKey,
|
|
amount: uint64,
|
|
): DepositData =
|
|
# Insecurely use pubkey as withdrawal key
|
|
DepositData(
|
|
pubkey: pubkey,
|
|
withdrawal_credentials: makeWithdrawalCredentials(pubkey),
|
|
amount: amount,
|
|
)
|
|
|
|
func mockDepositData(
|
|
pubkey: ValidatorPubKey,
|
|
privkey: ValidatorPrivKey,
|
|
amount: uint64,
|
|
# withdrawal_credentials: Eth2Digest,
|
|
flags: UpdateFlags = {}
|
|
): DepositData =
|
|
var ret = mockDepositData(pubkey, amount)
|
|
if skipBlsValidation notin flags:
|
|
ret.signature = defaultRuntimeConfig.get_deposit_signature(ret, privkey).toValidatorSig()
|
|
ret
|
|
|
|
template mockGenesisDepositsImpl(
|
|
result: seq[DepositData],
|
|
validatorCount: uint64,
|
|
amount: untyped,
|
|
flags: UpdateFlags = {},
|
|
updateAmount: untyped,
|
|
) =
|
|
# Genesis deposits with varying amounts
|
|
|
|
# NOTE: prefer to er on the side of caution and generate a valid Deposit
|
|
# (it can still be skipped later).
|
|
if skipBlsValidation in flags:
|
|
# 1st loop - build deposit data
|
|
for valIdx in 0 ..< validatorCount:
|
|
# Directly build the Deposit in-place for speed
|
|
result.setLen(valIdx + 1)
|
|
|
|
updateAmount
|
|
|
|
# DepositData
|
|
result[valIdx] =
|
|
mockDepositData(MockPubKeys[valIdx.ValidatorIndex], amount)
|
|
else: # With signing
|
|
var depositsDataHash: seq[Eth2Digest]
|
|
var depositsData: seq[DepositData]
|
|
|
|
# 1st loop - build deposit data
|
|
for valIdx in 0 ..< validatorCount:
|
|
# Directly build the Deposit in-place for speed
|
|
result.setLen(valIdx + 1)
|
|
|
|
updateAmount
|
|
|
|
# DepositData
|
|
result[valIdx] =
|
|
mockDepositData(
|
|
MockPubKeys[valIdx.ValidatorIndex],
|
|
MockPrivKeys[valIdx.ValidatorIndex],
|
|
amount, flags)
|
|
|
|
depositsData.add result[valIdx]
|
|
depositsDataHash.add hash_tree_root(result[valIdx])
|
|
|
|
proc mockGenesisBalancedDeposits*(
|
|
validatorCount: uint64,
|
|
amountInEth: Positive,
|
|
flags: UpdateFlags = {}
|
|
): seq[DepositData] =
|
|
## The amount should be strictly positive
|
|
## - 1 is the minimum deposit amount (MIN_DEPOSIT_AMOUNT)
|
|
## - 16 is the ejection balance (EJECTION_BALANCE)
|
|
## - 32 is the max effective balance (MAX_EFFECTIVE_BALANCE)
|
|
## ETH beyond do not contribute more for staking.
|
|
##
|
|
## Only validators with 32 ETH will be active at genesis
|
|
|
|
let amount = amountInEth.uint64 * 10'u64^9
|
|
mockGenesisDepositsImpl(result, validatorCount,amount,flags):
|
|
discard
|
|
|
|
proc mockUpdateStateForNewDeposit*[T](
|
|
state: var T,
|
|
validator_index: uint64,
|
|
amount: uint64,
|
|
# withdrawal_credentials: Eth2Digest
|
|
flags: UpdateFlags
|
|
): Deposit =
|
|
|
|
|
|
# TODO withdrawal credentials
|
|
|
|
result.data = mockDepositData(
|
|
MockPubKeys[validator_index.ValidatorIndex],
|
|
MockPrivKeys[validator_index.ValidatorIndex],
|
|
amount,
|
|
# withdrawal_credentials: Eth2Digest
|
|
flags
|
|
)
|
|
|
|
var result_seq = @[result]
|
|
attachMerkleProofs(result_seq)
|
|
result.proof = result_seq[0].proof
|
|
|
|
# TODO: this logic from the eth2.0-specs test suite seems strange
|
|
# but confirmed by running it
|
|
state.eth1_deposit_index = 0
|
|
state.eth1_data.deposit_root =
|
|
hash_tree_root(List[DepositData, 2'i64^DEPOSIT_CONTRACT_TREE_DEPTH](@[result.data]))
|
|
state.eth1_data.deposit_count = 1
|