github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-01-28 16:48:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

test: fix flaky arbitrary test (#2832)

Browse Source
This commit is contained in:
Matthias Seitz
2023-05-25 14:09:19 +02:00
committed by GitHub
parent 2305310ef8
commit 455eead32f
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -50,16 +50,21 @@ pub fn maybe_generate_tests(args: TokenStream, ast: &DeriveInput) -> TokenStream
use rand::RngCore;
// get random instance of type
let mut raw = [0u8;20 ];
let mut raw = [0u8;1024];
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 val = <super::#type_ident as arbitrary::Arbitrary>::arbitrary(&mut unstructured);
if val.is_err() {
// this can be flaky sometimes due to not enough data for iterator based types like Vec
return
}
let val = val.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();
let mut header = reth_rlp::Header::decode(decode_buf).expect("failed to decode header");
header.payload_length+=1;
let mut b = Vec::with_capacity(decode_buf.len());
header.encode(&mut b);