From 72a68e2bf3152889f8afdf1c1ccb8a7db6706ed6 Mon Sep 17 00:00:00 2001 From: DoTheBestToGetTheBest <146037313+DoTheBestToGetTheBest@users.noreply.github.com> Date: Sat, 4 Nov 2023 00:44:15 -0700 Subject: [PATCH] feat(rpc-testing-utils) : enhanced tansaction replay (#5275) Co-authored-by: Matthias Seitz --- crates/rpc/rpc-testing-util/tests/it/trace.rs | 28 +++++++++++++++++-- 1 file changed, 26 insertions(+), 2 deletions(-) diff --git a/crates/rpc/rpc-testing-util/tests/it/trace.rs b/crates/rpc/rpc-testing-util/tests/it/trace.rs index 0e3d6559fd65..eb83913f088b 100644 --- a/crates/rpc/rpc-testing-util/tests/it/trace.rs +++ b/crates/rpc/rpc-testing-util/tests/it/trace.rs @@ -1,7 +1,8 @@ +use futures::StreamExt; use jsonrpsee::http_client::HttpClientBuilder; use reth_rpc_api_testing_util::{trace::TraceApiExt, utils::parse_env_url}; -use std::time::Instant; - +use reth_rpc_types::trace::parity::TraceType; +use std::{collections::HashSet, time::Instant}; /// This is intended to be run locally against a running node. /// /// This is a noop of env var `RETH_RPC_TEST_NODE_URL` is not set. @@ -21,3 +22,26 @@ async fn trace_many_blocks() { } println!("Traced all blocks in {:?}", now.elapsed()); } + +/// Tests the replaying of transactions on a local Ethereum node. + +#[tokio::test(flavor = "multi_thread")] +#[ignore] +async fn replay_transactions() { + let url = parse_env_url("RETH_RPC_TEST_NODE_URL").unwrap(); + let client = HttpClientBuilder::default().build(url).unwrap(); + + let tx_hashes = vec![ + "0x4e08fe36db723a338e852f89f613e606b0c9a17e649b18b01251f86236a2cef3".parse().unwrap(), + "0xea2817f1aeeb587b82f4ab87a6dbd3560fc35ed28de1be280cb40b2a24ab48bb".parse().unwrap(), + ]; + + let trace_types = HashSet::from([TraceType::StateDiff, TraceType::VmTrace]); + + let mut stream = client.replay_transactions(tx_hashes, trace_types); + let now = Instant::now(); + while let Some(replay_txs) = stream.next().await { + println!("Transaction: {:?}", replay_txs); + println!("Replayed transactions in {:?}", now.elapsed()); + } +}