jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: roundtrip fuzz harness for PooledTransactions (#5125)

Browse Source
This commit is contained in:
alpharush
2023-10-30 12:11:25 -05:00
committed by GitHub
parent 2a56203563
commit 64d50643c8
1 changed files with 26 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
crates/net/eth-wire/tests/pooled_transactions.rs
View File

@ -3,21 +3,26 @@ use alloy_rlp::{Decodable, Encodable};
use reth_eth_wire::{EthVersion, PooledTransactions, ProtocolMessage};
use reth_primitives::{hex, Bytes, PooledTransactionsElement};
use std::{fs, path::PathBuf};
use test_fuzz::test_fuzz;
#[test]
fn decode_pooled_transactions_data() {
let network_data_path =
PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("testdata/pooled_transactions_with_blob");
let data = fs::read_to_string(network_data_path).expect("Unable to read file");
let hex_data = hex::decode(data.trim()).unwrap();
let txs = PooledTransactions::decode(&mut &hex_data[..]).unwrap();
/// Helper function to ensure encode-decode roundtrip works for [`PooledTransactions`].
#[test_fuzz]
fn roundtrip_pooled_transactions(hex_data: Vec<u8>) -> Result<(), alloy_rlp::Error> {
let input_rlp = &mut &hex_data[..];
let txs = match PooledTransactions::decode(input_rlp) {
Ok(txs) => txs,
Err(e) => return Err(e),
};
// get the amount of bytes decoded in `decode` by subtracting the length of the original buf,
// from the length of the remaining bytes
let decoded_len = hex_data.len() - input_rlp.len();
let expected_encoding = hex_data[..decoded_len].to_vec();
// do a roundtrip test
let mut buf = Vec::new();
txs.encode(&mut buf);
if hex_data != buf {
panic!("mixed pooled transaction roundtrip failed");
}
assert_eq!(expected_encoding, buf);
// now do another decoding, on what we encoded - this should succeed
let txs2 = PooledTransactions::decode(&mut &buf[..]).unwrap();
@ -27,6 +32,17 @@ fn decode_pooled_transactions_data() {
// ensure that the length is equal to the length of the encoded data
assert_eq!(txs.length(), buf.len());
Ok(())
}
#[test]
fn decode_pooled_transactions_data() {
let network_data_path =
PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("testdata/pooled_transactions_with_blob");
let data = fs::read_to_string(network_data_path).expect("Unable to read file");
let hex_data = hex::decode(data.trim()).expect("Unable to decode hex");
assert!(roundtrip_pooled_transactions(hex_data).is_ok());
}
#[test]