eth2.0-specs/test_libs/pyspec/eth2spec/test/test_finality.py

197 lines
7.6 KiB
Python
Raw Normal View History

from copy import deepcopy
from typing import List
2019-06-06 21:30:40 +00:00
from eth2spec.test.context import spec_state_test, never_bls, with_all_phases
from eth2spec.test.helpers.state import next_epoch, state_transition_and_sign_block
2019-05-30 20:57:18 +00:00
from eth2spec.test.helpers.block import build_empty_block_for_next_slot, apply_empty_block
from eth2spec.test.helpers.attestations import get_valid_attestation
def check_finality(spec,
state,
prev_state,
current_justified_changed,
previous_justified_changed,
finalized_changed):
if current_justified_changed:
2019-06-22 18:00:26 +00:00
assert state.current_justified_checkpoint.epoch > prev_state.current_justified_checkpoint.epoch
assert state.current_justified_checkpoint.root != prev_state.current_justified_checkpoint.root
else:
2019-06-22 18:00:26 +00:00
assert state.current_justified_checkpoint == prev_state.current_justified_checkpoint
if previous_justified_changed:
2019-06-22 18:00:26 +00:00
assert state.previous_justified_checkpoint.epoch > prev_state.previous_justified_checkpoint.epoch
assert state.previous_justified_checkpoint.root != prev_state.previous_justified_checkpoint.root
else:
2019-06-22 18:00:26 +00:00
assert state.previous_justified_checkpoint == prev_state.previous_justified_checkpoint
if finalized_changed:
2019-06-22 18:00:26 +00:00
assert state.finalized_checkpoint.epoch > prev_state.finalized_checkpoint.epoch
assert state.finalized_checkpoint.root != prev_state.finalized_checkpoint.root
else:
2019-06-22 18:00:26 +00:00
assert state.finalized_checkpoint == prev_state.finalized_checkpoint
2019-05-30 20:57:18 +00:00
def next_epoch_with_attestations(spec,
state,
2019-04-22 15:00:01 +00:00
fill_cur_epoch,
fill_prev_epoch):
post_state = deepcopy(state)
blocks = []
2019-04-24 17:45:41 +00:00
for _ in range(spec.SLOTS_PER_EPOCH):
2019-05-30 20:57:18 +00:00
block = build_empty_block_for_next_slot(spec, post_state)
2019-04-22 15:00:01 +00:00
if fill_cur_epoch:
slot_to_attest = post_state.slot - spec.MIN_ATTESTATION_INCLUSION_DELAY + 1
2019-05-30 20:57:18 +00:00
if slot_to_attest >= spec.get_epoch_start_slot(spec.get_current_epoch(post_state)):
cur_attestation = get_valid_attestation(spec, post_state, slot_to_attest)
2019-04-22 15:00:01 +00:00
block.body.attestations.append(cur_attestation)
2019-04-23 19:32:41 +00:00
if fill_prev_epoch:
slot_to_attest = post_state.slot - spec.SLOTS_PER_EPOCH + 1
2019-05-30 20:57:18 +00:00
prev_attestation = get_valid_attestation(spec, post_state, slot_to_attest)
2019-04-23 19:32:41 +00:00
block.body.attestations.append(prev_attestation)
state_transition_and_sign_block(spec, post_state, block)
2019-04-22 15:00:01 +00:00
blocks.append(block)
return state, blocks, post_state
2019-06-06 21:30:40 +00:00
@with_all_phases
2019-05-20 17:38:18 +00:00
@never_bls
@spec_state_test
2019-05-30 20:57:18 +00:00
def test_finality_rule_4(spec, state):
yield 'pre', state
blocks = []
for epoch in range(4):
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, True, False)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
2019-04-24 17:44:03 +00:00
# justification/finalization skipped at GENESIS_EPOCH
if epoch == 0:
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, False, False, False)
2019-04-24 17:44:03 +00:00
# justification/finalization skipped at GENESIS_EPOCH + 1
elif epoch == 1:
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, False, False, False)
elif epoch == 2:
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, False, False)
elif epoch >= 3:
# rule 4 of finality
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, True, True)
2019-06-22 18:00:26 +00:00
assert state.finalized_checkpoint == prev_state.current_justified_checkpoint
yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state
2019-06-06 21:30:40 +00:00
@with_all_phases
2019-05-20 17:38:18 +00:00
@never_bls
@spec_state_test
2019-05-30 20:57:18 +00:00
def test_finality_rule_1(spec, state):
# get past first two epochs that finality does not run on
2019-05-30 20:57:18 +00:00
next_epoch(spec, state)
apply_empty_block(spec, state)
next_epoch(spec, state)
apply_empty_block(spec, state)
yield 'pre', state
blocks = []
for epoch in range(3):
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, False, True)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
if epoch == 0:
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, False, False)
elif epoch == 1:
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, True, False)
elif epoch == 2:
# finalized by rule 1
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, True, True)
2019-06-22 18:00:26 +00:00
assert state.finalized_checkpoint == prev_state.previous_justified_checkpoint
yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state
2019-04-22 15:00:01 +00:00
2019-06-06 21:30:40 +00:00
@with_all_phases
2019-05-20 17:38:18 +00:00
@never_bls
@spec_state_test
2019-05-30 20:57:18 +00:00
def test_finality_rule_2(spec, state):
# get past first two epochs that finality does not run on
2019-05-30 20:57:18 +00:00
next_epoch(spec, state)
apply_empty_block(spec, state)
next_epoch(spec, state)
apply_empty_block(spec, state)
yield 'pre', state
blocks = []
for epoch in range(3):
2019-04-22 15:00:01 +00:00
if epoch == 0:
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, True, False)
check_finality(spec, state, prev_state, True, False, False)
elif epoch == 1:
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, False, False)
check_finality(spec, state, prev_state, False, True, False)
elif epoch == 2:
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, False, True)
2019-04-22 15:00:01 +00:00
# finalized by rule 2
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, False, True)
2019-06-22 18:00:26 +00:00
assert state.finalized_checkpoint == prev_state.previous_justified_checkpoint
2019-04-23 19:32:41 +00:00
blocks += new_blocks
yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state
2019-04-22 15:00:01 +00:00
2019-06-06 21:30:40 +00:00
@with_all_phases
2019-05-20 17:38:18 +00:00
@never_bls
@spec_state_test
2019-05-30 20:57:18 +00:00
def test_finality_rule_3(spec, state):
2019-04-23 19:32:41 +00:00
"""
Test scenario described here
https://github.com/ethereum/eth2.0-specs/issues/611#issuecomment-463612892
"""
2019-04-22 15:00:01 +00:00
# get past first two epochs that finality does not run on
2019-05-30 20:57:18 +00:00
next_epoch(spec, state)
apply_empty_block(spec, state)
next_epoch(spec, state)
apply_empty_block(spec, state)
yield 'pre', state
2019-04-22 15:00:01 +00:00
blocks = []
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, True, False)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, False, False)
2019-04-22 15:00:01 +00:00
2019-04-24 17:44:03 +00:00
# In epoch N, JE is set to N, prev JE is set to N-1
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, True, False)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, True, True)
2019-04-22 15:00:01 +00:00
2019-04-24 17:44:03 +00:00
# In epoch N+1, JE is N, prev JE is N-1, and not enough messages get in to do anything
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, False, False)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, False, True, False)
2019-04-22 15:00:01 +00:00
2019-04-24 17:44:03 +00:00
# In epoch N+2, JE is N, prev JE is N, and enough messages from the previous epoch get in to justify N+1.
# N+1 now becomes the JE. Not enough messages from epoch N+2 itself get in to justify N+2
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, False, True)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
# rule 2
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, False, True)
2019-04-22 15:00:01 +00:00
2019-04-24 17:44:03 +00:00
# In epoch N+3, LJE is N+1, prev LJE is N, and enough messages get in to justify epochs N+2 and N+3.
2019-05-30 20:57:18 +00:00
prev_state, new_blocks, state = next_epoch_with_attestations(spec, state, True, True)
2019-04-23 19:32:41 +00:00
blocks += new_blocks
# rule 3
2019-05-30 20:57:18 +00:00
check_finality(spec, state, prev_state, True, True, True)
2019-06-22 18:00:26 +00:00
assert state.finalized_checkpoint == prev_state.current_justified_checkpoint
2019-04-22 15:00:01 +00:00
yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state