1use std::time::Duration;
 2
 3use futures::Future;
 4use smol::future::FutureExt;
 5pub use util::*;
 6
 7pub async fn timeout<F, T>(timeout: Duration, f: F) -> Result<T, ()>
 8where
 9    F: Future<Output = T>,
10{
11    let timer = async {
12        smol::Timer::after(timeout).await;
13        Err(())
14    };
15    let future = async move { Ok(f.await) };
16    timer.race(future).await
17}
18
19#[cfg(any(test, feature = "test-support"))]
20pub struct CwdBacktrace<'a>(pub &'a backtrace::Backtrace);
21
22#[cfg(any(test, feature = "test-support"))]
23impl<'a> std::fmt::Debug for CwdBacktrace<'a> {
24    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
25        use backtrace::{BacktraceFmt, BytesOrWideString};
26
27        let cwd = std::env::current_dir().unwrap();
28        let cwd = cwd.parent().unwrap();
29        let mut print_path = |fmt: &mut std::fmt::Formatter<'_>, path: BytesOrWideString<'_>| {
30            std::fmt::Display::fmt(&path, fmt)
31        };
32        let mut fmt = BacktraceFmt::new(f, backtrace::PrintFmt::Full, &mut print_path);
33        for frame in self.0.frames() {
34            let mut formatted_frame = fmt.frame();
35            if frame
36                .symbols()
37                .iter()
38                .any(|s| s.filename().map_or(false, |f| f.starts_with(&cwd)))
39            {
40                formatted_frame.backtrace_frame(frame)?;
41            }
42        }
43        fmt.finish()
44    }
45}