Skip to main content

hala_io_test_derive/
lib.rs

1use proc_macro::TokenStream;
2use quote::{quote, quote_spanned};
3use syn::{parse_macro_input, spanned::Spanned, ItemFn};
4
5#[proc_macro_attribute]
6pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream {
7    let item_fn = parse_macro_input!(item as ItemFn);
8
9    if item_fn.sig.asyncness.is_none() {
10        return TokenStream::from(quote_spanned! { item_fn.span() =>
11            compile_error!("the async keyword is missing from the function declaration"),
12        });
13    }
14
15    let fn_name = &item_fn.sig.ident;
16
17    let test_name = item_fn.sig.ident.to_string();
18
19    quote! {
20        #[::core::prelude::v1::test]
21        fn #fn_name() {
22            #item_fn
23
24            hala_io_test::socket_tester(#test_name,#fn_name);
25        }
26
27
28    }
29    .into()
30}