macros: Error on function with arguments (#1419)

1 files changed, 13 insertions, 1 deletions

diff --git a/tokio-macros/src/lib.rs b/tokio-macros/src/lib.rs
index b7279b1e..ccb6ae7d 100644
--- a/tokio-macros/src/lib.rs
+++ b/tokio-macros/src/lib.rs
@@ -62,6 +62,12 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
         };
 
         return TokenStream::from(tokens);
+    } else if !input.decl.inputs.is_empty() {
+        let tokens = quote_spanned! { input.span() =>
+          compile_error!("the main function cannot accept arguments");
+        };
+
+        return TokenStream::from(tokens);
     }
 
     let mut runtime = RuntimeType::Multi;
@@ -102,7 +108,7 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
 ///
 /// # Examples
 ///
-/// ```ignore
+/// ```no_run
 /// #![feature(async_await)]
 ///
 /// #[tokio::test]
@@ -135,6 +141,12 @@ pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream {
         };
 
         return TokenStream::from(tokens);
+    } else if !input.decl.inputs.is_empty() {
+        let tokens = quote_spanned! { input.span() =>
+          compile_error!("the test function cannot accept arguments");
+        };
+
+        return TokenStream::from(tokens);
     }
 
     let result = quote! {