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

perf/test: run malformed header test separately (#2750)

Browse Source 
Co-authored-by: Matthias Seitz <matthias.seitz@outlook.de>
This commit is contained in:
chirag-bgh
2023-05-25 16:24:54 +05:30
committed by GitHub
parent 42683b8949
commit 2305310ef8
1 changed files with 21 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
crates/storage/codecs/derive/src/arbitrary.rs
View File

@ -16,6 +16,7 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
let mut traits = vec![];
let mut roundtrips = vec![];
let mut additional_tests = vec![];
for arg in args {
if arg.to_string() == "compact" {
@ -23,10 +24,8 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
roundtrips.push(quote! {
{
let mut buf = vec![];
let len = field.clone().to_compact(&mut buf);
let (decoded, _) = super::#type_ident::from_compact(&buf, len);
assert!(field == decoded);
}
});
@ -35,7 +34,6 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
roundtrips.push(quote! {
{
let mut buf = vec![];
let len = field.encode(&mut buf);
let mut b = &mut buf.as_slice();
let decoded = super::#type_ident::decode(b).unwrap();
@ -43,7 +41,23 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
// ensure buffer is fully consumed by decode
assert!(b.is_empty(), "buffer was not consumed entirely");
// malformed header check
}
});
additional_tests.push(quote! {
#[test]
fn malformed_rlp_header_check() {
use rand::RngCore;
// get random instance of type
let mut raw = [0u8;20 ];
rand::thread_rng().fill_bytes(&mut raw);
let mut unstructured = arbitrary::Unstructured::new(&raw[..]);
let val = <super::#type_ident as arbitrary::Arbitrary>::arbitrary(&mut unstructured).unwrap();
let mut buf = vec![];
let len = val.encode(&mut buf);
// malformed rlp-header check
let mut decode_buf = &mut buf.as_slice();
let mut header = reth_rlp::Header::decode(decode_buf).unwrap();
header.payload_length+=1;
@ -53,6 +67,7 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
let res = super::#type_ident::decode(&mut b.as_ref());
assert!(res.is_err(), "malformed header was decoded");
}
});
} else if let Ok(num) = arg.to_string().parse() {
default_cases = num;
@ -77,6 +92,8 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
#(#roundtrips)*
});
}
#(#additional_tests)*
}
}
}