tetcore_test_utils_derive/
lib.rsuse proc_macro::{Span, TokenStream};
use proc_macro_crate::crate_name;
use quote::quote;
use std::env;
#[proc_macro_attribute]
pub fn test(args: TokenStream, item: TokenStream) -> TokenStream {
impl_test(args, item)
}
fn impl_test(args: TokenStream, item: TokenStream) -> TokenStream {
let input = syn::parse_macro_input!(item as syn::ItemFn);
let args = syn::parse_macro_input!(args as syn::AttributeArgs);
parse_knobs(input, args).unwrap_or_else(|e| e.to_compile_error().into())
}
fn parse_knobs(
mut input: syn::ItemFn,
args: syn::AttributeArgs,
) -> Result<TokenStream, syn::Error> {
let sig = &mut input.sig;
let body = &input.block;
let attrs = &input.attrs;
let vis = input.vis;
if sig.inputs.len() != 1 {
let msg = "the test function accepts only one argument of type tc_service::TaskExecutor";
return Err(syn::Error::new_spanned(&sig, msg));
}
let (task_executor_name, task_executor_type) = match sig.inputs.pop().map(|x| x.into_value()) {
Some(syn::FnArg::Typed(x)) => (x.pat, x.ty),
_ => {
let msg =
"the test function accepts only one argument of type tc_service::TaskExecutor";
return Err(syn::Error::new_spanned(&sig, msg));
}
};
let crate_name = if env::var("CARGO_PKG_NAME").unwrap() == "tetcore-test-utils" {
syn::Ident::new("tetcore_test_utils", Span::call_site().into())
} else {
let crate_name = crate_name("tetcore-test-utils")
.map_err(|e| syn::Error::new_spanned(&sig, e))?;
syn::Ident::new(&crate_name, Span::call_site().into())
};
let header = {
quote! {
#[#crate_name::tokio::test(#(#args)*)]
}
};
let result = quote! {
#header
#(#attrs)*
#vis #sig {
use #crate_name::futures::future::FutureExt;
let #task_executor_name: #task_executor_type = (|fut, _| {
#crate_name::tokio::spawn(fut).map(drop)
})
.into();
let timeout_task = #crate_name::tokio::time::delay_for(
std::time::Duration::from_secs(
std::env::var("TETCORE_TEST_TIMEOUT")
.ok()
.and_then(|x| x.parse().ok())
.unwrap_or(600))
).fuse();
let actual_test_task = async move {
#body
}
.fuse();
#crate_name::futures::pin_mut!(timeout_task, actual_test_task);
#crate_name::futures::select! {
_ = timeout_task => {
panic!("The test took too long!");
},
_ = actual_test_task => {},
}
}
};
Ok(result.into())
}