From 5a8e54f32739d02458524ec874b7e5e88d92e948 Mon Sep 17 00:00:00 2001
From: daxpedda <daxpedda@gmail.com>
Date: Thu, 1 Apr 2021 12:42:22 +0200
Subject: [PATCH] Rely on #10 and update Tokio

---
 rusty-fork-macro/Cargo.toml |  2 +-
 rusty-fork-macro/src/lib.rs | 75 ++++++++++++++-----------------------
 2 files changed, 29 insertions(+), 48 deletions(-)

diff --git a/rusty-fork-macro/Cargo.toml b/rusty-fork-macro/Cargo.toml
index 8b014fc..56f3d12 100644
--- a/rusty-fork-macro/Cargo.toml
+++ b/rusty-fork-macro/Cargo.toml
@@ -23,4 +23,4 @@ syn = { version = "1", features = ["full"] }
 
 [dev-dependencies]
 rusty-fork = { path = "../", features = ["macro", "timeout"] }
-tokio = { version = "0.2", features = ["rt-threaded", "macros"] }
+tokio = { version = "1", features = ["macros", "rt"] }
diff --git a/rusty-fork-macro/src/lib.rs b/rusty-fork-macro/src/lib.rs
index efc0f30..7438683 100644
--- a/rusty-fork-macro/src/lib.rs
+++ b/rusty-fork-macro/src/lib.rs
@@ -77,7 +77,7 @@ pub fn fork_test(args: TokenStream, item: TokenStream) -> TokenStream {
 
     // defaults
     let mut crate_name = quote::quote! { rusty_fork };
-    let mut timeout = quote::quote! { 0 };
+    let mut timeout = quote::quote! {};
 
     // may be changed by the user
     for arg in args {
@@ -86,7 +86,7 @@ pub fn fork_test(args: TokenStream, item: TokenStream) -> TokenStream {
                 match ident.to_string().as_str() {
                     "timeout_ms" => {
                         if let Lit::Int(int) = name_value.lit {
-                            timeout = int.to_token_stream();
+                            timeout = quote::quote! { #![rusty_fork(timeout_ms = #int)] }
                         }
                     }
                     "crate" => {
@@ -113,16 +113,16 @@ pub fn fork_test(args: TokenStream, item: TokenStream) -> TokenStream {
 
     let item = syn::parse_macro_input!(item as ItemFn);
 
-    let fn_attrs = &item.attrs;
-    let fn_sig = &item.sig;
-    let fn_body = &item.block;
-    let fn_name = &fn_sig.ident;
+    let fn_attrs = item.attrs;
+    let fn_vis = item.vis;
+    let fn_sig = item.sig;
+    let fn_body = item.block;
 
     // the default is that we add the `#[test]` for the use
     let mut test = quote::quote! { #[test] };
 
     // we should still support a use case where the user adds it himself
-    for attr in fn_attrs {
+    for attr in &fn_attrs {
         if let Some(ident) = attr.path.get_ident() {
             if ident == "test" {
                 test = quote::quote! {};
@@ -141,45 +141,13 @@ pub fn fork_test(args: TokenStream, item: TokenStream) -> TokenStream {
         .into();
     }
 
-    // support returning `Result`
-    let body_fn = if let ReturnType::Type(_, ret_ty) = &fn_sig.output {
-        quote::quote! {
-            fn body_fn() {
-                fn body_fn() -> #ret_ty #fn_body
-                body_fn().unwrap();
-            }
-        }
-    } else {
-        quote::quote! {
-            fn body_fn() #fn_body
-        }
-    };
-
     (quote::quote! {
-        #test
-        #(#fn_attrs)*
-        fn #fn_name() {
-            // Eagerly convert everything to function pointers so that all
-            // tests use the same instantiation of `fork`.
-            #body_fn
-            let body: fn () = body_fn;
-
-            fn supervise_fn(
-                child: &mut ::#crate_name::ChildWrapper,
-                _file: &mut ::std::fs::File
-            ) {
-                ::#crate_name::fork_test::supervise_child(child, #timeout)
-            }
-            let supervise:
-                fn (&mut ::#crate_name::ChildWrapper, &mut ::std::fs::File) =
-                supervise_fn;
-            ::#crate_name::fork(
-                ::#crate_name::rusty_fork_test_name!(#fn_name),
-                ::#crate_name::rusty_fork_id!(),
-                ::#crate_name::fork_test::no_configure_child,
-                supervise,
-                body
-            ).expect("forking test failed")
+        ::#crate_name::rusty_fork_test! {
+            #timeout
+
+            #test
+            #(#fn_attrs)*
+            #fn_vis #fn_sig #fn_body
         }
     })
     .into()
@@ -235,7 +203,20 @@ mod test {
 
     #[tokio::test]
     #[fork_test]
-    async fn my_test() {
-        assert!(true);
+    async fn async_test() {
+        tokio::task::spawn(async {
+            println!("hello from child");
+        })
+        .await
+        .unwrap();
+    }
+
+    #[tokio::test]
+    #[fork_test]
+    async fn async_return_test() -> std::result::Result<(), tokio::task::JoinError> {
+        tokio::task::spawn(async {
+            println!("hello from child");
+        })
+        .await
     }
 }