status-im/eth2.0-specs
2
0
Fork 0
mirror of https://github.com/status-im/eth2.0-specs.git synced 2025-01-12 19:54:34 +00:00
Code Issues Projects Releases Wiki Activity

clean import, update proposer slashing test

Browse Source
This commit is contained in:
protolambda 2019-05-06 16:30:13 +02:00
parent 90a56e2f5b
commit 802f2710c3
No known key found for this signature in database
GPG Key ID: EC89FDBB2B4C7623
2 changed files with 29 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test_libs/pyspec/tests/block_processing/test_process_deposit.py
View File

@ -1,4 +1,3 @@
from copy import deepcopy
import pytest
import eth2spec.phase0.spec as spec

57
test_libs/pyspec/tests/block_processing/test_process_proposer_slashing.py
View File

@ -1,4 +1,3 @@
from copy import deepcopy
import pytest
import eth2spec.phase0.spec as spec
@ -11,78 +10,82 @@ from tests.helpers import (
get_valid_proposer_slashing,
)
# mark entire file as 'proposer_slashings'
pytestmark = pytest.mark.proposer_slashings
from .block_test_helpers import spec_state_test
def run_proposer_slashing_processing(state, proposer_slashing, valid=True):
"""
Run ``process_proposer_slashing`` returning the pre and post state.
Run ``process_proposer_slashing``, yielding:
- pre-state ('pre')
- proposer_slashing ('proposer_slashing')
- post-state ('post').
If ``valid == False``, run expecting ``AssertionError``
"""
post_state = deepcopy(state)
pre_proposer_balance = get_balance(state, proposer_slashing.proposer_index)
yield 'pre', state
yield 'proposer_slashing', proposer_slashing
if not valid:
with pytest.raises(AssertionError):
process_proposer_slashing(post_state, proposer_slashing)
return state, None
process_proposer_slashing(state, proposer_slashing)
yield 'post', None
return
process_proposer_slashing(post_state, proposer_slashing)
process_proposer_slashing(state, proposer_slashing)
yield 'post', state
slashed_validator = post_state.validator_registry[proposer_slashing.proposer_index]
# check if slashed
slashed_validator = state.validator_registry[proposer_slashing.proposer_index]
assert slashed_validator.slashed
assert slashed_validator.exit_epoch < spec.FAR_FUTURE_EPOCH
assert slashed_validator.withdrawable_epoch < spec.FAR_FUTURE_EPOCH
# lost whistleblower reward
assert (
get_balance(post_state, proposer_slashing.proposer_index) <
get_balance(state, proposer_slashing.proposer_index)
get_balance(state, proposer_slashing.proposer_index) <
pre_proposer_balance
)
return state, post_state
@spec_state_test
def test_success(state):
proposer_slashing = get_valid_proposer_slashing(state)
pre_state, post_state = run_proposer_slashing_processing(state, proposer_slashing)
return pre_state, proposer_slashing, post_state
yield from run_proposer_slashing_processing(state, proposer_slashing)
@spec_state_test
def test_epochs_are_different(state):
proposer_slashing = get_valid_proposer_slashing(state)
# set slots to be in different epochs
proposer_slashing.header_2.slot += spec.SLOTS_PER_EPOCH
pre_state, post_state = run_proposer_slashing_processing(state, proposer_slashing, False)
return pre_state, proposer_slashing, post_state
yield from run_proposer_slashing_processing(state, proposer_slashing, False)
@spec_state_test
def test_headers_are_same(state):
proposer_slashing = get_valid_proposer_slashing(state)
# set headers to be the same
proposer_slashing.header_2 = proposer_slashing.header_1
pre_state, post_state = run_proposer_slashing_processing(state, proposer_slashing, False)
return pre_state, proposer_slashing, post_state
yield from run_proposer_slashing_processing(state, proposer_slashing, False)
@spec_state_test
def test_proposer_is_slashed(state):
proposer_slashing = get_valid_proposer_slashing(state)
# set proposer to slashed
state.validator_registry[proposer_slashing.proposer_index].slashed = True
pre_state, post_state = run_proposer_slashing_processing(state, proposer_slashing, False)
return pre_state, proposer_slashing, post_state
yield from run_proposer_slashing_processing(state, proposer_slashing, False)
@spec_state_test
def test_proposer_is_withdrawn(state):
proposer_slashing = get_valid_proposer_slashing(state)
@ -91,6 +94,4 @@ def test_proposer_is_withdrawn(state):
proposer_index = proposer_slashing.proposer_index
state.validator_registry[proposer_index].withdrawable_epoch = current_epoch - 1
pre_state, post_state = run_proposer_slashing_processing(state, proposer_slashing, False)
return pre_state, proposer_slashing, post_state
yield from run_proposer_slashing_processing(state, proposer_slashing, False)