jasterv/test-context
1
0
Fork 0
mirror of https://codeberg.org/JasterV/test-context.git synced 2026-04-27 02:15:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: clean up the macro implementation

Browse source
This commit is contained in:
Victor Martinez 2024-02-26 15:56:59 +01:00
commit 0fed446341
2 changed files with 52 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

1
test-context-macros/Cargo.toml
View file

@ -16,3 +16,4 @@ proc-macro = true
[dependencies]
quote = "1.0.3"
syn = { version = "^2", features = ["full"] }
proc-macro2 = "1.0.78"

46
test-context-macros/src/lib.rs
View file

@ -1,5 +1,6 @@
use proc_macro::TokenStream;
use quote::{format_ident, quote};
use syn::Ident;
/// Macro to use on tests to add the setup/teardown functionality of your context.
///
@ -25,17 +26,37 @@ use quote::{format_ident, quote};
pub fn test_context(attr: TokenStream, item: TokenStream) -> TokenStream {
let context_type = syn::parse_macro_input!(attr as syn::Ident);
let input = syn::parse_macro_input!(item as syn::ItemFn);
let ret = &input.sig.output;
let name = &input.sig.ident;
let arguments = &input.sig.inputs;
let inner_body = &input.block;
let body = &input.block;
let attrs = &input.attrs;
let is_async = input.sig.asyncness.is_some();
let wrapped_name = format_ident!("__test_context_wrapped_{}", name);
let outer_body = if is_async {
let wrapper_body = if is_async {
async_wrapper_body(context_type, &wrapped_name)
} else {
sync_wrapper_body(context_type, &wrapped_name)
};
let async_tag = if is_async {
quote! { async }
} else {
quote! {}
};
quote! {
#(#attrs)*
#async_tag fn #name() #ret #wrapper_body
#async_tag fn #wrapped_name(#arguments) #ret #body
}
.into()
}
fn async_wrapper_body(context_type: Ident, wrapped_name: &Ident) -> proc_macro2::TokenStream {
quote! {
{
use test_context::futures::FutureExt;
@ -55,7 +76,9 @@ pub fn test_context(attr: TokenStream, item: TokenStream) -> TokenStream {
}
}
}
} else {
}
fn sync_wrapper_body(context_type: Ident, wrapped_name: &Ident) -> proc_macro2::TokenStream {
quote! {
{
let mut ctx = <#context_type as test_context::TestContext>::setup();
@ -72,19 +95,4 @@ pub fn test_context(attr: TokenStream, item: TokenStream) -> TokenStream {
}
}
}
};
let async_tag = if is_async {
quote! { async }
} else {
quote! {}
};
let result = quote! {
#(#attrs)*
#async_tag fn #name() #ret #outer_body
#async_tag fn #wrapped_name(#arguments) #ret #inner_body
};
result.into()
}