Add passing exit tests

This commit is contained in:
Paul Hauner 2019-05-23 10:11:15 +10:00
parent 4db6a8a058
commit 13f78342c3
No known key found for this signature in database
GPG Key ID: 5E2CFF9B75FA63DF
4 changed files with 60 additions and 2 deletions

View File

@ -8,6 +8,7 @@ mod bls_g2_uncompressed;
mod bls_priv_to_pub;
mod bls_sign_msg;
mod operations_deposit;
mod operations_exit;
mod operations_transfer;
mod ssz_generic;
mod ssz_static;
@ -19,6 +20,7 @@ pub use bls_g2_uncompressed::*;
pub use bls_priv_to_pub::*;
pub use bls_sign_msg::*;
pub use operations_deposit::*;
pub use operations_exit::*;
pub use operations_transfer::*;
pub use ssz_generic::*;
pub use ssz_static::*;

View File

@ -0,0 +1,38 @@
use super::*;
use crate::case_result::compare_beacon_state_results_without_caches;
use serde_derive::Deserialize;
use state_processing::per_block_processing::process_exits;
use types::{BeaconState, EthSpec, VoluntaryExit};
#[derive(Debug, Clone, Deserialize)]
pub struct OperationsExit<E: EthSpec> {
pub description: String,
#[serde(bound = "E: EthSpec")]
pub pre: BeaconState<E>,
pub voluntary_exit: VoluntaryExit,
#[serde(bound = "E: EthSpec")]
pub post: Option<BeaconState<E>>,
}
impl<E: EthSpec> YamlDecode for OperationsExit<E> {
fn yaml_decode(yaml: &String) -> Result<Self, Error> {
Ok(serde_yaml::from_str(&yaml.as_str()).unwrap())
}
}
impl<E: EthSpec> Case for OperationsExit<E> {
fn result(&self, _case_index: usize) -> Result<(), Error> {
let mut state = self.pre.clone();
let exit = self.voluntary_exit.clone();
let mut expected = self.post.clone();
// Epoch processing requires the epoch cache.
state.build_all_caches(&E::spec()).unwrap();
let result = process_exits(&mut state, &[exit], &E::spec());
let mut result = result.and_then(|_| Ok(state));
compare_beacon_state_results_without_caches(&mut result, &mut expected)
}
}

View File

@ -63,6 +63,12 @@ impl Doc {
("operations", "transfer", "minimal") => {
run_test::<OperationsTransfer<MinimalEthSpec>>(self)
}
("operations", "voluntary_exit", "mainnet") => {
run_test::<OperationsExit<MainnetEthSpec>>(self)
}
("operations", "voluntary_exit", "minimal") => {
run_test::<OperationsExit<MinimalEthSpec>>(self)
}
(runner, handler, config) => panic!(
"No implementation for runner: \"{}\", handler: \"{}\", config: \"{}\"",
runner, handler, config

View File

@ -11,7 +11,10 @@ fn yaml_files_in_test_dir(dir: &Path) -> Vec<PathBuf> {
assert!(
base_path.exists(),
"Unable to locate test files. Did you init git submoules?"
format!(
"Unable to locate {:?}. Did you init git submoules?",
base_path
)
);
let mut paths: Vec<PathBuf> = WalkDir::new(base_path)
@ -70,7 +73,6 @@ fn operations_deposit() {
// No transfers are permitted in phase 0.
/*
#[test]
#[should_panic]
#[cfg(not(feature = "fake_crypto"))]
fn operations_transfer() {
yaml_files_in_test_dir(&Path::new("operations").join("transfer"))
@ -83,6 +85,16 @@ fn operations_transfer() {
}
*/
#[test]
#[cfg(not(feature = "fake_crypto"))]
fn operations_exit() {
yaml_files_in_test_dir(&Path::new("operations").join("voluntary_exit"))
.into_par_iter()
.for_each(|file| {
Doc::assert_tests_pass(file);
});
}
#[test]
#[cfg(not(feature = "fake_crypto"))]
fn bls() {