fix: benchmarks

AstarNetwork · Oct 19, 2023 · 09b30d8 · 09b30d8
1 parent 6744d84
commit 09b30d8
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/chain-extensions/unified-accounts/Cargo.toml b/chain-extensions/unified-accounts/Cargo.toml
@@ -41,5 +41,5 @@ std = [
 	# Astar
 	"astar-primitives/std",
 	"pallet-unified-accounts/std",
-    "unified-accounts-chain-extension-types/std"
+	"unified-accounts-chain-extension-types/std",
 ]
diff --git a/pallets/unified-accounts/src/benchmarking.rs b/pallets/unified-accounts/src/benchmarking.rs
@@ -20,6 +20,7 @@
 
 use super::*;
 use frame_benchmarking::v2::*;
+use frame_support::assert_ok;
 use frame_system::RawOrigin;
 
 /// Assert that the last event equals the provided one.
@@ -42,6 +43,10 @@ mod benchmarks {
         )
         .into();
 
+        assert_ok!(T::Currency::mint_into(
+            &caller,
+            T::AccountMappingStorageFee::get()
+        ));
         let caller_clone = caller.clone();
 
         #[extrinsic_call]
@@ -62,6 +67,11 @@ mod benchmarks {
         let caller_clone = caller.clone();
         let evm_address = T::DefaultNativeToEvm::into_h160(caller.clone());
 
+        assert_ok!(T::Currency::mint_into(
+            &caller,
+            T::AccountMappingStorageFee::get()
+        ));
+
         #[extrinsic_call]
         _(RawOrigin::Signed(caller));
 

diff --git a/pallets/unified-accounts/src/lib.rs b/pallets/unified-accounts/src/lib.rs
@@ -114,7 +114,7 @@ pub mod pallet {
         /// The overarching event type
         type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;
         /// The Currency for managing evm address assets
-        type Currency: FungibleMutate<Self::AccountId>;
+        type Currency: FungibleMutate<Self::AccountId, Balance = Balance>;
         /// Default evm address to account id conversion
         type DefaultEvmToNative: AddressMapping<Self::AccountId>;
         /// Default account id to evm address conversion