9d6209725f
## Issue Addressed This PR addresses partially #3651 ## Proposed Changes This PR adds the following methods: * a new method to trait `TreeHash`, `hash_tree_leaves` which returns all the Merkle leaves of the ssz object. * a new method to `BeaconState`: `compute_merkle_proof` which generates a specific merkle proof for given depth and index by using the `hash_tree_leaves` as leaves function. ## Additional Info Now here is some rationale on why I decided to go down this route: adding a new function to commonly used trait is a pain but was necessary to make sure we have all merkle leaves for every object, that is why I just added `hash_tree_leaves` in the trait and not `compute_merkle_proof` as well. although it would make sense it gives us code duplication/harder review time and we just need it from one specific object in one specific usecase so not worth the effort YET. In my humble opinion. Co-authored-by: Michael Sproul <micsproul@gmail.com>
108 lines
2.9 KiB
Rust
108 lines
2.9 KiB
Rust
use super::*;
|
|
use rayon::prelude::*;
|
|
use std::fmt::Debug;
|
|
use std::path::{Path, PathBuf};
|
|
use types::ForkName;
|
|
|
|
mod bls_aggregate_sigs;
|
|
mod bls_aggregate_verify;
|
|
mod bls_batch_verify;
|
|
mod bls_eth_aggregate_pubkeys;
|
|
mod bls_eth_fast_aggregate_verify;
|
|
mod bls_fast_aggregate_verify;
|
|
mod bls_sign_msg;
|
|
mod bls_verify_msg;
|
|
mod common;
|
|
mod epoch_processing;
|
|
mod fork;
|
|
mod fork_choice;
|
|
mod genesis_initialization;
|
|
mod genesis_validity;
|
|
mod merkle_proof_validity;
|
|
mod operations;
|
|
mod rewards;
|
|
mod sanity_blocks;
|
|
mod sanity_slots;
|
|
mod shuffling;
|
|
mod ssz_generic;
|
|
mod ssz_static;
|
|
mod transition;
|
|
|
|
pub use self::fork_choice::*;
|
|
pub use bls_aggregate_sigs::*;
|
|
pub use bls_aggregate_verify::*;
|
|
pub use bls_batch_verify::*;
|
|
pub use bls_eth_aggregate_pubkeys::*;
|
|
pub use bls_eth_fast_aggregate_verify::*;
|
|
pub use bls_fast_aggregate_verify::*;
|
|
pub use bls_sign_msg::*;
|
|
pub use bls_verify_msg::*;
|
|
pub use common::SszStaticType;
|
|
pub use epoch_processing::*;
|
|
pub use fork::ForkTest;
|
|
pub use genesis_initialization::*;
|
|
pub use genesis_validity::*;
|
|
pub use merkle_proof_validity::*;
|
|
pub use operations::*;
|
|
pub use rewards::RewardsTest;
|
|
pub use sanity_blocks::*;
|
|
pub use sanity_slots::*;
|
|
pub use shuffling::*;
|
|
pub use ssz_generic::*;
|
|
pub use ssz_static::*;
|
|
pub use transition::TransitionTest;
|
|
|
|
pub trait LoadCase: Sized {
|
|
/// Load the test case from a test case directory.
|
|
fn load_from_dir(_path: &Path, _fork_name: ForkName) -> Result<Self, Error>;
|
|
}
|
|
|
|
pub trait Case: Debug + Sync {
|
|
/// An optional field for implementing a custom description.
|
|
///
|
|
/// Defaults to "no description".
|
|
fn description(&self) -> String {
|
|
"no description".to_string()
|
|
}
|
|
|
|
/// Whether or not this test exists for the given `fork_name`.
|
|
///
|
|
/// Returns `true` by default.
|
|
fn is_enabled_for_fork(_fork_name: ForkName) -> bool {
|
|
true
|
|
}
|
|
|
|
/// Execute a test and return the result.
|
|
///
|
|
/// `case_index` reports the index of the case in the set of test cases. It is not strictly
|
|
/// necessary, but it's useful when troubleshooting specific failing tests.
|
|
fn result(&self, case_index: usize, fork_name: ForkName) -> Result<(), Error>;
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct Cases<T> {
|
|
pub test_cases: Vec<(PathBuf, T)>,
|
|
}
|
|
|
|
impl<T: Case> Cases<T> {
|
|
pub fn test_results(&self, fork_name: ForkName, use_rayon: bool) -> Vec<CaseResult> {
|
|
if use_rayon {
|
|
self.test_cases
|
|
.into_par_iter()
|
|
.enumerate()
|
|
.map(|(i, (ref path, ref tc))| {
|
|
CaseResult::new(i, path, tc, tc.result(i, fork_name))
|
|
})
|
|
.collect()
|
|
} else {
|
|
self.test_cases
|
|
.iter()
|
|
.enumerate()
|
|
.map(|(i, (ref path, ref tc))| {
|
|
CaseResult::new(i, path, tc, tc.result(i, fork_name))
|
|
})
|
|
.collect()
|
|
}
|
|
}
|
|
}
|