fix: Skip simulation if skip_simulation = True

This is why we weren't getting BAL510 for `test_erc4626_linear_pool_creation` in python - simulation was being skipped, though not skipped in the rust porting. The simulation is skipped here since no liquidity has been added in more than 100k blocks.
This commit is contained in:
TAMARA LIPOWSKI
2025-09-07 01:09:29 -04:00
committed by Tamara
parent 02ed0fe216
commit dc288bcf29
3 changed files with 70 additions and 13 deletions

View File

@@ -69,13 +69,17 @@ impl RPCProvider {
}
}
// TODO: Implement
// async fn get_block_header(&self, _block_number: u64) {
// let provider = ProviderBuilder::new().on_http(self.url);
// let block_id: BlockId = BlockId::from(block_number);
//
// let block = provider.get_block(block_id)
// }
pub async fn get_block_header(&self, block_number: u64) -> miette::Result<Block> {
let provider = ProviderBuilder::new().connect_http(self.url.clone());
let block_id: BlockId = BlockId::from(block_number);
provider
.get_block(block_id)
.await
.into_diagnostic()
.wrap_err("Failed to fetch block header")
.and_then(|block_opt| block_opt.ok_or_else(|| miette::miette!("Block not found")))
}
}
#[cfg(test)]
@@ -122,4 +126,23 @@ mod tests {
assert_eq!(balance, U256::from(717250938432_u64));
}
#[tokio::test]
async fn get_block_header_test() {
let eth_rpc_url = env::var("RPC_URL").expect("Missing RPC_URL in environment");
let rpc_provider = RPCProvider::new(eth_rpc_url);
let block_number = 21998530;
let block_header = rpc_provider
.get_block_header(block_number)
.await
.unwrap();
// Verify that we got a block with the correct number
assert_eq!(block_number, block_header.header.number);
// Verify that the timestamp is non-zero
assert!(block_header.header.timestamp > 0);
}
}