Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Testing setup #36

Merged
merged 12 commits into from
Dec 4, 2024
6 changes: 6 additions & 0 deletions .github/workflows/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,12 @@ jobs:
run: cargo check
- name: Test
run: cargo test -- --test-threads=1 --nocapture
- name: Downloading ethereum/tests
run: git clone https://github.com/ethereum/tests ethereum-tests
- name: Downloading EELS fixtures released at Cancun
run: curl -LO https://github.com/ethereum/execution-spec-tests/releases/download/v2.1.1/fixtures.tar.gz && tar -xzf fixtures.tar.gz
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Was going to suggest caching, but this is very fast already

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

the curl -LO is done within a sec, but git clone does take some time tbh

- name: Test specs (EELS and ethereum/tests)
run: cargo test --features testing

docs:
name: Docs
Expand Down
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1 +1,3 @@
/target
ethereum-tests/
fixtures/
3 changes: 3 additions & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ reth-ethereum-consensus = { git = "https://github.com/paradigmxyz/reth", rev = "
reth-chainspec = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
reth-chain-state = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
reth-consensus = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
reth-cli = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
reth-cli-util = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
# reth-auto-seal-consensus = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
reth-prune-types = { git = "https://github.com/paradigmxyz/reth", rev = "496bf0bf715f0a1fafc198f8d72ccd71913d1a40" }
Expand Down Expand Up @@ -80,4 +81,6 @@ libc = "0.2"
[features]
default = ["jemalloc"]
jemalloc = ["dep:tikv-jemallocator"]
testing = []
failing-tests = []

3 changes: 2 additions & 1 deletion scripts/chiado_genesis_alloc.json
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@
},
"registrar": "0x6000000000000000000000000000000000000000"
},
"eip1559collector": "0x1559000000000000000000000000000000000000"
"eip1559collector": "0x1559000000000000000000000000000000000000",
"depositContractAddress": "0xbabe2bed00000000000000000000000000000003"
},
"baseFeePerGas": "0x3b9aca00",
"difficulty": "0x01",
Expand Down
1 change: 1 addition & 0 deletions src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ mod evm_config;
pub mod execute;
mod gnosis;
mod payload_builder;
mod testing;

#[derive(Debug, Clone, Default, PartialEq, Eq, clap::Args)]
#[command(next_help_heading = "Gnosis")]
Expand Down
18 changes: 18 additions & 0 deletions src/testing/assert.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
//! Various assertion helpers.

use crate::testing::Error;
use std::fmt::Debug;

/// A helper like `assert_eq!` that instead returns `Err(Error::Assertion)` on failure.
pub fn assert_equal<T>(left: T, right: T, msg: &str) -> Result<(), Error>
where
T: PartialEq + Debug,
{
if left == right {
Ok(())
} else {
Err(Error::Assertion(format!(
"{msg}\n left `{left:?}`,\n right `{right:?}`"
)))
}
}
42 changes: 42 additions & 0 deletions src/testing/case.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
//! Test case definitions

use crate::testing::result::{CaseResult, Error};
use std::{
fmt::Debug,
path::{Path, PathBuf},
};

/// A single test case, capable of loading a JSON description of itself and running it.
///
/// See <https://ethereum-tests.readthedocs.io/> for test specs.
pub trait Case: Debug + Sync + Sized {
/// A description of the test.
fn description(&self) -> String {
"no description".to_string()
}

/// Load the test from the given file path.
///
/// The file can be assumed to be a valid EF test case as described on <https://ethereum-tests.readthedocs.io/>.
fn load(path: &Path) -> Result<Self, Error>;

/// Run the test.
fn run(&self) -> Result<(), Error>;
}

/// A container for multiple test cases.
#[derive(Debug)]
pub struct Cases<T> {
/// The contained test cases and the path to each test.
pub test_cases: Vec<(PathBuf, T)>,
}

impl<T: Case> Cases<T> {
/// Run the contained test cases.
pub fn run(&self) -> Vec<CaseResult> {
self.test_cases
.iter()
.map(|(path, case)| CaseResult::new(path, case, case.run()))
.collect()
}
}
286 changes: 286 additions & 0 deletions src/testing/cases/blockchain_test.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,286 @@
//! Test runners for `BlockchainTests` in <https://github.com/ethereum/tests>

use crate::{
execute::GnosisExecutorProvider,
testing::{
models::{BlockchainTest, ForkSpec},
Case, Error, Suite,
},
};
use alloy_rlp::Decodable;
use rayon::iter::{ParallelBridge, ParallelIterator};
use reth_chainspec::ChainSpec;
use reth_cli::chainspec::parse_genesis;
use reth_primitives::{BlockBody, SealedBlock, StaticFileSegment};
use reth_provider::{
providers::StaticFileWriter, test_utils::create_test_provider_factory_with_chain_spec,
DatabaseProviderFactory, HashingWriter, StaticFileProviderFactory,
};
use reth_stages::{stages::ExecutionStage, ExecInput, Stage};
use std::{
collections::BTreeMap,
fs,
path::{Path, PathBuf},
sync::Arc,
};

/// A handler for the blockchain test suite.
#[derive(Debug)]
pub struct BlockchainTests {
suite: String,
}

impl BlockchainTests {
/// Create a new handler for a subset of the blockchain test suite.
pub const fn new(suite: String) -> Self {
Self { suite }
}
}

impl Suite for BlockchainTests {
type Case = BlockchainTestCase;

fn suite_name(&self) -> String {
self.suite.clone()
}
}

/// An Ethereum blockchain test.
#[derive(Debug, PartialEq, Eq)]
pub struct BlockchainTestCase {
tests: BTreeMap<String, BlockchainTest>,
skip: bool,
}

impl Case for BlockchainTestCase {
fn load(path: &Path) -> Result<Self, Error> {
Ok(Self {
tests: {
let s = fs::read_to_string(path).map_err(|error| Error::Io {
path: path.into(),
error,
})?;
let test =
serde_json::from_str(&s).map_err(|error| Error::CouldNotDeserialize {
path: path.into(),
error,
})?;
test
},
skip: should_skip(path),
})
}

/// Runs the test cases for the Ethereum Forks test suite.
///
/// # Errors
/// Returns an error if the test is flagged for skipping or encounters issues during execution.
fn run(&self) -> Result<(), Error> {
// If the test is marked for skipping, return a Skipped error immediately.
if self.skip {
return Err(Error::Skipped);
}

let chainspec_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
.join("scripts")
.join("chiado_genesis_alloc.json");
let original_chain_spec: ChainSpec = parse_genesis(chainspec_path.to_str().unwrap())
.unwrap()
.into();
// let chain_spec: Arc<ChainSpec> = Arc::new(chain_spec);

// Iterate through test cases, filtering by the network type to exclude specific forks.
self.tests
.values()
.filter(|case| {
!matches!(
case.network,
ForkSpec::ByzantiumToConstantinopleAt5
| ForkSpec::Frontier
| ForkSpec::Homestead
| ForkSpec::Byzantium
| ForkSpec::Istanbul
| ForkSpec::Berlin
| ForkSpec::London
| ForkSpec::Constantinople
| ForkSpec::ConstantinopleFix
| ForkSpec::MergeEOF
| ForkSpec::MergeMeterInitCode
| ForkSpec::MergePush0
| ForkSpec::Unknown
)
})
.par_bridge()
.try_for_each(|case| {
// Create a new test database and initialize a provider for the test case.
let mut chain_spec: ChainSpec = case.network.clone().into();
chain_spec.genesis.config.extra_fields.insert(
String::from("eip1559collector"),
original_chain_spec
.genesis
.config
.extra_fields
.get("eip1559collector")
.unwrap()
.clone(),
);
chain_spec.genesis.config.extra_fields.insert(
String::from("blockRewardsContract"),
original_chain_spec
.genesis
.config
.extra_fields
.get("blockRewardsContract")
.unwrap()
.clone(),
);
chain_spec.deposit_contract = original_chain_spec.deposit_contract;
let chain_spec: Arc<ChainSpec> = Arc::new(chain_spec);
let provider = create_test_provider_factory_with_chain_spec(chain_spec.clone());

let provider = provider.database_provider_rw().unwrap();

// provider.

// Insert initial test state into the provider.
provider.insert_historical_block(
SealedBlock::new(
case.genesis_block_header.clone().into(),
BlockBody::default(),
)
.try_seal_with_senders()
.unwrap(),
)?;
case.pre.write_to_db(provider.tx_ref())?;

// Initialize receipts static file with genesis
{
let static_file_provider = provider.static_file_provider();
let mut receipts_writer = static_file_provider
.latest_writer(StaticFileSegment::Receipts)
.unwrap();
receipts_writer.increment_block(0).unwrap();
receipts_writer.commit_without_sync_all().unwrap();
}

// Decode and insert blocks, creating a chain of blocks for the test case.
let last_block = case.blocks.iter().try_fold(None, |_, block| {
let decoded = SealedBlock::decode(&mut block.rlp.as_ref())?;
dbg!(
"printing block during insertion: {:?}",
decoded.clone().try_seal_with_senders().unwrap()
);
provider.insert_historical_block(
decoded.clone().try_seal_with_senders().unwrap(),
)?;
Ok::<Option<SealedBlock>, Error>(Some(decoded))
})?;
provider
.static_file_provider()
.latest_writer(StaticFileSegment::Headers)
.unwrap()
.commit_without_sync_all()
.unwrap();

let gnosis_executor_provider = GnosisExecutorProvider::gnosis(chain_spec.clone());

// Execute the execution stage using the EVM processor factory for the test case
// network.
let result = ExecutionStage::new_with_executor(gnosis_executor_provider).execute(
&provider,
ExecInput {
target: last_block.as_ref().map(|b| b.number),
checkpoint: None,
},
);
if let Err(e) = result {
return Err(Error::Custom(
format!("error in execution stage {:?}", e).to_string(),
));
}

// Validate the post-state for the test case.
match (&case.post_state, &case.post_state_hash) {
(Some(state), None) => {
// Validate accounts in the state against the provider's database.
for (&address, account) in state {
account.assert_db(address, provider.tx_ref())?;
}
}
(None, Some(expected_state_root)) => {
// Insert state hashes into the provider based on the expected state root.
let last_block = last_block.unwrap_or_default();
provider.insert_hashes(
0..=last_block.number,
last_block.hash(),
*expected_state_root,
)?;
}
_ => {
return Err(Error::MissingPostState);
}
}

// Drop the provider without committing to the database.
drop(provider);
Ok(())
})?;

Ok(())
}
}

/// Returns whether the test at the given path should be skipped.
///
/// Some tests are edge cases that cannot happen on mainnet, while others are skipped for
/// convenience (e.g. they take a long time to run) or are temporarily disabled.
///
/// The reason should be documented in a comment above the file name(s).
pub fn should_skip(path: &Path) -> bool {
let path_str = path.to_str().expect("Path is not valid UTF-8");
let name = path.file_name().unwrap().to_str().unwrap();
false && matches!(
name,
// funky test with `bigint 0x00` value in json :) not possible to happen on mainnet and require
// custom json parser. https://github.com/ethereum/tests/issues/971
| "ValueOverflow.json"
| "ValueOverflowParis.json"

// txbyte is of type 02 and we don't parse tx bytes for this test to fail.
| "typeTwoBerlin.json"

// Test checks if nonce overflows. We are handling this correctly but we are not parsing
// exception in testsuite There are more nonce overflow tests that are in internal
// call/create, and those tests are passing and are enabled.
| "CreateTransactionHighNonce.json"

// Test check if gas price overflows, we handle this correctly but does not match tests specific
// exception.
| "HighGasPrice.json"
| "HighGasPriceParis.json"

// Skip test where basefee/accesslist/difficulty is present but it shouldn't be supported in
// London/Berlin/TheMerge. https://github.com/ethereum/tests/blob/5b7e1ab3ffaf026d99d20b17bb30f533a2c80c8b/GeneralStateTests/stExample/eip1559.json#L130
// It is expected to not execute these tests.
| "accessListExample.json"
| "basefeeExample.json"
| "eip1559.json"
| "mergeTest.json"

// These tests are passing, but they take a lot of time to execute so we are going to skip them.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

How long for curiosity?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

as long as it's running now (10 mins)
we don't exclude any tests now (not the false &&)

| "loopExp.json"
| "Call50000_sha256.json"
| "static_Call50000_sha256.json"
| "loopMul.json"
| "CALLBlake2f_MaxRounds.json"
| "shiftCombinations.json"
)
// Ignore outdated EOF tests that haven't been updated for Cancun yet.
|| path_contains(path_str, &["EIPTests", "stEOF"])
}

/// `str::contains` but for a path. Takes into account the OS path separator (`/` or `\`).
fn path_contains(path_str: &str, rhs: &[&str]) -> bool {
let rhs = rhs.join(std::path::MAIN_SEPARATOR_STR);
path_str.contains(&rhs)
}
Loading
Loading