Skip to content

Commit

Permalink
fix: rpc equivalence tests (#553)
Browse files Browse the repository at this point in the history 
* fix: permission failures on rpc equivalence tests

* fix inconsistent state in tests
  • Loading branch information
@ncitron
ncitron authored Mar 7, 2025
1 parent b2edcf2 commit 92cdb71
Showing 1 changed file with 179 additions and 102 deletions.
281 changes: 179 additions & 102 deletions tests/rpc_equivalence.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ async fn setup() -> (
};

// Start Verifiable API server that'd wrap the given RPC
let api_port = rng.gen_range(0..=65535);
let api_port = rng.gen_range(1024..=65535);
let mut api_server = VerifiableApiServer::new(ApiNetwork::Ethereum(ServerArgs {
server_address: format!("127.0.0.1:{api_port}").parse().unwrap(),
execution_rpc: Url::parse(&execution_rpc).unwrap(),
Expand All @@ -66,7 +66,7 @@ async fn setup() -> (

// Helios provider (Verifiable API)
let (helios_client_api, helios_provider_api) = {
let port = rng.gen_range(0..=65535);
let port = rng.gen_range(1024..=65535);
let mut helios_client = EthereumClientBuilder::new()
.network(Network::Mainnet)
.execution_verifiable_api(&format!("http://localhost:{api_port}"))
Expand Down Expand Up @@ -110,31 +110,46 @@ async fn get_transaction_by_hash() {
}

let (_handle1, _handle2, _handle3, providers) = setup().await;
let helios_api = providers[0].clone();
let helios_rpc = providers[1].clone();
let provider = providers[2].clone();

let block = providers[0]
let block_api = helios_api
.get_block_by_number(BlockNumberOrTag::Latest, BlockTransactionsKind::Hashes)
.await
.unwrap()
.unwrap();
let block_rpc = helios_rpc
.get_block_by_number(BlockNumberOrTag::Latest, BlockTransactionsKind::Hashes)
.await
.unwrap()
.unwrap();

let tx_hash = block.transactions.hashes().next().unwrap();

let results = join_all(
providers
.into_iter()
.map(|provider| async move {
provider
.get_transaction_by_hash(tx_hash)
.await
.unwrap()
.unwrap()
})
.collect::<Vec<_>>(),
)
.await;
let tx_hash = block_api.transactions.hashes().next().unwrap();
let tx = helios_api
.get_transaction_by_hash(tx_hash)
.await
.unwrap()
.unwrap();
let expected = provider
.get_transaction_by_hash(tx_hash)
.await
.unwrap()
.unwrap();
assert_eq!(tx, expected);

assert_eq!(results[0], results[1]);
assert_eq!(results[0], results[2]);
let tx_hash = block_rpc.transactions.hashes().next().unwrap();
let tx = helios_rpc
.get_transaction_by_hash(tx_hash)
.await
.unwrap()
.unwrap();
let expected = provider
.get_transaction_by_hash(tx_hash)
.await
.unwrap()
.unwrap();
assert_eq!(tx, expected);
}

// #[tokio::test]
Expand Down Expand Up @@ -163,31 +178,46 @@ async fn get_transaction_receipt() {
}

let (_handle1, _handle2, _handle3, providers) = setup().await;
let helios_api = providers[0].clone();
let helios_rpc = providers[1].clone();
let provider = providers[2].clone();

let block = providers[0]
let block_api = helios_api
.get_block_by_number(BlockNumberOrTag::Latest, BlockTransactionsKind::Hashes)
.await
.unwrap()
.unwrap();
let block_rpc = helios_rpc
.get_block_by_number(BlockNumberOrTag::Latest, BlockTransactionsKind::Hashes)
.await
.unwrap()
.unwrap();

let tx_hash = block.transactions.hashes().next().unwrap();

let results = join_all(
providers
.into_iter()
.map(|provider| async move {
provider
.get_transaction_receipt(tx_hash)
.await
.unwrap()
.unwrap()
})
.collect::<Vec<_>>(),
)
.await;
let tx_hash = block_api.transactions.hashes().next().unwrap();
let receipt = helios_api
.get_transaction_receipt(tx_hash)
.await
.unwrap()
.unwrap();
let expected = provider
.get_transaction_receipt(tx_hash)
.await
.unwrap()
.unwrap();
assert_eq!(receipt, expected);

assert_eq!(results[0], results[1]);
assert_eq!(results[0], results[2]);
let tx_hash = block_rpc.transactions.hashes().next().unwrap();
let receipt = helios_rpc
.get_transaction_receipt(tx_hash)
.await
.unwrap()
.unwrap();
let expected = provider
.get_transaction_receipt(tx_hash)
.await
.unwrap()
.unwrap();
assert_eq!(receipt, expected);
}

#[tokio::test]
Expand All @@ -197,25 +227,36 @@ async fn get_block_receipts() {
}

let (_handle1, _handle2, _handle3, providers) = setup().await;
let helios_api = providers[0].clone();
let helios_rpc = providers[1].clone();
let provider = providers[2].clone();

let block_num = providers[0].get_block_number().await.unwrap();

let results = join_all(
providers
.into_iter()
.map(|provider| async move {
provider
.get_block_receipts(block_num.into())
.await
.unwrap()
.unwrap()
})
.collect::<Vec<_>>(),
)
.await;
let helios_api_block_num = helios_api.get_block_number().await.unwrap();
let helios_rpc_block_num = helios_rpc.get_block_number().await.unwrap();

let receipts = helios_api
.get_block_receipts(helios_api_block_num.into())
.await
.unwrap()
.unwrap();
let expected = provider
.get_block_receipts(helios_api_block_num.into())
.await
.unwrap()
.unwrap();
assert_eq!(receipts, expected);

assert_eq!(results[0], results[1]);
assert_eq!(results[0], results[2]);
let receipts = helios_rpc
.get_block_receipts(helios_rpc_block_num.into())
.await
.unwrap()
.unwrap();
let expected = provider
.get_block_receipts(helios_rpc_block_num.into())
.await
.unwrap()
.unwrap();
assert_eq!(receipts, expected);
}

#[tokio::test]
Expand All @@ -225,27 +266,38 @@ async fn get_balance() {
}

let (_handle1, _handle2, _handle3, providers) = setup().await;
let helios_api = providers[0].clone();
let helios_rpc = providers[1].clone();
let provider = providers[2].clone();

let block_num = providers[0].get_block_number().await.unwrap();
let helios_api_block_num = helios_api.get_block_number().await.unwrap();
let helios_rpc_block_num = helios_rpc.get_block_number().await.unwrap();

let address = address!("00000000219ab540356cBB839Cbe05303d7705Fa");

let results = join_all(
providers
.into_iter()
.map(|provider| async move {
provider
.get_balance(address)
.block_id(block_num.into())
.await
.unwrap()
})
.collect::<Vec<_>>(),
)
.await;
let balance = helios_api
.get_balance(address)
.block_id(helios_api_block_num.into())
.await
.unwrap();
let expected = provider
.get_balance(address)
.block_id(helios_api_block_num.into())
.await
.unwrap();
assert_eq!(balance, expected);

assert_eq!(results[0], results[1]);
assert_eq!(results[0], results[2]);
let balance = helios_rpc
.get_balance(address)
.block_id(helios_rpc_block_num.into())
.await
.unwrap();
let expected = provider
.get_balance(address)
.block_id(helios_rpc_block_num.into())
.await
.unwrap();
assert_eq!(balance, expected);
}

#[tokio::test]
Expand All @@ -255,8 +307,12 @@ async fn call() {
}

let (_handle1, _handle2, _handle3, providers) = setup().await;
let helios_api = providers[0].clone();
let helios_rpc = providers[1].clone();
let provider = providers[2].clone();

let block_num = providers[0].get_block_number().await.unwrap();
let helios_api_block_num = helios_api.get_block_number().await.unwrap();
let helios_rpc_block_num = helios_rpc.get_block_number().await.unwrap();

let usdc = address!("a0b86991c6218b36c1d19d4a2e9eb0ce3606eb48");
let user = address!("99C9fc46f92E8a1c0deC1b1747d010903E884bE1");
Expand All @@ -268,25 +324,41 @@ async fn call() {
}
}

let results = join_all(
providers
.into_iter()
.map(|provider| async move {
let token = ERC20::new(usdc, provider);
token
.balanceOf(user)
.block(block_num.into())
.call()
.await
.unwrap()
._0
})
.collect::<Vec<_>>(),
)
.await;
let token_api = ERC20::new(usdc, helios_api);
let token_rpc = ERC20::new(usdc, helios_rpc);
let token_provider = ERC20::new(usdc, provider);

let balance = token_api
.balanceOf(user)
.block(helios_api_block_num.into())
.call()
.await
.unwrap()
._0;
let expected = token_provider
.balanceOf(user)
.block(helios_api_block_num.into())
.call()
.await
.unwrap()
._0;
assert_eq!(balance, expected);

assert_eq!(results[0], results[1]);
assert_eq!(results[0], results[2]);
let balance = token_rpc
.balanceOf(user)
.block(helios_rpc_block_num.into())
.call()
.await
.unwrap()
._0;
let expected = token_provider
.balanceOf(user)
.block(helios_rpc_block_num.into())
.call()
.await
.unwrap()
._0;
assert_eq!(balance, expected);
}

#[tokio::test]
Expand All @@ -296,23 +368,28 @@ async fn get_logs() {
}

let (_handle1, _handle2, _handle3, providers) = setup().await;
let helios_api = providers[0].clone();
let helios_rpc = providers[1].clone();
let provider = providers[2].clone();

let block = providers[0]
let block_api = helios_api
.get_block_by_number(BlockNumberOrTag::Latest, BlockTransactionsKind::Hashes)
.await
.unwrap()
.unwrap();
let block_rpc = helios_rpc
.get_block_by_number(BlockNumberOrTag::Latest, BlockTransactionsKind::Hashes)
.await
.unwrap()
.unwrap();

let filter = Filter::new().at_block_hash(block.header.hash);

let results = join_all(
providers
.iter()
.map(|provider| async { provider.get_logs(&filter).await.unwrap() })
.collect::<Vec<_>>(),
)
.await;
let filter = Filter::new().at_block_hash(block_api.header.hash);
let logs = helios_api.get_logs(&filter).await.unwrap();
let expected = provider.get_logs(&filter).await.unwrap();
assert_eq!(logs, expected);

assert_eq!(results[0], results[1]);
assert_eq!(results[0], results[2]);
let filter = Filter::new().at_block_hash(block_rpc.header.hash);
let logs = helios_rpc.get_logs(&filter).await.unwrap();
let expected = provider.get_logs(&filter).await.unwrap();
assert_eq!(logs, expected);
}

0 comments on commit 92cdb71

Please sign in to comment.