Substitute mock-accountant macro by a mock builder pallet

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};
@@ -22,7 +26,7 @@ pub mod pallet {
 		type Balance;
 		type BalanceRatio;
 		type CurrencyId;
-		type TrancheCurrency;
+		type TrancheCurrency: Default;
 	}
 
 	#[pallet::pallet]
@@ -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;
@@ -131,7 +203,7 @@ pub mod pallet {
 		type PoolId = T::PoolId;
 
 		fn bench_default_investment_id(_: Self::PoolId) -> Self::InvestmentId {
-			unimplemented!();
+			T::TrancheCurrency::default()
 		}
 	}
 }