fix test

AstarNetwork · Aug 29, 2024 · 13c0d19 · 13c0d19
1 parent a6ee8e6
commit 13c0d19
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pallets/vesting-mbm/src/mock.rs b/pallets/vesting-mbm/src/mock.rs
@@ -60,7 +60,10 @@ parameter_types! {
 
 #[derive_impl(pallet_migrations::config_preludes::TestDefaultConfig)]
 impl pallet_migrations::Config for Runtime {
+    #[cfg(not(feature = "runtime-benchmarks"))]
     type Migrations = (crate::LazyMigration<Runtime, crate::weights::SubstrateWeight<Runtime>>,);
+    #[cfg(feature = "runtime-benchmarks")]
+    type Migrations = pallet_migrations::mock_helpers::MockedMigrations;
     type MigrationStatusHandler = ();
     type MaxServiceWeight = MaxServiceWeight;
 }

diff --git a/pallets/vesting-mbm/src/tests.rs b/pallets/vesting-mbm/src/tests.rs
@@ -16,7 +16,7 @@
 // You should have received a copy of the GNU General Public License
 // along with Astar. If not, see <http://www.gnu.org/licenses/>.
 
-#![cfg(test)]
+#![cfg(all(test, not(feature = "runtime-benchmarks")))]
 
 use crate::mock::{
     new_test_ext, run_to_block, AllPalletsWithSystem, Balances, RuntimeOrigin, Vesting, ALICE, BOB,