1use proc_macro::TokenStream;
2use quote::quote;
3use syn::{parse_macro_input, ItemFn};
4
5#[proc_macro_attribute]
6pub fn async_test(_attr: TokenStream, item: TokenStream) -> TokenStream {
7 let input = parse_macro_input!(item as ItemFn);
8 let name = &input.sig.ident;
9 let new_name = format!("{}_wrapper", name);
10 let new_ident = syn::Ident::new(&new_name, name.span());
11
12 let output = quote! {
13 #input
14
15 #[test]
16 fn #new_ident() {
17 futures_lite::future::block_on(async {
18 #name().await;
19 });
20 }
21 };
22
23 TokenStream::from(output)
24}