Fix #[itest] not preserving attributes

Author: Bromeon

godot-macros/src/bench.rs

@@ -8,7 +8,7 @@
 use proc_macro2::TokenStream;
 use quote::quote;
 
-use crate::util::{bail, KvParser};
+use crate::util::{bail, retain_attributes_except, KvParser};
 use crate::ParseResult;
 
 const DEFAULT_REPETITIONS: usize = 100;
@@ -42,7 +42,11 @@ pub fn attribute_bench(input_decl: venial::Item) -> ParseResult<TokenStream> {
 
     let body = &func.body;
 
+    // Filter out #[bench] itself, but preserve other attributes like #[allow], #[expect], etc.
+    let other_attributes = retain_attributes_except(&func.attributes, "bench");
+
     Ok(quote! {
+        #(#other_attributes)*
         pub fn #bench_name() {
             for _ in 0..#repetitions {
                 let __ret: #ret = #body;

godot-macros/src/itest.rs

@@ -8,7 +8,9 @@
 use proc_macro2::TokenStream;
 use quote::{quote, ToTokens};
 
-use crate::util::{bail, extract_typename, ident, path_ends_with, KvParser};
+use crate::util::{
+    bail, extract_typename, ident, path_ends_with, retain_attributes_except, KvParser,
+};
 use crate::ParseResult;
 
 pub fn attribute_itest(input_item: venial::Item) -> ParseResult<TokenStream> {
@@ -85,7 +87,11 @@ pub fn attribute_itest(input_item: venial::Item) -> ParseResult<TokenStream> {
         plugin_name = ident("__GODOT_ITEST");
     };
 
+    // Filter out #[itest] itself, but preserve other attributes like #[allow], #[expect], etc.
+    let other_attributes = retain_attributes_except(&func.attributes, "itest");
+
     Ok(quote! {
+        #(#other_attributes)*
         pub fn #test_name(#param) #return_tokens {
             #body
         }

godot-macros/src/util/mod.rs

@@ -85,6 +85,18 @@ pub(crate) use bail;
 pub(crate) use error;
 pub(crate) use require_api_version;
 
+/// Keeps all attributes except the one specified (e.g. `"itest"`).
+pub fn retain_attributes_except<'a>(
+    attributes: &'a [venial::Attribute],
+    macro_name: &'a str,
+) -> impl Iterator<Item = &'a venial::Attribute> {
+    attributes.iter().filter(move |attr| {
+        !attr
+            .get_single_path_segment()
+            .is_some_and(|segment| segment == macro_name)
+    })
+}
+
 pub fn reduce_to_signature(function: &venial::Function) -> venial::Function {
     let mut reduced = function.clone();
     reduced.vis_marker = None; // retained outside in the case of #[signal].

itest/rust/src/engine_tests/codegen_test.rs

@@ -190,3 +190,11 @@ trait TraitA {
 impl TraitA for CodegenTest3 {
     fn exit_tree(&mut self) {}
 }
+
+// Verifies that attributes (here #[expect]) are preserved by #[itest] macro.
+// See retain_attributes_except() function.
+#[itest]
+#[expect(unused_variables)]
+fn test_itest_macro_attribute_retention() {
+    let unused_var = 42; // Should not generate warning.
+}