Skip to content

Commit 3940220

Browse files
committed
fix: preserve outer attributes in freenet_test macro
1 parent 2ebef93 commit 3940220

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

crates/freenet-macros/src/codegen.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ fn is_gateway(args: &FreenetTestArgs, node_label: &str, node_idx: usize) -> bool
2020
pub fn generate_test_code(args: FreenetTestArgs, input_fn: ItemFn) -> Result<TokenStream> {
2121
let test_fn_name = &input_fn.sig.ident;
2222
let inner_fn_name = format_ident!("{}_inner", test_fn_name);
23+
let attrs = &input_fn.attrs;
2324

2425
// Extract test body
2526
let test_body = &input_fn.block;
@@ -53,6 +54,7 @@ pub fn generate_test_code(args: FreenetTestArgs, input_fn: ItemFn) -> Result<Tok
5354

5455
let generated = quote! {
5556
#tokio_attr
57+
#(#attrs)*
5658
async fn #test_fn_name() -> freenet::test_utils::TestResult {
5759
use freenet::test_utils::{TestContext, TestLogger, NodeInfo};
5860
use std::time::{Duration, Instant};

0 commit comments

Comments
 (0)