merkle proof test generator
Building merkle proofs is required functionality for implementing light client sync. Although the spec currently only defines a function to verify merkle proofs (`is_valid_merkle_branch`) there are still a few PySpec unit tests that produce merkle proofs. This patch adds a new generator to extract test vectors from those static unit tests, so that light client implementations can validate their merkle proof logic.
This commit is contained in:
parent
adfcb79815
commit
e7317e2283
|
@ -9,7 +9,11 @@ from eth2spec.test.helpers.merkle import build_proof
|
|||
@with_phases([ALTAIR])
|
||||
@spec_state_test
|
||||
def test_next_sync_committee_tree(spec, state):
|
||||
yield "state", state
|
||||
yield "leaf", state.next_sync_committee.hash_tree_root()
|
||||
yield "leaf_index", "meta", spec.NEXT_SYNC_COMMITTEE_INDEX
|
||||
next_sync_committee_branch = build_proof(state.get_backing(), spec.NEXT_SYNC_COMMITTEE_INDEX)
|
||||
yield "proof", next_sync_committee_branch
|
||||
assert spec.is_valid_merkle_branch(
|
||||
leaf=state.next_sync_committee.hash_tree_root(),
|
||||
branch=next_sync_committee_branch,
|
||||
|
@ -22,7 +26,11 @@ def test_next_sync_committee_tree(spec, state):
|
|||
@with_phases([ALTAIR])
|
||||
@spec_state_test
|
||||
def test_finality_root_tree(spec, state):
|
||||
yield "state", state
|
||||
yield "leaf", state.finalized_checkpoint.root
|
||||
yield "leaf_index", "meta", spec.FINALIZED_ROOT_INDEX
|
||||
finality_branch = build_proof(state.get_backing(), spec.FINALIZED_ROOT_INDEX)
|
||||
yield "proof", finality_branch
|
||||
assert spec.is_valid_merkle_branch(
|
||||
leaf=state.finalized_checkpoint.root,
|
||||
branch=finality_branch,
|
||||
|
|
|
@ -0,0 +1,36 @@
|
|||
# Merkle tests
|
||||
|
||||
This series of tests provides reference test vectors for validating correct
|
||||
generation and verification of merkle proofs based on static data.
|
||||
|
||||
## Test case format
|
||||
|
||||
### `meta.yaml`
|
||||
|
||||
```yaml
|
||||
leaf_index: int -- Generalized leaf index, verifying against the proof.
|
||||
proof_count: int -- Amount of proof elements.
|
||||
```
|
||||
|
||||
### `state.ssz_snappy`
|
||||
|
||||
An SSZ-snappy encoded `BeaconState` object from which other data is generated.
|
||||
|
||||
### `leaf.ssz_snappy`
|
||||
|
||||
An SSZ-snappy encoded `Bytes32` reflecting the merkle root of `leaf_index` at
|
||||
the given `state`.
|
||||
|
||||
### `proof_<index>.ssz_snappy`
|
||||
|
||||
A series of files, with `<index>` in range `[0, proof_count)`. Each file is an
|
||||
SSZ-snappy encoded `Bytes32` and represents one element of the merkle proof for
|
||||
`leaf_index` at the given `state`.
|
||||
|
||||
## Condition
|
||||
|
||||
A test-runner can implement the following assertions:
|
||||
- Check that `is_valid_merkle_branch` confirms `leaf` at `leaf_index` to verify
|
||||
against `has_tree_root(state)` and `proof`.
|
||||
- If the implementation supports generating merkle proofs, check that the
|
||||
self-generated proof matches the `proof` provided with the test.
|
|
@ -0,0 +1,6 @@
|
|||
# Merkle
|
||||
|
||||
The purpose of this test-generator is to provide test-vectors for validating the
|
||||
correct merkleization of objects and corresponding merkle proofs.
|
||||
|
||||
Test-format documentation can be found [here](../../formats/merkle/README.md).
|
|
@ -0,0 +1,14 @@
|
|||
from eth2spec.test.helpers.constants import ALTAIR
|
||||
from eth2spec.gen_helpers.gen_from_tests.gen import run_state_test_generators
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
altair_mods = {key: 'eth2spec.test.altair.unittests.test_' + key for key in [
|
||||
'helpers',
|
||||
]}
|
||||
|
||||
all_mods = {
|
||||
ALTAIR: altair_mods
|
||||
}
|
||||
|
||||
run_state_test_generators(runner_name="merkle", all_mods=all_mods)
|
|
@ -0,0 +1,2 @@
|
|||
pytest>=4.4
|
||||
../../../[generator]
|
Loading…
Reference in New Issue