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