initial setup the benchmarking for the rewards pallet

Author: mshankarrao

pallets/rewards/src/benchmarking.rs

@@ -4,17 +4,25 @@ use super::*;
 
 #[allow(unused)]
 use crate::Pallet as Rewards;
-use frame_benchmarking::{benchmarks, whitelisted_caller};
-use frame_system::RawOrigin;
+use frame_benchmarking::{account, benchmarks, whitelisted_caller};
+use frame_support::traits::{OnFinalize, OnInitialize,Currency};
+use frame_system::{DigestItemOf, EventRecord, RawOrigin};
+use sp_runtime::traits::Bounded;
 
 benchmarks! {
-	do_something {
-		let s in 0 .. 100;
-		let caller: T::AccountId = whitelisted_caller();
-	}: _(RawOrigin::Signed(caller), s)
+	on_initialize {
+		let author: T::AccountId = account("author", 0, 0);
+		let author_digest = DigestItemOf::<T>::PreRuntime(sp_consensus_pow::POW_ENGINE_ID, author.encode());
+		frame_system::Module::<T>::deposit_log(author_digest);
+
+		Reward::<T>::put(T::Currency::minimum_balance());
+
+		// Whitelist transient storage items
+		frame_benchmarking::benchmarking::add_to_whitelist(Author::<T>::hashed_key().to_vec().into());
+
+		let block_number = frame_system::Module::<T>::block_number();
+	}: { crate::Module::<T>::on_initialize(block_number); }
 	verify {
-		assert_eq!(Something::<T>::get(), Some(s));
+		assert_eq!(Author::<T>::get(), Some(author));
 	}
-
-	impl_benchmark_test_suite!(Rewards, crate::mock::new_test_ext(), crate::mock::Test);
 }

runtime/src/lib.rs

@@ -635,6 +635,7 @@ impl pallet_fast_unstake::Config for Runtime {
 	type Staking = Staking;
 	type MaxErasToCheckPerBlock = ConstU32<1>;
 	type WeightInfo = ();
+	type MaxBackersPerValidator = ();
 }
 
 parameter_types! {
@@ -1569,8 +1570,8 @@ impl pallet_nfts::Config for Runtime {
 	type MaxDeadlineDuration = MaxDeadlineDuration;
 	type Features = Features;
 	type WeightInfo = pallet_nfts::weights::SubstrateWeight<Runtime>;
-	#[cfg(feature = "runtime-benchmarks")]
-	type Helper = ();
+	// #[cfg(feature = "runtime-benchmarks")]
+	//type Helper = ();
 	type CreateOrigin = AsEnsureOriginWithArg<EnsureSigned<AccountId>>;
 	type Locker = ();
 }
@@ -1863,6 +1864,7 @@ mod benches {
 		[pallet_utility, Utility]
 		[pallet_vesting, Vesting]
 		[pallet_whitelist, Whitelist]
+		[rewards, Rewards]
 	);
 }
 
@@ -2252,6 +2254,23 @@ impl_runtime_apis! {
 			let mut batches = Vec::<BenchmarkBatch>::new();
 			let params = (&config, &whitelist);
 			add_benchmarks!(params, batches);
+			// add_benchmark!(params, batches, balances, Balances);
+			// add_benchmark!(params, batches, collective, Council);
+			// add_benchmark!(params, batches, democracy, Democracy);
+			// add_benchmark!(params, batches, identity, Identity);
+			// add_benchmark!(params, batches, indices, Indices);
+			// add_benchmark!(params, batches, multisig, Multisig);
+			// add_benchmark!(params, batches, proxy, Proxy);
+			// add_benchmark!(params, batches, scheduler, Scheduler);
+			// add_benchmark!(params, batches, system, SystemBench::<Runtime>);
+			// add_benchmark!(params, batches, timestamp, Timestamp);
+			// add_benchmark!(params, batches, treasury, Treasury);
+			// add_benchmark!(params, batches, utility, Utility);
+			// add_benchmark!(params, batches, vesting, Vesting);
+			// add_benchmark!(params, batches, rewards, Rewards);
+			// add_benchmark!(params, batches, staking, Staking);
+
+
 			Ok(batches)
 		}
 	}