status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

cleanup deposit tests

This commit is contained in:
protolambda 2019-04-15 16:01:17 +10:00
parent 9bb902217d
commit 9591bea42c
No known key found for this signature in database
GPG Key ID: EC89FDBB2B4C7623
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
test_generators/operations/deposits.py
View File

@ -58,11 +58,12 @@ def build_deposit(state,
index=index, index=index,
data=deposit_data, data=deposit_data,
) )
assert spec.verify_merkle_branch(item, proof, spec.DEPOSIT_CONTRACT_TREE_DEPTH, index, get_merkle_root(tuple(deposit_data_leaves)))
return deposit return deposit
def build_deposit_for_index(initial_validator_count: int, index: int) -> Tuple[spec.Deposit, spec.BeaconState, List[spec.Bytes32]]: def build_deposit_for_index(initial_validator_count: int, index: int) -> Tuple[spec.Deposit, spec.BeaconState]:
genesis_deposits = genesis.create_deposits( genesis_deposits = genesis.create_deposits(
keys.pubkeys[:initial_validator_count], keys.pubkeys[:initial_validator_count],
keys.withdrawal_creds[:initial_validator_count] keys.withdrawal_creds[:initial_validator_count]
@ -83,14 +84,12 @@ def build_deposit_for_index(initial_validator_count: int, index: int) -> Tuple[s
state.latest_eth1_data.deposit_root = get_merkle_root(tuple(deposit_data_leaves)) state.latest_eth1_data.deposit_root = get_merkle_root(tuple(deposit_data_leaves))
state.latest_eth1_data.deposit_count = len(deposit_data_leaves) state.latest_eth1_data.deposit_count = len(deposit_data_leaves)
return deposit, state, deposit_data_leaves return deposit, state
@to_dict @to_dict
def valid_deposit(): def valid_deposit():
new_dep, state, leaves = build_deposit_for_index(10, 10) new_dep, state = build_deposit_for_index(10, 10)
state.latest_eth1_data.deposit_root = get_merkle_root(tuple(leaves))
state.latest_eth1_data.deposit_count = len(leaves)
yield 'description', 'valid deposit to add new validator' yield 'description', 'valid deposit to add new validator'
yield 'pre', encode(state, spec.BeaconState) yield 'pre', encode(state, spec.BeaconState)
yield 'deposit', encode(new_dep, spec.Deposit) yield 'deposit', encode(new_dep, spec.Deposit)
@ -100,9 +99,7 @@ def valid_deposit():
@to_dict @to_dict
def valid_topup(): def valid_topup():
new_dep, state, leaves = build_deposit_for_index(10, 3) new_dep, state = build_deposit_for_index(10, 3)
state.latest_eth1_data.deposit_root = get_merkle_root(tuple(leaves))
state.latest_eth1_data.deposit_count = len(leaves)
yield 'description', 'valid deposit to top-up existing validator' yield 'description', 'valid deposit to top-up existing validator'
yield 'pre', encode(state, spec.BeaconState) yield 'pre', encode(state, spec.BeaconState)
yield 'deposit', encode(new_dep, spec.Deposit) yield 'deposit', encode(new_dep, spec.Deposit)
@ -112,9 +109,7 @@ def valid_topup():
@to_dict @to_dict
def invalid_deposit_index(): def invalid_deposit_index():
new_dep, state, leaves = build_deposit_for_index(10, 10) new_dep, state = build_deposit_for_index(10, 10)
state.latest_eth1_data.deposit_root = get_merkle_root(tuple(leaves))
state.latest_eth1_data.deposit_count = len(leaves)
# Mess up deposit index, 1 too small # Mess up deposit index, 1 too small
state.deposit_index = 9 state.deposit_index = 9
@ -125,9 +120,7 @@ def invalid_deposit_index():
@to_dict @to_dict
def invalid_deposit_proof(): def invalid_deposit_proof():
new_dep, state, leaves = build_deposit_for_index(10, 10) new_dep, state = build_deposit_for_index(10, 10)
state.latest_eth1_data.deposit_root = get_merkle_root(tuple(leaves))
state.latest_eth1_data.deposit_count = len(leaves)
# Make deposit proof invalid (at bottom of proof) # Make deposit proof invalid (at bottom of proof)
new_dep.proof[-1] = spec.ZERO_HASH new_dep.proof[-1] = spec.ZERO_HASH