1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

extern crate proc_macro;

use proc_macro::TokenStream;
use quote::quote;
use syn::ItemFn;
use syn::Ident;
use proc_macro2::Span;


#[proc_macro_attribute]
pub fn test_async(_attr: TokenStream, item: TokenStream) -> TokenStream {

    let input = syn::parse_macro_input!(item as ItemFn);
    let name = &input.sig.ident;
    let sync_name = format!("{}_sync",name);
    let out_fn_iden = Ident::new(&sync_name, Span::call_site());

    let expression = quote! {

        #[test]
        fn #out_fn_iden()  {

            ::fluvio_future::subscriber::init_logger();
          
            #input
            
            let ft = async {
                #name().await
            };

            if let Err(err) = ::fluvio_future::task::run_block_on(ft) {
                assert!(false,"error: {:?}",err);
            }            
            
        }
    };
    
    expression.into()
   
}