feat(papyrus_base_layer): automock the base layer

starkware-libs · Feb 13, 2025 · ad5d208 · ad5d208
1 parent 7eab8a9
commit ad5d208
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/papyrus_base_layer/Cargo.toml b/crates/papyrus_base_layer/Cargo.toml
@@ -25,6 +25,7 @@ alloy-transport-http.workspace = true
 async-trait.workspace = true
 colored = { workspace = true, optional = true }
 ethers.workspace = true
+mockall.workspace = true
 papyrus_config.workspace = true
 serde.workspace = true
 starknet-types-core.workspace = true

diff --git a/crates/papyrus_base_layer/src/lib.rs b/crates/papyrus_base_layer/src/lib.rs
@@ -3,11 +3,13 @@ use std::fmt::{Debug, Display};
 use std::ops::RangeInclusive;
 
 use async_trait::async_trait;
+use mockall::automock;
 use serde::{Deserialize, Serialize};
 use starknet_api::block::BlockHashAndNumber;
 use starknet_api::core::{ContractAddress, EntryPointSelector, EthAddress, Nonce};
 use starknet_api::transaction::fields::{Calldata, Fee};
 use starknet_api::transaction::L1HandlerTransaction;
+use thiserror::Error;
 
 pub mod constants;
 pub mod ethereum_base_layer_contract;
@@ -22,8 +24,12 @@ mod base_layer_test;
 
 pub type L1BlockNumber = u64;
 
+#[derive(Debug, Error)]
+pub enum MockError {}
+
 /// Interface for getting data from the Starknet base contract.
 #[async_trait]
+#[automock(type Error = MockError;)]
 pub trait BaseLayerContract {
     type Error: Error + Display + Debug;