Replace remaining usages of finish_pending_tasks with condition

This commit is contained in:
Antonio Scandurra 2021-04-20 12:28:30 +02:00
parent ebb7124405
commit cd7dccd30c
4 changed files with 50 additions and 225 deletions

View file

@ -6,7 +6,6 @@ use simplelog::SimpleLogger;
use std::{
collections::BTreeMap,
path::{Path, PathBuf},
time::{Duration, Instant},
};
use tempdir::TempDir;
@ -144,18 +143,3 @@ fn write_tree(path: &Path, tree: serde_json::Value) {
panic!("You must pass a JSON object to this helper")
}
}
pub async fn assert_condition(poll_interval: u64, timeout: u64, mut f: impl FnMut() -> bool) {
let poll_interval = Duration::from_millis(poll_interval);
let timeout = Duration::from_millis(timeout);
let start = Instant::now();
loop {
if f() {
return;
} else if Instant::now().duration_since(start) < timeout {
smol::Timer::after(poll_interval).await;
} else {
panic!("timed out waiting on condition");
}
}
}