Refactor with_parsed_values

crates/snforge-scarb-plugin/src/attributes/internal_config_statement.rs

@@ -1,6 +1,7 @@
 use super::AttributeInfo;
 use crate::{
     args::Arguments,
+    asserts::assert_is_used_once,
     common::{into_proc_macro_result, with_parsed_values},
     config_statement::append_config_statements,
 };
@@ -37,6 +38,7 @@ fn internal_config_statement_internal(
     args: Arguments,
     _warns: &mut Vec<Diagnostic>,
 ) -> Result<TokenStream, Diagnostics> {
+    assert_is_used_once::<InternalConfigStatementCollector>(db, func)?;
     args.assert_is_empty::<InternalConfigStatementCollector>()?;
 
     Ok(append_config_statements(db, func, TokenStream::empty()))

crates/snforge-scarb-plugin/src/attributes/test.rs

@@ -1,4 +1,5 @@
 use super::{AttributeInfo, ErrorExt, internal_config_statement::InternalConfigStatementCollector};
+use crate::asserts::assert_is_used_once;
 use crate::attributes::fuzzer::wrapper::FuzzerWrapperCollector;
 use crate::attributes::fuzzer::{FuzzerCollector, FuzzerConfigCollector};
 use crate::external_inputs::ExternalInput;
@@ -36,6 +37,7 @@ fn test_internal(
     args: Arguments,
     _warns: &mut Vec<Diagnostic>,
 ) -> Result<TokenStream, Diagnostics> {
+    assert_is_used_once::<TestCollector>(db, func)?;
     args.assert_is_empty::<TestCollector>()?;
     ensure_parameters_only_with_fuzzer_attribute(db, func)?;
 

crates/snforge-scarb-plugin/src/common.rs

@@ -1,6 +1,5 @@
 use crate::{
     args::Arguments,
-    asserts::assert_is_used_once,
     attributes::AttributeInfo,
     parse::{parse, parse_args},
 };
@@ -52,8 +51,6 @@ where
 
     let db = db.upcast();
 
-    assert_is_used_once::<Collector>(db, &func)?;
-
     let (args_db, args) = parse_args(args);
     let args_db = args_db.upcast();
 

crates/snforge-scarb-plugin/src/config_statement.rs

@@ -1,3 +1,4 @@
+use crate::asserts::assert_is_used_once;
 use crate::utils::{create_single_token, get_statements};
 use crate::{
     args::Arguments,
@@ -32,6 +33,8 @@ fn with_config_cheatcodes<Collector>(
 where
     Collector: AttributeCollector,
 {
+    assert_is_used_once::<Collector>(db, func)?;
+
     let value = Collector::args_into_config_expression(args_db, args, warns)?;
 
     let cheatcode_name = Collector::CHEATCODE_NAME;