fix: pallet_contract benchmarks

paritytech · Nov 22, 2023 · f6eeeb9 · f6eeeb9
1 parent 7f9baed
commit f6eeeb9
Show file tree

Hide file tree

Showing 3 changed files with 620 additions and 605 deletions.
diff --git a/runtime/stout/src/contracts.rs b/runtime/stout/src/contracts.rs
@@ -23,8 +23,6 @@ use frame_support::{
 	parameter_types,
 	traits::{ConstU32, Nothing},
 };
-#[cfg(feature = "runtime-benchmarks")]
-use pallet_contracts::NoopMigration;
 use pallet_contracts::{
 	weights::SubstrateWeight, Config, DebugInfo, DefaultAddressGenerator, Frame, Schedule,
 };
@@ -73,7 +71,7 @@ impl Config for Runtime {
 	#[cfg(not(feature = "runtime-benchmarks"))]
 	type Migrations = ();
 	#[cfg(feature = "runtime-benchmarks")]
-	type Migrations = (NoopMigration<1>, NoopMigration<2>);
+	type Migrations = pallet_contracts::migration::codegen::BenchMigrations;
 	type MaxDelegateDependencies = ConstU32<32>;
 	type CodeHashLockupDepositPercent = CodeHashLockupDepositPercent;
 	type Debug = ();

diff --git a/runtime/trappist/src/contracts.rs b/runtime/trappist/src/contracts.rs
@@ -23,8 +23,6 @@ use frame_support::{
 	parameter_types,
 	traits::{ConstBool, ConstU32, Nothing},
 };
-#[cfg(feature = "runtime-benchmarks")]
-use pallet_contracts::NoopMigration;
 use pallet_contracts::{
 	migration::{v13, v14, v15},
 	Config, DebugInfo, DefaultAddressGenerator, Frame, Schedule,
@@ -73,7 +71,7 @@ impl Config for Runtime {
 	#[cfg(not(feature = "runtime-benchmarks"))]
 	type Migrations = (v13::Migration<Self>, v14::Migration<Self, Balances>, v15::Migration<Self>);
 	#[cfg(feature = "runtime-benchmarks")]
-	type Migrations = (NoopMigration<1>, NoopMigration<2>);
+	type Migrations = pallet_contracts::migration::codegen::BenchMigrations;
 	type MaxDelegateDependencies = ConstU32<32>;
 	type CodeHashLockupDepositPercent = CodeHashLockupDepositPercent;
 	type Debug = ();