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

Add `test_ejection`

This commit is contained in:
Hsiao-Wei Wang 2019-04-26 14:53:58 +08:00
parent 0de772fc1c
commit 70cd3d2253
No known key found for this signature in database
GPG Key ID: 95B070122902DEA4
1 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
test_libs/pyspec/tests/epoch_processing/test_process_registry_updates.py
View File

@ -12,12 +12,13 @@ from tests.helpers import (
def test_activation(state):
# Mock a new deposit
index = 0
assert is_active_validator(state.validator_registry[index], get_current_epoch(state))
# Mock a new deposit
state.validator_registry[index].activation_eligibility_epoch = spec.FAR_FUTURE_EPOCH
state.validator_registry[index].activation_epoch = spec.FAR_FUTURE_EPOCH
state.validator_registry[index].effective_balance = spec.MAX_EFFECTIVE_BALANCE
assert not is_active_validator(state.validator_registry[index], get_current_epoch(state))
pre_state = deepcopy(state)
@ -33,3 +34,25 @@ def test_activation(state):
)
return pre_state, state
def test_ejection(state):
index = 0
assert is_active_validator(state.validator_registry[index], get_current_epoch(state))
assert state.validator_registry[index].exit_epoch == spec.FAR_FUTURE_EPOCH
# Mock an ejection
state.validator_registry[index].effective_balance = spec.EJECTION_BALANCE
pre_state = deepcopy(state)
for _ in range(spec.ACTIVATION_EXIT_DELAY + 1):
next_epoch(state)
assert state.validator_registry[index].exit_epoch != spec.FAR_FUTURE_EPOCH
assert not is_active_validator(
state.validator_registry[index],
get_current_epoch(state),
)
return pre_state, state