Improvement for #[tokio::main]

tokio-rs · Jun 14, 2024 · 13325e7 · 13325e7
1 parent c17e8d5
commit 13325e7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tokio-macros/src/entry.rs b/tokio-macros/src/entry.rs
@@ -480,7 +480,12 @@ fn parse_knobs(mut input: ItemFn, is_test: bool, config: FinalConfig) -> TokenSt
     };
 
     let input = input.into_tokens(generated_attrs, body, last_block);
-    replace_with_prefix(input)
+
+    if is_test {
+        input
+    } else {
+        replace_with_prefix(input)
+    }
 }
 
 fn replace_with_prefix(input: TokenStream) -> TokenStream {