chore: add testing feature, use automock only as test

commit-id:c208af25

crates/starknet_gateway_types/Cargo.toml

@@ -5,14 +5,17 @@ edition.workspace = true
 license.workspace = true
 repository.workspace = true
 
+[features]
+testing = ["mockall"]
+
 [lints]
 workspace = true
 
 [dependencies]
 async-trait.workspace = true
 axum.workspace = true
 enum-assoc.workspace = true
-mockall.workspace = true
+mockall = { workspace = true, optional = true }
 papyrus_network_types.workspace = true
 papyrus_proc_macros.workspace = true
 papyrus_rpc.workspace = true
@@ -22,3 +25,7 @@ starknet_api.workspace = true
 starknet_sequencer_infra.workspace = true
 thiserror.workspace = true
 tracing.workspace = true
+
+[dev-dependencies]
+# Enable self with "testing" feature in tests.
+starknet_gateway_types = { workspace = true, features = ["testing"] }

crates/starknet_gateway_types/src/communication.rs

@@ -1,8 +1,8 @@
 use std::sync::Arc;
 
 use async_trait::async_trait;
-use mockall::predicate::*;
-use mockall::*;
+#[cfg(any(feature = "testing", test))]
+use mockall::automock;
 use papyrus_proc_macros::handle_response_variants;
 use serde::{Deserialize, Serialize};
 use starknet_api::transaction::TransactionHash;
@@ -30,7 +30,7 @@ use tracing::{error, instrument};
 
 /// Serves as the gateway's shared interface. Requires `Send + Sync` to allow transferring
 /// and sharing resources (inputs, futures) across threads.
-#[automock]
+#[cfg_attr(any(feature = "testing", test), automock)]
 #[async_trait]
 pub trait GatewayClient: Send + Sync {
     async fn add_tx(&self, gateway_input: GatewayInput) -> GatewayClientResult<TransactionHash>;