use proc_macro::TokenStream;
use quote::{quote, quote_spanned};
use syn::{parse_macro_input, spanned::Spanned, ItemFn, TypePath};
#[proc_macro_attribute]
pub fn test(attr: TokenStream, item: TokenStream) -> TokenStream {
let runner_path = parse_macro_input!(attr as TypePath);
let item_fn = parse_macro_input!(item as ItemFn);
if item_fn.sig.asyncness.is_none() {
return TokenStream::from(quote_spanned! { item_fn.span() =>
compile_error!("the async keyword is missing from the function declaration"),
});
}
let fn_name = &item_fn.sig.ident;
let test_name = item_fn.sig.ident.to_string();
quote! {
#[::core::prelude::v1::test]
fn #fn_name() {
#item_fn
#runner_path(#test_name,#fn_name);
}
}
.into()
}