mirror of
https://github.com/status-im/eth2.0-specs.git
synced 2025-01-12 19:54:34 +00:00
test rule 3
This commit is contained in:
parent
9f4e59b0bc
commit
edb24ce9dd
@ -1682,12 +1682,16 @@ def process_justification_and_finalization(state: BeaconState) -> None:
|
|||||||
state.previous_justified_root = state.current_justified_root
|
state.previous_justified_root = state.current_justified_root
|
||||||
state.justification_bitfield = (state.justification_bitfield << 1) % 2**64
|
state.justification_bitfield = (state.justification_bitfield << 1) % 2**64
|
||||||
previous_boundary_attesting_balance = get_attesting_balance(state, get_previous_epoch_boundary_attestations(state))
|
previous_boundary_attesting_balance = get_attesting_balance(state, get_previous_epoch_boundary_attestations(state))
|
||||||
|
print(previous_boundary_attesting_balance)
|
||||||
if previous_boundary_attesting_balance * 3 >= get_previous_total_balance(state) * 2:
|
if previous_boundary_attesting_balance * 3 >= get_previous_total_balance(state) * 2:
|
||||||
|
print("prev success")
|
||||||
state.current_justified_epoch = get_previous_epoch(state)
|
state.current_justified_epoch = get_previous_epoch(state)
|
||||||
state.current_justified_root = get_block_root(state, get_epoch_start_slot(state.current_justified_epoch))
|
state.current_justified_root = get_block_root(state, get_epoch_start_slot(state.current_justified_epoch))
|
||||||
state.justification_bitfield |= (1 << 1)
|
state.justification_bitfield |= (1 << 1)
|
||||||
current_boundary_attesting_balance = get_attesting_balance(state, get_current_epoch_boundary_attestations(state))
|
current_boundary_attesting_balance = get_attesting_balance(state, get_current_epoch_boundary_attestations(state))
|
||||||
|
print(current_boundary_attesting_balance)
|
||||||
if current_boundary_attesting_balance * 3 >= get_current_total_balance(state) * 2:
|
if current_boundary_attesting_balance * 3 >= get_current_total_balance(state) * 2:
|
||||||
|
print("cur success")
|
||||||
state.current_justified_epoch = get_current_epoch(state)
|
state.current_justified_epoch = get_current_epoch(state)
|
||||||
state.current_justified_root = get_block_root(state, get_epoch_start_slot(state.current_justified_epoch))
|
state.current_justified_root = get_block_root(state, get_epoch_start_slot(state.current_justified_epoch))
|
||||||
state.justification_bitfield |= (1 << 0)
|
state.justification_bitfield |= (1 << 0)
|
||||||
|
@ -47,11 +47,50 @@ def check_finality(state,
|
|||||||
assert state.finalized_root == prev_state.finalized_root
|
assert state.finalized_root == prev_state.finalized_root
|
||||||
|
|
||||||
|
|
||||||
|
def next_epoch_with_attestations(state,
|
||||||
|
fill_cur_epoch,
|
||||||
|
fill_prev_epoch):
|
||||||
|
post_state = deepcopy(state)
|
||||||
|
blocks = []
|
||||||
|
for slot in range(spec.SLOTS_PER_EPOCH):
|
||||||
|
print("slot: %s", post_state.slot)
|
||||||
|
block = build_empty_block_for_next_slot(post_state)
|
||||||
|
if fill_prev_epoch:
|
||||||
|
print("prev")
|
||||||
|
slot_to_attest = post_state.slot - spec.SLOTS_PER_EPOCH + 1
|
||||||
|
prev_attestation = get_valid_attestation(post_state, slot_to_attest)
|
||||||
|
block.body.attestations.append(prev_attestation)
|
||||||
|
|
||||||
|
if fill_cur_epoch:
|
||||||
|
print("cur")
|
||||||
|
slot_to_attest = post_state.slot - spec.MIN_ATTESTATION_INCLUSION_DELAY + 1
|
||||||
|
if slot_to_attest >= get_epoch_start_slot(get_current_epoch(post_state)):
|
||||||
|
cur_attestation = get_valid_attestation(post_state, slot_to_attest)
|
||||||
|
fill_aggregate_attestation(post_state, cur_attestation)
|
||||||
|
block.body.attestations.append(cur_attestation)
|
||||||
|
|
||||||
|
state_transition(post_state, block)
|
||||||
|
blocks.append(block)
|
||||||
|
|
||||||
|
# if fill_prev_epoch:
|
||||||
|
# assert len(post_state.previous_epoch_attestations) >= 0
|
||||||
|
# else:
|
||||||
|
# assert len(post_state.previous_epoch_attestations) == 0
|
||||||
|
|
||||||
|
# if fill_cur_epoch:
|
||||||
|
# assert len(post_state.current_epoch_attestations) >= 0
|
||||||
|
# else:
|
||||||
|
# assert len(post_state.current_epoch_attestations) == 0
|
||||||
|
|
||||||
|
return state, blocks, post_state
|
||||||
|
|
||||||
|
|
||||||
def test_finality_from_genesis_rule_4(state):
|
def test_finality_from_genesis_rule_4(state):
|
||||||
test_state = deepcopy(state)
|
test_state = deepcopy(state)
|
||||||
|
|
||||||
blocks = []
|
blocks = []
|
||||||
for epoch in range(6):
|
for epoch in range(6):
|
||||||
|
prev_state = deepcopy(test_state)
|
||||||
old_current_justified_epoch = test_state.current_justified_epoch
|
old_current_justified_epoch = test_state.current_justified_epoch
|
||||||
old_current_justified_root = test_state.current_justified_root
|
old_current_justified_root = test_state.current_justified_root
|
||||||
for slot in range(spec.SLOTS_PER_EPOCH):
|
for slot in range(spec.SLOTS_PER_EPOCH):
|
||||||
@ -67,14 +106,14 @@ def test_finality_from_genesis_rule_4(state):
|
|||||||
blocks.append(block)
|
blocks.append(block)
|
||||||
|
|
||||||
if epoch == 0:
|
if epoch == 0:
|
||||||
check_finality(test_state, state, False, False, False)
|
check_finality(test_state, prev_state, False, False, False)
|
||||||
elif epoch == 1:
|
elif epoch == 1:
|
||||||
check_finality(test_state, state, False, False, False)
|
check_finality(test_state, prev_state, False, False, False)
|
||||||
elif epoch == 2:
|
elif epoch == 2:
|
||||||
check_finality(test_state, state, True, False, False)
|
check_finality(test_state, prev_state, True, False, False)
|
||||||
elif epoch >= 3:
|
elif epoch >= 3:
|
||||||
# rule 4 of finaliy
|
# rule 4 of finaliy
|
||||||
check_finality(test_state, state, True, True, True)
|
check_finality(test_state, prev_state, True, True, True)
|
||||||
assert test_state.finalized_epoch == old_current_justified_epoch
|
assert test_state.finalized_epoch == old_current_justified_epoch
|
||||||
assert test_state.finalized_root == old_current_justified_root
|
assert test_state.finalized_root == old_current_justified_root
|
||||||
|
|
||||||
@ -90,6 +129,7 @@ def test_finality_rule_1(state):
|
|||||||
|
|
||||||
blocks = []
|
blocks = []
|
||||||
for epoch in range(3):
|
for epoch in range(3):
|
||||||
|
prev_state = deepcopy(test_state)
|
||||||
old_previous_justified_epoch = test_state.previous_justified_epoch
|
old_previous_justified_epoch = test_state.previous_justified_epoch
|
||||||
old_previous_justified_root = test_state.previous_justified_root
|
old_previous_justified_root = test_state.previous_justified_root
|
||||||
for slot in range(spec.SLOTS_PER_EPOCH):
|
for slot in range(spec.SLOTS_PER_EPOCH):
|
||||||
@ -106,15 +146,17 @@ def test_finality_rule_1(state):
|
|||||||
blocks.append(block)
|
blocks.append(block)
|
||||||
|
|
||||||
if epoch == 0:
|
if epoch == 0:
|
||||||
check_finality(test_state, state, True, False, False)
|
check_finality(test_state, prev_state, True, False, False)
|
||||||
elif epoch == 1:
|
elif epoch == 1:
|
||||||
check_finality(test_state, state, True, True, False)
|
check_finality(test_state, prev_state, True, True, False)
|
||||||
elif epoch == 2:
|
elif epoch == 2:
|
||||||
# finalized by rule 1
|
# finalized by rule 1
|
||||||
check_finality(test_state, state, True, True, True)
|
check_finality(test_state, prev_state, True, True, True)
|
||||||
assert test_state.finalized_epoch == old_previous_justified_epoch
|
assert test_state.finalized_epoch == old_previous_justified_epoch
|
||||||
assert test_state.finalized_root == old_previous_justified_root
|
assert test_state.finalized_root == old_previous_justified_root
|
||||||
|
|
||||||
|
return state, blocks, test_state
|
||||||
|
|
||||||
|
|
||||||
def test_finality_rule_2(state):
|
def test_finality_rule_2(state):
|
||||||
# get past first two epochs that finality does not run on
|
# get past first two epochs that finality does not run on
|
||||||
@ -127,30 +169,86 @@ def test_finality_rule_2(state):
|
|||||||
for epoch in range(3):
|
for epoch in range(3):
|
||||||
old_previous_justified_epoch = test_state.previous_justified_epoch
|
old_previous_justified_epoch = test_state.previous_justified_epoch
|
||||||
old_previous_justified_root = test_state.previous_justified_root
|
old_previous_justified_root = test_state.previous_justified_root
|
||||||
for slot in range(spec.SLOTS_PER_EPOCH):
|
|
||||||
attestation = None
|
|
||||||
if epoch == 0:
|
|
||||||
slot_to_attest = test_state.slot - spec.MIN_ATTESTATION_INCLUSION_DELAY + 1
|
|
||||||
if slot_to_attest >= get_epoch_start_slot(get_current_epoch(state)):
|
|
||||||
attestation = get_valid_attestation(test_state, slot_to_attest)
|
|
||||||
fill_aggregate_attestation(test_state, attestation)
|
|
||||||
if epoch == 2:
|
|
||||||
slot_to_attest = test_state.slot - spec.SLOTS_PER_EPOCH + 1
|
|
||||||
attestation = get_valid_attestation(test_state, slot_to_attest)
|
|
||||||
fill_aggregate_attestation(test_state, attestation)
|
|
||||||
|
|
||||||
block = build_empty_block_for_next_slot(test_state)
|
|
||||||
if attestation:
|
|
||||||
block.body.attestations.append(attestation)
|
|
||||||
state_transition(test_state, block)
|
|
||||||
blocks.append(block)
|
|
||||||
|
|
||||||
if epoch == 0:
|
if epoch == 0:
|
||||||
check_finality(test_state, state, True, False, False)
|
prev_state, blocks, test_state = next_epoch_with_attestations(test_state, True, False)
|
||||||
elif epoch == 1:
|
check_finality(test_state, prev_state, True, False, False)
|
||||||
check_finality(test_state, state, True, True, False)
|
if epoch == 1:
|
||||||
elif epoch == 2:
|
prev_state, blocks, test_state = next_epoch_with_attestations(test_state, False, False)
|
||||||
|
check_finality(test_state, prev_state, False, True, False)
|
||||||
|
if epoch == 2:
|
||||||
|
prev_state, blocks, test_state = next_epoch_with_attestations(test_state, False, True)
|
||||||
# finalized by rule 2
|
# finalized by rule 2
|
||||||
check_finality(test_state, state, True, True, True)
|
check_finality(test_state, prev_state, True, False, True)
|
||||||
assert test_state.finalized_epoch == old_previous_justified_epoch
|
assert test_state.finalized_epoch == old_previous_justified_epoch
|
||||||
assert test_state.finalized_root == old_previous_justified_root
|
assert test_state.finalized_root == old_previous_justified_root
|
||||||
|
return state, blocks, test_state
|
||||||
|
|
||||||
|
|
||||||
|
def test_finality_rule_3(state):
|
||||||
|
# get past first two epochs that finality does not run on
|
||||||
|
next_epoch(state)
|
||||||
|
next_epoch(state)
|
||||||
|
|
||||||
|
test_state = deepcopy(state)
|
||||||
|
|
||||||
|
blocks = []
|
||||||
|
for epoch in range(2):
|
||||||
|
prev_state = deepcopy(test_state)
|
||||||
|
for slot in range(spec.SLOTS_PER_EPOCH):
|
||||||
|
slot_to_attest = test_state.slot - spec.MIN_ATTESTATION_INCLUSION_DELAY + 1
|
||||||
|
attestation = get_valid_attestation(test_state, slot_to_attest)
|
||||||
|
fill_aggregate_attestation(test_state, attestation)
|
||||||
|
block = build_empty_block_for_next_slot(test_state)
|
||||||
|
block.body.attestations.append(attestation)
|
||||||
|
state_transition(test_state, block)
|
||||||
|
|
||||||
|
blocks.append(block)
|
||||||
|
if epoch == 0:
|
||||||
|
check_finality(test_state, prev_state, True, False, False)
|
||||||
|
if epoch == 1:
|
||||||
|
check_finality(test_state, prev_state, True, True, True)
|
||||||
|
|
||||||
|
prev_state = deepcopy(test_state)
|
||||||
|
next_epoch(test_state)
|
||||||
|
check_finality(test_state, prev_state, False, True, False)
|
||||||
|
|
||||||
|
|
||||||
|
prev_state = deepcopy(test_state)
|
||||||
|
for slot in range(spec.SLOTS_PER_EPOCH):
|
||||||
|
slot_to_attest = test_state.slot - spec.SLOTS_PER_EPOCH + 1
|
||||||
|
attestation = get_valid_attestation(test_state, slot_to_attest)
|
||||||
|
fill_aggregate_attestation(test_state, attestation)
|
||||||
|
block = build_empty_block_for_next_slot(test_state)
|
||||||
|
block.body.attestations.append(attestation)
|
||||||
|
state_transition(test_state, block)
|
||||||
|
|
||||||
|
assert len(test_state.previous_epoch_attestations) >= 0
|
||||||
|
assert len(test_state.current_epoch_attestations) == 0
|
||||||
|
|
||||||
|
blocks.append(block)
|
||||||
|
check_finality(test_state, prev_state, True, False, True)
|
||||||
|
|
||||||
|
|
||||||
|
prev_state = deepcopy(test_state)
|
||||||
|
for slot in range(spec.SLOTS_PER_EPOCH):
|
||||||
|
prev_slot_to_attest = test_state.slot - spec.SLOTS_PER_EPOCH + 1
|
||||||
|
prev_attestation = get_valid_attestation(test_state, prev_slot_to_attest)
|
||||||
|
fill_aggregate_attestation(test_state, prev_attestation)
|
||||||
|
|
||||||
|
cur_slot_to_attest = test_state.slot - spec.MIN_ATTESTATION_INCLUSION_DELAY + 1
|
||||||
|
cur_attestation = get_valid_attestation(test_state, cur_slot_to_attest)
|
||||||
|
fill_aggregate_attestation(test_state, cur_attestation)
|
||||||
|
|
||||||
|
block = build_empty_block_for_next_slot(test_state)
|
||||||
|
block.body.attestations.append(prev_attestation)
|
||||||
|
block.body.attestations.append(cur_attestation)
|
||||||
|
|
||||||
|
state_transition(test_state, block)
|
||||||
|
|
||||||
|
assert len(test_state.previous_epoch_attestations) >= 0
|
||||||
|
assert len(test_state.current_epoch_attestations) >= 0
|
||||||
|
|
||||||
|
blocks.append(block)
|
||||||
|
check_finality(test_state, prev_state, True, True, True)
|
||||||
|
|
||||||
|
return state, blocks, test_state
|
||||||
|
Loading…
x
Reference in New Issue
Block a user