lite_async_test/
lib.rs

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}