fix benchmarks

chainflip-io · Oct 11, 2023 · f01b83b · f01b83b
1 parent 942fd22
commit f01b83b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/state-chain/pallets/cf-pools/src/benchmarking.rs b/state-chain/pallets/cf-pools/src/benchmarking.rs
@@ -17,6 +17,18 @@ fn new_lp_account<T: Chainflip>() -> T::AccountId {
 	let caller: T::AccountId = whitelisted_caller();
 	<T as frame_system::Config>::OnNewAccount::on_new_account(&caller);
 	T::AccountRoleRegistry::register_as_liquidity_provider(&caller).unwrap();
+
+	for encoded_address in [
+		EncodedAddress::Eth(Default::default()),
+		EncodedAddress::Dot(Default::default()),
+		EncodedAddress::Btc("bcrt1qs758ursh4q9z627kt3pp5yysm78ddny6txaqgw".as_bytes().to_vec()),
+	] {
+		LiquidityProvider::register_liquidity_refund_address(
+			RuntimeOrigin::signed(caller.clone()),
+			encoded_address
+		).unwrap();
+	}
+
 	caller
 }