Replace mock-accountant macro with mock builder pallet and Rate by

Quantity
centrifuge · Dec 12, 2023 · 083d77f · 083d77f
1 parent 6d6cd79
commit 083d77f
Show file tree

Hide file tree

Showing 5 changed files with 185 additions and 39 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/libs/mocks/src/pools.rs b/libs/mocks/src/pools.rs
@@ -1,6 +1,10 @@
 #[frame_support::pallet]
 pub mod pallet {
-	use cfg_traits::{PoolInspect, PoolReserve, PriceValue, Seconds, TrancheTokenPrice};
+	use cfg_traits::{
+		investments::InvestmentAccountant, PoolInspect, PoolReserve, PriceValue, Seconds,
+		TrancheTokenPrice,
+	};
+	use cfg_types::investments::InvestmentInfo;
 	use codec::{Decode, Encode, MaxEncodedLen};
 	use frame_support::pallet_prelude::*;
 	use mock_builder::{execute_call, register_call};
@@ -64,6 +68,42 @@ pub mod pallet {
 		) {
 			register_call!(move |(a, b, c)| f(a, b, c));
 		}
+
+		pub fn mock_info(
+			f: impl Fn(
+					T::TrancheCurrency,
+				) -> Result<
+					InvestmentInfo<T::AccountId, T::CurrencyId, T::TrancheCurrency>,
+					DispatchError,
+				> + 'static,
+		) {
+			register_call!(f);
+		}
+
+		pub fn mock_balance(f: impl Fn(T::TrancheCurrency, &T::AccountId) -> T::Balance + 'static) {
+			register_call!(move |(a, b)| f(a, b));
+		}
+
+		pub fn mock_transfer(
+			f: impl Fn(T::TrancheCurrency, &T::AccountId, &T::AccountId, T::Balance) -> DispatchResult
+				+ 'static,
+		) {
+			register_call!(move |(a, b, c, d)| f(a, b, c, d));
+		}
+
+		#[allow(non_snake_case)]
+		pub fn mock_InvestmentAccountant_deposit(
+			f: impl Fn(&T::AccountId, T::TrancheCurrency, T::Balance) -> DispatchResult + 'static,
+		) {
+			register_call!(move |(a, b, c)| f(a, b, c));
+		}
+
+		#[allow(non_snake_case)]
+		pub fn mock_InvestmentAccountant_withdraw(
+			f: impl Fn(&T::AccountId, T::TrancheCurrency, T::Balance) -> DispatchResult + 'static,
+		) {
+			register_call!(move |(a, b, c)| f(a, b, c));
+		}
 	}
 
 	impl<T: Config> PoolInspect<T::AccountId, T::CurrencyId> for Pallet<T> {
@@ -88,6 +128,38 @@ pub mod pallet {
 		}
 	}
 
+	impl<T: Config> InvestmentAccountant<T::AccountId> for Pallet<T> {
+		type Amount = T::Balance;
+		type Error = DispatchError;
+		type InvestmentId = T::TrancheCurrency;
+		type InvestmentInfo = InvestmentInfo<T::AccountId, T::CurrencyId, Self::InvestmentId>;
+
+		fn info(a: Self::InvestmentId) -> Result<Self::InvestmentInfo, DispatchError> {
+			execute_call!(a)
+		}
+
+		fn balance(a: Self::InvestmentId, b: &T::AccountId) -> Self::Amount {
+			execute_call!((a, b))
+		}
+
+		fn transfer(
+			a: Self::InvestmentId,
+			b: &T::AccountId,
+			c: &T::AccountId,
+			d: Self::Amount,
+		) -> DispatchResult {
+			execute_call!((a, b, c, d))
+		}
+
+		fn deposit(a: &T::AccountId, b: Self::InvestmentId, c: Self::Amount) -> DispatchResult {
+			execute_call!((a, b, c))
+		}
+
+		fn withdraw(a: &T::AccountId, b: Self::InvestmentId, c: Self::Amount) -> DispatchResult {
+			execute_call!((a, b, c))
+		}
+	}
+
 	impl<T: Config> TrancheTokenPrice<T::AccountId, T::CurrencyId> for Pallet<T> {
 		type BalanceRatio = T::BalanceRatio;
 		type Moment = Seconds;

diff --git a/pallets/investments/Cargo.toml b/pallets/investments/Cargo.toml
@@ -29,6 +29,7 @@ frame-benchmarking = { git = "https://github.com/paritytech/substrate", default-
 [dev-dependencies]
 cfg-test-utils = { path = "../../libs/test-utils" }
 cfg-types = { path = "../../libs/types" }
+cfg-mocks = { workspace = true, default-features = true }
 orml-tokens = { git = "https://github.com/open-web3-stack/open-runtime-module-library", branch = "polkadot-v0.9.43" }
 orml-traits = { git = "https://github.com/open-web3-stack/open-runtime-module-library", branch = "polkadot-v0.9.43" }
 pallet-balances = { git = "https://github.com/paritytech/substrate", branch = "polkadot-v0.9.43" }
@@ -44,6 +45,7 @@ runtime-benchmarks = [
   "cfg-test-utils/runtime-benchmarks",
   "cfg-traits/runtime-benchmarks",
   "cfg-types/runtime-benchmarks",
+  "cfg-mocks/runtime-benchmarks",
   "frame-benchmarking/runtime-benchmarks",
   "frame-support/runtime-benchmarks",
   "frame-system/runtime-benchmarks",
@@ -67,6 +69,7 @@ try-runtime = [
   "cfg-primitives/try-runtime",
   "cfg-test-utils/try-runtime",
   "cfg-traits/try-runtime",
+  "cfg-mocks/try-runtime",
   "frame-support/try-runtime",
   "frame-system/try-runtime",
   "sp-runtime/try-runtime",