test: fix flaky test (#8673)

This commit is contained in:
Matthias Seitz
2024-06-07 15:01:20 +02:00
committed by GitHub
parent 007b0ebd6d
commit 264437a5c7

View File

@ -407,6 +407,7 @@ impl ChunkedFileReader {
T: FromReader, T: FromReader,
{ {
if self.file_byte_len == 0 && self.chunk.is_empty() { if self.file_byte_len == 0 && self.chunk.is_empty() {
dbg!(self.chunk.is_empty());
// eof // eof
return Ok(None) return Ok(None)
} }
@ -605,8 +606,9 @@ mod tests {
// Generate some random blocks // Generate some random blocks
let (file, headers, _) = generate_bodies_file(0..=14).await; let (file, headers, _) = generate_bodies_file(0..=14).await;
// calculate min for chunk byte length range // calculate min for chunk byte length range, pick a lower bound that guarantees at least
let chunk_byte_len = rand::thread_rng().gen_range(1..=10_000); // one block will be read
let chunk_byte_len = rand::thread_rng().gen_range(2000..=10_000);
trace!(target: "downloaders::file::test", chunk_byte_len); trace!(target: "downloaders::file::test", chunk_byte_len);
// init reader // init reader
@ -619,6 +621,7 @@ mod tests {
// test // test
while let Some(client) = reader.next_chunk::<FileClient>().await.unwrap() { while let Some(client) = reader.next_chunk::<FileClient>().await.unwrap() {
let sync_target = client.tip_header().unwrap(); let sync_target = client.tip_header().unwrap();
let sync_target_hash = sync_target.hash(); let sync_target_hash = sync_target.hash();
// construct headers downloader and use first header // construct headers downloader and use first header