1234567891011121314151617#![cfg(feature = "runtime-benchmarks")]1819use super::*;20use crate::Pallet as PromototionPallet;2122use sp_runtime::traits::Bounded;2324use frame_benchmarking::{benchmarks, account};25use frame_support::traits::OnInitialize;26use frame_system::RawOrigin;27use pallet_unique::benchmarking::create_nft_collection;28use pallet_evm_migration::Pallet as EvmMigrationPallet;2930const SEED: u32 = 0;3132fn set_admin<T>() -> Result<T::AccountId, sp_runtime::DispatchError>33where34 T: Config + pallet_unique::Config + pallet_evm_migration::Config,35 T::BlockNumber: From<u32> + Into<u32>,36 <<T as Config>::Currency as Currency<T::AccountId>>::Balance: Sum + From<u128>,37{38 let pallet_admin = account::<T::AccountId>("admin", 0, SEED);3940 <T as Config>::Currency::make_free_balance_be(41 &pallet_admin,42 Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value(),43 );4445 PromototionPallet::<T>::set_admin_address(46 RawOrigin::Root.into(),47 T::CrossAccountId::from_sub(pallet_admin.clone()),48 )?;4950 Ok(pallet_admin)51}5253benchmarks! {54 where_clause{55 where T: Config + pallet_unique::Config + pallet_evm_migration::Config ,56 T::BlockNumber: From<u32> + Into<u32>,57 <<T as Config>::Currency as Currency<T::AccountId>>::Balance: Sum + From<u128>58 }5960 on_initialize {61 let b in 0..PENDING_LIMIT_PER_BLOCK;62 set_admin::<T>()?;6364 (0..b).try_for_each(|index| {65 let staker = account::<T::AccountId>("staker", index, SEED);66 <T as Config>::Currency::make_free_balance_be(&staker, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());67 PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())?;68 PromototionPallet::<T>::unstake(RawOrigin::Signed(staker.clone()).into()).map_err(|e| e.error)?;69 Result::<(), sp_runtime::DispatchError>::Ok(())70 })?;71 let block_number = <frame_system::Pallet<T>>::current_block_number() + T::PendingInterval::get();72 }: {PromototionPallet::<T>::on_initialize(block_number)}7374 set_admin_address {75 let pallet_admin = account::<T::AccountId>("admin", 0, SEED);76 let _ = <T as Config>::Currency::make_free_balance_be(&pallet_admin, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());77 } : _(RawOrigin::Root, T::CrossAccountId::from_sub(pallet_admin))7879 payout_stakers{80 let b in 1..101;8182 let pallet_admin = account::<T::AccountId>("admin", 1, SEED);83 let share = Perbill::from_rational(1u32, 20);84 PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;85 <T as Config>::Currency::make_free_balance_be(&pallet_admin, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());8687 let staker: T::AccountId = account("caller", 0, SEED);88 <T as Config>::Currency::make_free_balance_be(&staker, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());89 let stakers: Vec<T::AccountId> = (0..b).map(|index| account("staker", index, SEED)).collect();90 stakers.iter().for_each(|staker| {91 <T as Config>::Currency::make_free_balance_be(&staker, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());92 });93 (0..10).try_for_each(|_| {94 stakers.iter()95 .map(|staker| {96 PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())97 }).collect::<Result<Vec<_>, _>>()?;98 <frame_system::Pallet<T>>::finalize();99 Result::<(), sp_runtime::DispatchError>::Ok(())100 })?;101 } : _(RawOrigin::Signed(pallet_admin.clone()), Some(b as u8))102103 stake {104 let caller = account::<T::AccountId>("caller", 0, SEED);105 let share = Perbill::from_rational(1u32, 10);106 let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());107 } : _(RawOrigin::Signed(caller.clone()), share * <T as Config>::Currency::total_balance(&caller))108109 unstake {110 let caller = account::<T::AccountId>("caller", 0, SEED);111 let share = Perbill::from_rational(1u32, 20);112 let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());113 (0..10).map(|_| {114 115 <frame_system::Pallet<T>>::finalize();116 PromototionPallet::<T>::stake(RawOrigin::Signed(caller.clone()).into(), share * <T as Config>::Currency::total_balance(&caller))117 }).collect::<Result<Vec<_>, _>>()?;118119 } : _(RawOrigin::Signed(caller.clone()))120121 sponsor_collection {122 let pallet_admin = account::<T::AccountId>("admin", 0, SEED);123 PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;124 let _ = <T as Config>::Currency::make_free_balance_be(&pallet_admin, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());125 let caller: T::AccountId = account("caller", 0, SEED);126 let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());127 let collection = create_nft_collection::<T>(caller.clone())?;128 } : _(RawOrigin::Signed(pallet_admin.clone()), collection)129130 stop_sponsoring_collection {131 let pallet_admin = account::<T::AccountId>("admin", 0, SEED);132 PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;133 let _ = <T as Config>::Currency::make_free_balance_be(&pallet_admin, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());134 let caller: T::AccountId = account("caller", 0, SEED);135 let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());136 let collection = create_nft_collection::<T>(caller.clone())?;137 PromototionPallet::<T>::sponsor_collection(RawOrigin::Signed(pallet_admin.clone()).into(), collection)?;138 } : _(RawOrigin::Signed(pallet_admin.clone()), collection)139140 sponsor_contract {141 let pallet_admin = account::<T::AccountId>("admin", 0, SEED);142 PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;143144 let _ = <T as Config>::Currency::make_free_balance_be(&pallet_admin, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());145 let address = H160::from_low_u64_be(SEED as u64);146 let data: Vec<u8> = (0..20 as u8).collect();147 <EvmMigrationPallet<T>>::begin(RawOrigin::Root.into(), address)?;148 <EvmMigrationPallet<T>>::finish(RawOrigin::Root.into(), address, data)?;149 } : _(RawOrigin::Signed(pallet_admin.clone()), address)150151 stop_sponsoring_contract {152 let pallet_admin = account::<T::AccountId>("admin", 0, SEED);153 PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;154155 let _ = <T as Config>::Currency::make_free_balance_be(&pallet_admin, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());156 let address = H160::from_low_u64_be(SEED as u64);157 let data: Vec<u8> = (0..20 as u8).collect();158 <EvmMigrationPallet<T>>::begin(RawOrigin::Root.into(), address)?;159 <EvmMigrationPallet<T>>::finish(RawOrigin::Root.into(), address, data)?;160 PromototionPallet::<T>::sponsor_contract(RawOrigin::Signed(pallet_admin.clone()).into(), address)?;161 } : _(RawOrigin::Signed(pallet_admin.clone()), address)162}