address comment

Signed-off-by: Xinye <xinye.tao@metabit-trading.com>

Author: Xinye

src/lib.rs

@@ -867,7 +867,7 @@ macro_rules! fail_point {
 #[cfg(feature = "async")]
 mod async_imp {
     use super::*;
-    type BoxFuture<'a, T> = Pin<Box<dyn Future<Output = T> + Send + 'a, Global>>;
+    type BoxFuture<'a, T> = std::pin::Pin<Box<dyn std::future::Future<Output = T> + Send + 'a>>;
 
     #[derive(Clone)]
     pub(crate) struct AsyncCallback(
@@ -997,9 +997,16 @@ mod async_imp {
             match task {
                 Task::Off => {}
                 Task::Return(s) => return Some(s),
-                Task::Sleep(_) => panic!(
-                    "fail does not support async sleep, please use a async closure to sleep."
-                ),
+                Task::Sleep(t) => {
+                    let not = Arc::new(tokio::sync::Notify::new());
+                    let not_for_thread = not.clone();
+                    let handle = std::thread::spawn(move || {
+                        std::thread::sleep(Duration::from_millis(t));
+                        not_for_thread.notify_waiters();
+                    });
+                    not.notified().await;
+                    handle.join().unwrap();
+                }
                 Task::Panic(msg) => match msg {
                     Some(ref msg) => panic!("{}", msg),
                     None => panic!("failpoint {} panic", name),
@@ -1010,9 +1017,11 @@ mod async_imp {
                 },
                 Task::Pause => unreachable!(),
                 Task::Yield => thread::yield_now(),
-                Task::Delay(_) => panic!(
-                    "fail does not support async delay, please use a async closure to delay."
-                ),
+                Task::Delay(_) => {
+                    let timer = Instant::now();
+                    let timeout = Duration::from_millis(t);
+                    while timer.elapsed() < timeout {}
+                }
                 Task::Callback(f) => {
                     f.run();
                 }
@@ -1279,5 +1288,21 @@ mod tests {
             .await
             .unwrap();
         handle.await.unwrap();
+
+        cfg("sleep", "sleep(500)").unwrap();
+        let (tx, mut rx) = tokio::sync::mpsc::channel(1);
+        let handle = tokio::spawn(async move {
+            tx.send(()).await.unwrap();
+            async_fail_point!("sleep");
+            tx.send(()).await.unwrap();
+        });
+        rx.recv().await.unwrap();
+        tokio::time::timeout(Duration::from_millis(300), rx.recv())
+            .await
+            .unwrap_err();
+        tokio::time::timeout(Duration::from_millis(300), rx.recv())
+            .await
+            .unwrap();
+        handle.await.unwrap();
     }
 }