Create a macro which automates creation of benchmark test suites.

frame/assets/src/benchmarking.rs

@@ -20,7 +20,7 @@
 use super::*;
 use sp_runtime::traits::Bounded;
 use frame_system::RawOrigin as SystemOrigin;
-use frame_benchmarking::{benchmarks, account, whitelisted_caller};
+use frame_benchmarking::{benchmarks, account, whitelisted_caller, impl_benchmark_test_suite};
 use frame_support::traits::Get;
 
 use crate::Module as Assets;
@@ -233,120 +233,4 @@ benchmarks! {
 	}
 }
 
-#[cfg(test)]
-mod tests {
-	use super::*;
-	use crate::tests::{new_test_ext, Test};
-
-	#[test]
-	fn create() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_create::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn force_create() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_force_create::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn destroy() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_destroy::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn force_destroy() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_force_destroy::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn mint() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_mint::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn burn() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_burn::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn transfer() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_transfer::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn force_transfer() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_force_transfer::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn freeze() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_freeze::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn thaw() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_thaw::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn freeze_asset() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_freeze_asset::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn thaw_asset() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_thaw_asset::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn transfer_ownership() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_transfer_ownership::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn set_team() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_set_team::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn set_max_zombies() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_set_max_zombies::<Test>().is_ok());
-		});
-	}
-
-	#[test]
-	fn set_metadata() {
-		new_test_ext().execute_with(|| {
-			assert!(test_benchmark_set_metadata::<Test>().is_ok());
-		});
-	}
-}
+impl_benchmark_test_suite!(Assets, crate::tests::new_test_ext(), crate::tests::Test);

frame/balances/src/benchmarking.rs

@@ -22,7 +22,7 @@
 use super::*;
 
 use frame_system::RawOrigin;
-use frame_benchmarking::{benchmarks, account, whitelisted_caller};
+use frame_benchmarking::{benchmarks, account, whitelisted_caller, impl_benchmark_test_suite};
 use sp_runtime::traits::Bounded;
 
 use crate::Module as Balances;
@@ -144,51 +144,8 @@ benchmarks! {
 	}
 }
 
-#[cfg(test)]
-mod tests {
-	use super::*;
-	use crate::tests_composite::{ExtBuilder, Test};
-	use frame_support::assert_ok;
-
-	#[test]
-	fn transfer() {
-		ExtBuilder::default().build().execute_with(|| {
-			assert_ok!(test_benchmark_transfer::<Test>());
-		});
-	}
-
-	#[test]
-	fn transfer_best_case() {
-		ExtBuilder::default().build().execute_with(|| {
-			assert_ok!(test_benchmark_transfer_best_case::<Test>());
-		});
-	}
-
-	#[test]
-	fn transfer_keep_alive() {
-		ExtBuilder::default().build().execute_with(|| {
-			assert_ok!(test_benchmark_transfer_keep_alive::<Test>());
-		});
-	}
-
-	#[test]
-	fn transfer_set_balance_creating() {
-		ExtBuilder::default().build().execute_with(|| {
-			assert_ok!(test_benchmark_set_balance_creating::<Test>());
-		});
-	}
-
-	#[test]
-	fn transfer_set_balance_killing() {
-		ExtBuilder::default().build().execute_with(|| {
-			assert_ok!(test_benchmark_set_balance_killing::<Test>());
-		});
-	}
-
-	#[test]
-	fn force_transfer() {
-		ExtBuilder::default().build().execute_with(|| {
-			assert_ok!(test_benchmark_force_transfer::<Test>());
-		});
-	}
-}
+impl_benchmark_test_suite!(
+	Balances,
+	crate::tests_composite::ExtBuilder::default().build(),
+	crate::tests_composite::Test,
+);