hala-io-test-derive 0.1.2

Hala asynchronous io test proc-macro toolkit
Documentation
use proc_macro::TokenStream;
use quote::{quote, quote_spanned};
use syn::{parse_macro_input, spanned::Spanned, ItemFn};

#[proc_macro_attribute]
pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream {
    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

            hala_io_test::socket_tester(#test_name,#fn_name);
        }


    }
    .into()
}