refactor: reduce tokio::test

tmtmtoo · Jan 5, 2024 · 0fcf78c · 0fcf78c
1 parent 730bd88
commit 0fcf78c
Show file tree

Hide file tree

Showing 5 changed files with 49 additions and 10 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -25,5 +25,7 @@ structopt = "0.3.26"
 
 [dev-dependencies]
 assert_cmd = "1.0.1"
+futures-lite = "2.1.0"
+lite-async-test = "0.1.2"
 mockall = "0.12.1"
 predicates = "1.0.5"
diff --git a/src/app.rs b/src/app.rs
@@ -52,7 +52,7 @@ mod tests {
         }
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn run_must_be_done() {
         let actual = run(0).await;
         let expected = 5;

diff --git a/src/app/retry.rs b/src/app/retry.rs
@@ -155,7 +155,7 @@ mod tests {
         }
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_done_with_success() {
         let app = RetryApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE {
@@ -170,7 +170,7 @@ mod tests {
         ));
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_sleep_without_limit() {
         let app = RetryApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE {
@@ -190,7 +190,7 @@ mod tests {
         });
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_sleep_with_limit() {
         let app = RetryApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE {
@@ -218,7 +218,7 @@ mod tests {
         );
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_done_with_fail() {
         let app = RetryApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE {
@@ -233,7 +233,7 @@ mod tests {
         ));
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn sleep_to_exec() {
         let app = RetryApp::<TestE, TestS> {
             state: State::Sleep(TestS),

diff --git a/src/app/supervise.rs b/src/app/supervise.rs
@@ -146,7 +146,7 @@ mod tests {
         }
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_sleep_without_limit() {
         let app = SuperviseApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE),
@@ -164,7 +164,7 @@ mod tests {
         });
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_sleep_with_limit() {
         let app = SuperviseApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE),
@@ -182,7 +182,7 @@ mod tests {
         });
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn exec_cmd_to_done() {
         let app = SuperviseApp::<TestE, TestS> {
             state: State::ExecuteCommand(TestE),
@@ -195,7 +195,7 @@ mod tests {
         });
     }
 
-    #[tokio::test]
+    #[lite_async_test::async_test]
     async fn sleep_to_exec() {
         let app = SuperviseApp::<TestE, TestS> {
             state: State::Sleep(TestS),