implement assertion sanity check suggested in PR

This commit is contained in:
protolambda 2019-04-15 23:55:32 +10:00
parent c3d321ada8
commit d2b7a8f5de
No known key found for this signature in database
GPG Key ID: EC89FDBB2B4C7623
1 changed files with 14 additions and 2 deletions

View File

@ -116,7 +116,13 @@ def invalid_deposit_index():
yield 'description', 'invalid deposit index'
yield 'pre', encode(state, spec.BeaconState)
yield 'deposit', encode(new_dep, spec.Deposit)
yield 'post', None
try:
spec.process_deposit(state, new_dep)
except AssertionError:
# expected
yield 'post', None
return
raise Exception('invalid_deposit_index has unexpectedly allowed deposit')
@to_dict
@ -128,7 +134,13 @@ def invalid_deposit_proof():
yield 'description', 'invalid deposit proof'
yield 'pre', encode(state, spec.BeaconState)
yield 'deposit', encode(new_dep, spec.Deposit)
yield 'post', None
try:
spec.process_deposit(state, new_dep)
except AssertionError:
# expected
yield 'post', None
return
raise Exception('invalid_deposit_index has unexpectedly allowed deposit')
@to_tuple