Combine `process_crosslinks` and `verify_empty_shard_transition` into `process_shard_transitions`
This commit is contained in:
parent
b16e6d7a86
commit
cceeab2657
|
@ -64,6 +64,7 @@
|
||||||
- [`apply_shard_transition`](#apply_shard_transition)
|
- [`apply_shard_transition`](#apply_shard_transition)
|
||||||
- [`process_crosslink_for_shard`](#process_crosslink_for_shard)
|
- [`process_crosslink_for_shard`](#process_crosslink_for_shard)
|
||||||
- [`process_crosslinks`](#process_crosslinks)
|
- [`process_crosslinks`](#process_crosslinks)
|
||||||
|
- [`process_shard_transitions`](#process_shard_transitions)
|
||||||
- [`process_attestation`](#process_attestation)
|
- [`process_attestation`](#process_attestation)
|
||||||
- [New Attester slashing processing](#new-attester-slashing-processing)
|
- [New Attester slashing processing](#new-attester-slashing-processing)
|
||||||
- [Verify empty shard transition](#verify-empty-shard-transition)
|
- [Verify empty shard transition](#verify-empty-shard-transition)
|
||||||
|
@ -671,7 +672,6 @@ def process_block(state: BeaconState, block: BeaconBlock) -> None:
|
||||||
process_eth1_data(state, block.body)
|
process_eth1_data(state, block.body)
|
||||||
process_light_client_signatures(state, block.body)
|
process_light_client_signatures(state, block.body)
|
||||||
process_operations(state, block.body)
|
process_operations(state, block.body)
|
||||||
verify_empty_shard_transition(state, block.body)
|
|
||||||
```
|
```
|
||||||
|
|
||||||
#### Operations
|
#### Operations
|
||||||
|
@ -695,7 +695,7 @@ def process_operations(state: BeaconState, body: BeaconBlockBody) -> None:
|
||||||
# See custody game spec.
|
# See custody game spec.
|
||||||
process_custody_game_operations(state, body)
|
process_custody_game_operations(state, body)
|
||||||
|
|
||||||
process_crosslinks(state, body.shard_transitions, body.attestations)
|
process_shard_transitions(state, body.shard_transitions, body.attestations)
|
||||||
|
|
||||||
# TODO process_operations(body.shard_receipt_proofs, process_shard_receipt_proofs)
|
# TODO process_operations(body.shard_receipt_proofs, process_shard_receipt_proofs)
|
||||||
```
|
```
|
||||||
|
@ -882,6 +882,18 @@ def process_crosslinks(state: BeaconState,
|
||||||
pending_attestation.crosslink_success = True
|
pending_attestation.crosslink_success = True
|
||||||
```
|
```
|
||||||
|
|
||||||
|
###### `process_shard_transitions`
|
||||||
|
|
||||||
|
```python
|
||||||
|
def process_shard_transitions(state: BeaconState,
|
||||||
|
shard_transitions: Sequence[ShardTransition],
|
||||||
|
attestations: Sequence[Attestation]) -> None:
|
||||||
|
# Process crosslinks
|
||||||
|
process_crosslinks(state, shard_transitions, attestations)
|
||||||
|
# Verify the empty proposal shard states
|
||||||
|
assert verify_empty_shard_transition(state, shard_transitions)
|
||||||
|
```
|
||||||
|
|
||||||
###### `process_attestation`
|
###### `process_attestation`
|
||||||
|
|
||||||
```python
|
```python
|
||||||
|
@ -893,7 +905,7 @@ def process_attestation(state: BeaconState, attestation: Attestation) -> None:
|
||||||
data=attestation.data,
|
data=attestation.data,
|
||||||
inclusion_delay=state.slot - attestation.data.slot,
|
inclusion_delay=state.slot - attestation.data.slot,
|
||||||
proposer_index=get_beacon_proposer_index(state),
|
proposer_index=get_beacon_proposer_index(state),
|
||||||
crosslink_success=False, # To be filled in during process_crosslinks
|
crosslink_success=False, # To be filled in during process_shard_transitions
|
||||||
)
|
)
|
||||||
if attestation.data.target.epoch == get_current_epoch(state):
|
if attestation.data.target.epoch == get_current_epoch(state):
|
||||||
state.current_epoch_attestations.append(pending_attestation)
|
state.current_epoch_attestations.append(pending_attestation)
|
||||||
|
@ -946,13 +958,15 @@ def process_attester_slashing(state: BeaconState, attester_slashing: AttesterSla
|
||||||
#### Verify empty shard transition
|
#### Verify empty shard transition
|
||||||
|
|
||||||
```python
|
```python
|
||||||
def verify_empty_shard_transition(state: BeaconState, block_body: BeaconBlockBody) -> None:
|
def verify_empty_shard_transition(state: BeaconState, shard_transitions: Sequence[ShardTransition]) -> bool:
|
||||||
"""
|
"""
|
||||||
Verify that ``shard_transitions`` are empty if a crosslink was not formed for the associated shard in this slot.
|
Verify that ``shard_transitions`` are empty if a crosslink was not formed for the associated shard in this slot.
|
||||||
"""
|
"""
|
||||||
for shard in range(get_active_shard_count(state)):
|
for shard in range(get_active_shard_count(state)):
|
||||||
if state.shard_states[shard].slot != compute_previous_slot(state.slot):
|
if state.shard_states[shard].slot != compute_previous_slot(state.slot):
|
||||||
assert block_body.shard_transitions[shard] == ShardTransition()
|
if shard_transitions[shard] != ShardTransition():
|
||||||
|
return False
|
||||||
|
return True
|
||||||
```
|
```
|
||||||
|
|
||||||
#### Light client processing
|
#### Light client processing
|
||||||
|
|
|
@ -1,9 +1,9 @@
|
||||||
from eth2spec.test.context import expect_assertion_error
|
from eth2spec.test.context import expect_assertion_error
|
||||||
|
|
||||||
|
|
||||||
def run_crosslinks_processing(spec, state, shard_transitions, attestations, valid=True):
|
def run_shard_transitions_processing(spec, state, shard_transitions, attestations, valid=True):
|
||||||
"""
|
"""
|
||||||
Run ``process_attestation``, yielding:
|
Run ``process_shard_transitions``, yielding:
|
||||||
- pre-state ('pre')
|
- pre-state ('pre')
|
||||||
- shard_transitions ('shard_transitions')
|
- shard_transitions ('shard_transitions')
|
||||||
- attestations ('attestations')
|
- attestations ('attestations')
|
||||||
|
@ -17,12 +17,12 @@ def run_crosslinks_processing(spec, state, shard_transitions, attestations, vali
|
||||||
|
|
||||||
# If the attestation is invalid, processing is aborted, and there is no post-state.
|
# If the attestation is invalid, processing is aborted, and there is no post-state.
|
||||||
if not valid:
|
if not valid:
|
||||||
expect_assertion_error(lambda: spec.process_crosslinks(state, shard_transitions, attestations))
|
expect_assertion_error(lambda: spec.process_shard_transitions(state, shard_transitions, attestations))
|
||||||
yield 'post', None
|
yield 'post', None
|
||||||
return
|
return
|
||||||
|
|
||||||
# process crosslinks
|
# process crosslinks
|
||||||
spec.process_crosslinks(state, shard_transitions, attestations)
|
spec.process_shard_transitions(state, shard_transitions, attestations)
|
||||||
|
|
||||||
# yield post-state
|
# yield post-state
|
||||||
yield 'post', state
|
yield 'post', state
|
|
@ -4,7 +4,7 @@ from eth2spec.test.context import (
|
||||||
spec_state_test,
|
spec_state_test,
|
||||||
always_bls,
|
always_bls,
|
||||||
)
|
)
|
||||||
from eth2spec.test.helpers.crosslinks import run_crosslinks_processing
|
from eth2spec.test.helpers.shard_transitions import run_shard_transitions_processing
|
||||||
from eth2spec.test.helpers.shard_block import (
|
from eth2spec.test.helpers.shard_block import (
|
||||||
build_attestation_with_shard_transition,
|
build_attestation_with_shard_transition,
|
||||||
build_shard_block,
|
build_shard_block,
|
||||||
|
@ -46,7 +46,7 @@ def run_basic_crosslink_tests(spec, state, target_len_offset_slot, valid=True):
|
||||||
transition_to(spec, state, state.slot + target_len_offset_slot)
|
transition_to(spec, state, state.slot + target_len_offset_slot)
|
||||||
pre_shard_state = state.shard_states[shard]
|
pre_shard_state = state.shard_states[shard]
|
||||||
|
|
||||||
yield from run_crosslinks_processing(spec, state, shard_transitions, [attestation], valid=valid)
|
yield from run_shard_transitions_processing(spec, state, shard_transitions, [attestation], valid=valid)
|
||||||
|
|
||||||
if valid:
|
if valid:
|
||||||
# After state transition,
|
# After state transition,
|
|
@ -184,7 +184,7 @@ def create_provider(handler_name: str, tests_src, config_name: str) -> gen_typin
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
gen_runner.run_generator("epoch_processing", [
|
gen_runner.run_generator("epoch_processing", [
|
||||||
create_provider('crosslinks', test_process_crosslinks, 'minimal'),
|
create_provider('final_updates', test_process_final_updates, 'minimal'),
|
||||||
...
|
...
|
||||||
])
|
])
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue