98 lines
3.0 KiB
Python
98 lines
3.0 KiB
Python
|
from copy import deepcopy
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import build.phase0.spec as spec
|
||
|
|
from build.phase0.spec import (
|
||
|
|
get_balance,
|
||
|
|
get_current_epoch,
|
||
|
|
process_proposer_slashing,
|
||
|
|
)
|
||
|
|
from tests.phase0.helpers import (
|
||
|
|
get_valid_proposer_slashing,
|
||
|
|
)
|
||
|
|
|
||
|
|
# mark entire file as 'header'
|
||
|
|
pytestmark = pytest.mark.proposer_slashings
|
||
|
|
|
||
|
|
|
||
|
|
def run_proposer_slashing_processing(state, proposer_slashing, valid=True):
|
||
|
|
"""
|
||
|
|
Run ``process_proposer_slashing`` returning the pre and post state.
|
||
|
|
If ``valid == False``, run expecting ``AssertionError``
|
||
|
|
"""
|
||
|
|
post_state = deepcopy(state)
|
||
|
|
|
||
|
|
if not valid:
|
||
|
|
with pytest.raises(AssertionError):
|
||
|
|
process_proposer_slashing(post_state, proposer_slashing)
|
||
|
|
return state, None
|
||
|
|
|
||
|
|
process_proposer_slashing(post_state, proposer_slashing)
|
||
|
|
|
||
|
|
slashed_validator = post_state.validator_registry[proposer_slashing.proposer_index]
|
||
|
|
assert not slashed_validator.initiated_exit
|
||
|
|
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)
|
||
|
|
)
|
||
|
|
|
||
|
|
return state, post_state
|
||
|
|
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
def test_proposer_is_withdrawn(state):
|
||
|
|
proposer_slashing = get_valid_proposer_slashing(state)
|
||
|
|
|
||
|
|
# set proposer withdrawable_epoch in past
|
||
|
|
current_epoch = get_current_epoch(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
|