fix test build

moonbeam-foundation · Nov 29, 2024 · 328b3d0 · 328b3d0
1 parent 59bb6d4
commit 328b3d0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/pallets/moonbeam-lazy-migrations/src/mock.rs b/pallets/moonbeam-lazy-migrations/src/mock.rs
@@ -173,6 +173,19 @@ parameter_types! {
 	pub const AssetAccountDeposit: u128 = 1;
 }
 
+// Required for runtime benchmarks
+pallet_assets::runtime_benchmarks_enabled! {
+	pub struct BenchmarkHelper;
+	impl<AssetIdParameter> pallet_assets::BenchmarkHelper<AssetIdParameter> for BenchmarkHelper
+	where
+		AssetIdParameter: From<u128>,
+	{
+		fn create_asset_id_parameter(id: u32) -> AssetIdParameter {
+			(id as u128).into()
+		}
+	}
+}
+
 impl pallet_assets::Config<()> for Test {
 	type RuntimeEvent = RuntimeEvent;
 	type Balance = Balance;
@@ -192,6 +205,9 @@ impl pallet_assets::Config<()> for Test {
 	type AssetIdParameter = AssetId;
 	type CreateOrigin = AsEnsureOriginWithArg<EnsureSigned<AccountId>>;
 	type CallbackHandle = ();
+	pallet_assets::runtime_benchmarks_enabled! {
+		type BenchmarkHelper = BenchmarkHelper;
+	}
 }
 
 #[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Encode, Decode, RuntimeDebug, TypeInfo)]