Skip to content

Commit e67bb28

Browse files
authored
chore(net): test utils improvements (#8493)
1 parent fd495eb commit e67bb28

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

crates/net/downloaders/src/bodies/test_utils.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,15 @@ pub(crate) fn zip_blocks<'a>(
3030
.collect()
3131
}
3232

33-
pub(crate) fn create_raw_bodies<'a>(
34-
headers: impl Iterator<Item = &'a SealedHeader>,
33+
pub(crate) fn create_raw_bodies(
34+
headers: impl IntoIterator<Item = SealedHeader>,
3535
bodies: &mut HashMap<B256, BlockBody>,
3636
) -> Vec<Block> {
3737
headers
3838
.into_iter()
3939
.map(|header| {
4040
let body = bodies.remove(&header.hash()).expect("body exists");
41-
body.create_block(header.as_ref().clone())
41+
body.create_block(header.unseal())
4242
})
4343
.collect()
4444
}

crates/net/downloaders/src/test_utils/mod.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,15 +48,16 @@ pub(crate) async fn generate_bodies_file(
4848
range: RangeInclusive<u64>,
4949
) -> (tokio::fs::File, Vec<SealedHeader>, HashMap<B256, BlockBody>) {
5050
let (headers, bodies) = generate_bodies(range);
51-
let raw_block_bodies = create_raw_bodies(headers.clone().iter(), &mut bodies.clone());
51+
let raw_block_bodies = create_raw_bodies(headers.iter().cloned(), &mut bodies.clone());
5252

5353
let file: File = tempfile::tempfile().unwrap().into();
5454
let mut writer = FramedWrite::new(file, BlockFileCodec);
5555

5656
// rlp encode one after the other
5757
for block in raw_block_bodies {
58-
writer.send(block).await.unwrap();
58+
writer.feed(block).await.unwrap();
5959
}
60+
writer.flush().await.unwrap();
6061

6162
// get the file back
6263
let mut file: File = writer.into_inner();

0 commit comments

Comments
 (0)