`4dd148686c6b`

source

pallets/app-promotion/src/benchmarking.rs9.1 KiBsource history
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617#![cfg(feature = "runtime-benchmarks")]1819use frame_benchmarking::{account, benchmarks};20use frame_support::traits::{fungible::Unbalanced, OnInitialize};21use frame_system::RawOrigin;22use pallet_evm_migration::Pallet as EvmMigrationPallet;23use pallet_unique::benchmarking::create_nft_collection;24use sp_runtime::traits::Bounded;2526use super::*;27use crate::Pallet as PromototionPallet;2829const SEED: u32 = 0;3031fn set_admin<T>() -> Result<T::AccountId, sp_runtime::DispatchError>32where33	T: Config + pallet_unique::Config + pallet_evm_migration::Config,34	BlockNumberFor<T>: From<u32> + Into<u32>,35	BalanceOf<T>: Sum + From<u128>,36{37	let pallet_admin = account::<T::AccountId>("admin", 0, SEED);3839	<T as Config>::Currency::set_balance(40		&pallet_admin,41		Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value(),42	);4344	PromototionPallet::<T>::set_admin_address(45		RawOrigin::Root.into(),46		T::CrossAccountId::from_sub(pallet_admin.clone()),47	)?;4849	Ok(pallet_admin)50}5152benchmarks! {53	where_clause{54		where T:  Config + pallet_unique::Config + pallet_evm_migration::Config ,55		BlockNumberFor<T>: From<u32> + Into<u32>,56		BalanceOf<T>: Sum + From<u128>57	}5859	on_initialize {60		let b in 0..PENDING_LIMIT_PER_BLOCK;61		set_admin::<T>()?;6263		(0..b).try_for_each(|index| {64			let staker = account::<T::AccountId>("staker", index, SEED);65			<T as Config>::Currency::write_balance(&staker,  Into::<BalanceOf<T>>::into(10_000u128) * T::Nominal::get())?;66			PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())?;67			PromototionPallet::<T>::unstake_all(RawOrigin::Signed(staker).into())?;68			Result::<(), sp_runtime::DispatchError>::Ok(())69		})?;70		let block_number = <frame_system::Pallet<T>>::current_block_number() + T::PendingInterval::get();71	}: {PromototionPallet::<T>::on_initialize(block_number)}7273	set_admin_address {74		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);75		let _ = <T as Config>::Currency::set_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());76	} : _(RawOrigin::Root, T::CrossAccountId::from_sub(pallet_admin))7778	payout_stakers{79		let b in 1..100;8081		let pallet_admin = account::<T::AccountId>("admin", 1, SEED);82		let share = Perbill::from_rational(1u32, 20);83		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;84		<T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value())?;85		<T as Config>::Currency::write_balance(&<T as pallet::Config>::TreasuryAccountId::get(),  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value())?;8687		let stakers: Vec<T::AccountId> = (0..b).map(|index| account("staker", index, SEED)).collect();88		stakers.iter().try_for_each(|staker| {89			<T as Config>::Currency::write_balance(staker,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value())?;90			Result::<(), sp_runtime::DispatchError>::Ok(())91		})?;92		(1..11).try_for_each(|i| {93			<frame_system::Pallet<T>>::set_block_number(i.into());94			T::RelayBlockNumberProvider::set_block_number((2*i).into());95			assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());96			assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());97			stakers.iter()98			.map(|staker| {99				PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())100			}).collect::<Result<Vec<_>, _>>()?;101102			Result::<(), sp_runtime::DispatchError>::Ok(())103		})?;104105		let stakes = Staked::<T>::iter_prefix((&stakers[0],)).collect::<Vec<_>>();106		assert_eq!(stakes.len(), 10);107108		<frame_system::Pallet<T>>::set_block_number(15_000.into());109		T::RelayBlockNumberProvider::set_block_number(30_000.into());110	} : _(RawOrigin::Signed(pallet_admin.clone()), Some(b as u8))111112	stake {113		let caller = account::<T::AccountId>("caller", 0, SEED);114		let share = Perbill::from_rational(1u32, 10);115		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());116	} : _(RawOrigin::Signed(caller.clone()), share * <T as Config>::Currency::total_balance(&caller))117118	unstake_all {119		let caller = account::<T::AccountId>("caller", 0, SEED);120		let share = Perbill::from_rational(1u32, 20);121		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());122		(1..11).map(|i| {123			// used to change block number124			<frame_system::Pallet<T>>::set_block_number(i.into());125			T::RelayBlockNumberProvider::set_block_number((2*i).into());126			assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());127			assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());128			PromototionPallet::<T>::stake(RawOrigin::Signed(caller.clone()).into(), share * <T as Config>::Currency::total_balance(&caller))129		}).collect::<Result<Vec<_>, _>>()?;130131	} : _(RawOrigin::Signed(caller.clone()))132133	unstake_partial {134		let caller = account::<T::AccountId>("caller", 0, SEED);135		let share = Perbill::from_rational(1u32, 20);136		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());137		(1..11).map(|i| {138			// used to change block number139			<frame_system::Pallet<T>>::set_block_number(i.into());140			T::RelayBlockNumberProvider::set_block_number((2*i).into());141			assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());142			assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());143			PromototionPallet::<T>::stake(RawOrigin::Signed(caller.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())144		}).collect::<Result<Vec<_>, _>>()?;145146	} : _(RawOrigin::Signed(caller.clone()), Into::<BalanceOf<T>>::into(1000u128) * T::Nominal::get())147148	sponsor_collection {149		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);150		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;151		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());152		let caller: T::AccountId = account("caller", 0, SEED);153		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());154		let collection = create_nft_collection::<T>(caller)?;155	} : _(RawOrigin::Signed(pallet_admin.clone()), collection)156157	stop_sponsoring_collection {158		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);159		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;160		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());161		let caller: T::AccountId = account("caller", 0, SEED);162		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());163		let collection = create_nft_collection::<T>(caller)?;164		PromototionPallet::<T>::sponsor_collection(RawOrigin::Signed(pallet_admin.clone()).into(), collection)?;165	} : _(RawOrigin::Signed(pallet_admin.clone()), collection)166167	sponsor_contract {168		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);169		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;170171		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());172		let address = H160::from_low_u64_be(SEED as u64);173		let data: Vec<u8> = (0..20).collect();174		<EvmMigrationPallet<T>>::begin(RawOrigin::Root.into(), address)?;175		<EvmMigrationPallet<T>>::finish(RawOrigin::Root.into(), address, data)?;176	} : _(RawOrigin::Signed(pallet_admin.clone()), address)177178	stop_sponsoring_contract {179		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);180		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;181182		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());183		let address = H160::from_low_u64_be(SEED as u64);184		let data: Vec<u8> = (0..20).collect();185		<EvmMigrationPallet<T>>::begin(RawOrigin::Root.into(), address)?;186		<EvmMigrationPallet<T>>::finish(RawOrigin::Root.into(), address, data)?;187		PromototionPallet::<T>::sponsor_contract(RawOrigin::Signed(pallet_admin.clone()).into(), address)?;188	} : _(RawOrigin::Signed(pallet_admin.clone()), address)189}