From 3a55aba251e445b1ff5e03ee2319d7a2920c1ee9 Mon Sep 17 00:00:00 2001 From: Taiki Endo Date: Mon, 23 Sep 2019 04:09:30 +0900 Subject: macros: add build tests for #[tokio::main] and #[tokio::test] (#1591) --- tokio-macros/src/lib.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tokio-macros') diff --git a/tokio-macros/src/lib.rs b/tokio-macros/src/lib.rs index cabfd8be..9a33c5fd 100644 --- a/tokio-macros/src/lib.rs +++ b/tokio-macros/src/lib.rs @@ -124,8 +124,9 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream { /// } /// ``` #[proc_macro_attribute] -pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream { +pub fn test(args: TokenStream, item: TokenStream) -> TokenStream { let input = syn::parse_macro_input!(item as syn::ItemFn); + let _ = syn::parse_macro_input!(args as syn::parse::Nothing); let ret = &input.sig.output; let name = &input.sig.ident; @@ -143,7 +144,7 @@ pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream { if input.sig.asyncness.is_none() { let msg = "the async keyword is missing from the function declaration"; - return syn::Error::new_spanned(&input, msg) + return syn::Error::new_spanned(&input.sig.fn_token, msg) .to_compile_error() .into(); } else if !input.sig.inputs.is_empty() { -- cgit v1.2.3