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"

94
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,53 +26,19 @@ 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 {
quote! {
{
use test_context::futures::FutureExt;
let mut ctx = <#context_type as test_context::AsyncTestContext>::setup().await;
let wrapped_ctx = &mut ctx;
let result = async move {
std::panic::AssertUnwindSafe(
#wrapped_name(wrapped_ctx)
).catch_unwind().await
}.await;
<#context_type as test_context::AsyncTestContext>::teardown(ctx).await;
match result {
Ok(returned_value) => returned_value,
Err(err) => {
std::panic::resume_unwind(err);
}
}
}
}
let wrapper_body = if is_async {
async_wrapper_body(context_type, &wrapped_name)
} else {
quote! {
{
let mut ctx = <#context_type as test_context::TestContext>::setup();
let mut wrapper = std::panic::AssertUnwindSafe(&mut ctx);
let result = std::panic::catch_unwind(move || {
#wrapped_name(*wrapper)
});
<#context_type as test_context::TestContext>::teardown(ctx);
match result {
Ok(returned_value) => returned_value,
Err(err) => {
std::panic::resume_unwind(err);
}
}
}
}
sync_wrapper_body(context_type, &wrapped_name)
};
let async_tag = if is_async {
@ -80,11 +47,52 @@ pub fn test_context(attr: TokenStream, item: TokenStream) -> TokenStream {
quote! {}
};
let result = quote! {
quote! {
#(#attrs)*
#async_tag fn #name() #ret #outer_body
#async_tag fn #name() #ret #wrapper_body
#async_tag fn #wrapped_name(#arguments) #ret #inner_body
};
result.into()
#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;
let mut ctx = <#context_type as test_context::AsyncTestContext>::setup().await;
let wrapped_ctx = &mut ctx;
let result = async move {
std::panic::AssertUnwindSafe(
#wrapped_name(wrapped_ctx)
).catch_unwind().await
}.await;
<#context_type as test_context::AsyncTestContext>::teardown(ctx).await;
match result {
Ok(returned_value) => returned_value,
Err(err) => {
std::panic::resume_unwind(err);
}
}
}
}
}
fn sync_wrapper_body(context_type: Ident, wrapped_name: &Ident) -> proc_macro2::TokenStream {
quote! {
{
let mut ctx = <#context_type as test_context::TestContext>::setup();
let mut wrapper = std::panic::AssertUnwindSafe(&mut ctx);
let result = std::panic::catch_unwind(move || {
#wrapped_name(*wrapper)
});
<#context_type as test_context::TestContext>::teardown(ctx);
match result {
Ok(returned_value) => returned_value,
Err(err) => {
std::panic::resume_unwind(err);
}
}
}
}
}