git.delta.rocks / unique-network / refs/commits / 65414d643dd4

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 super::*;20use crate::Pallet as PromototionPallet;21use frame_support::traits::fungible::Unbalanced;22use 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	BalanceOf<T>: Sum + From<u128>,37{38	let pallet_admin = account::<T::AccountId>("admin", 0, SEED);3940	<T as Config>::Currency::set_balance(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		BalanceOf<T>: 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::write_balance(&staker,  Into::<BalanceOf<T>>::into(10_000u128) * T::Nominal::get())?;67			PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())?;68			PromototionPallet::<T>::unstake_all(RawOrigin::Signed(staker).into())?;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::set_balance(&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..100;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::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value())?;86		<T as Config>::Currency::write_balance(&<T as pallet::Config>::TreasuryAccountId::get(),  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value())?;8788		let stakers: Vec<T::AccountId> = (0..b).map(|index| account("staker", index, SEED)).collect();89		stakers.iter().try_for_each(|staker| {90			<T as Config>::Currency::write_balance(staker,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value())?;91			Result::<(), sp_runtime::DispatchError>::Ok(())92		})?;93		(1..11).try_for_each(|i| {94			<frame_system::Pallet<T>>::set_block_number(i.into());95			T::RelayBlockNumberProvider::set_block_number((2*i).into());96			assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());97			assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());98			stakers.iter()99			.map(|staker| {100				PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())101			}).collect::<Result<Vec<_>, _>>()?;102103			Result::<(), sp_runtime::DispatchError>::Ok(())104		})?;105106		let stakes = Staked::<T>::iter_prefix((&stakers[0],)).collect::<Vec<_>>();107		assert_eq!(stakes.len(), 10);108109		<frame_system::Pallet<T>>::set_block_number(15_000.into());110		T::RelayBlockNumberProvider::set_block_number(30_000.into());111	} : _(RawOrigin::Signed(pallet_admin.clone()), Some(b as u8))112113	stake {114		let caller = account::<T::AccountId>("caller", 0, SEED);115		let share = Perbill::from_rational(1u32, 10);116		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());117	} : _(RawOrigin::Signed(caller.clone()), share * <T as Config>::Currency::total_balance(&caller))118119	unstake_all {120		let caller = account::<T::AccountId>("caller", 0, SEED);121		let share = Perbill::from_rational(1u32, 20);122		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());123		(1..11).map(|i| {124			// used to change block number125			<frame_system::Pallet<T>>::set_block_number(i.into());126			T::RelayBlockNumberProvider::set_block_number((2*i).into());127			assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());128			assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());129			PromototionPallet::<T>::stake(RawOrigin::Signed(caller.clone()).into(), share * <T as Config>::Currency::total_balance(&caller))130		}).collect::<Result<Vec<_>, _>>()?;131132	} : _(RawOrigin::Signed(caller.clone()))133134	unstake_partial {135		let caller = account::<T::AccountId>("caller", 0, SEED);136		let share = Perbill::from_rational(1u32, 20);137		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());138		(1..11).map(|i| {139			// used to change block number140			<frame_system::Pallet<T>>::set_block_number(i.into());141			T::RelayBlockNumberProvider::set_block_number((2*i).into());142			assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());143			assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());144			PromototionPallet::<T>::stake(RawOrigin::Signed(caller.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())145		}).collect::<Result<Vec<_>, _>>()?;146147	} : _(RawOrigin::Signed(caller.clone()), Into::<BalanceOf<T>>::into(1000u128) * T::Nominal::get())148149	sponsor_collection {150		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);151		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;152		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());153		let caller: T::AccountId = account("caller", 0, SEED);154		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());155		let collection = create_nft_collection::<T>(caller)?;156	} : _(RawOrigin::Signed(pallet_admin.clone()), collection)157158	stop_sponsoring_collection {159		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);160		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;161		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());162		let caller: T::AccountId = account("caller", 0, SEED);163		let _ = <T as Config>::Currency::write_balance(&caller,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());164		let collection = create_nft_collection::<T>(caller)?;165		PromototionPallet::<T>::sponsor_collection(RawOrigin::Signed(pallet_admin.clone()).into(), collection)?;166	} : _(RawOrigin::Signed(pallet_admin.clone()), collection)167168	sponsor_contract {169		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);170		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;171172		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());173		let address = H160::from_low_u64_be(SEED as u64);174		let data: Vec<u8> = (0..20).collect();175		<EvmMigrationPallet<T>>::begin(RawOrigin::Root.into(), address)?;176		<EvmMigrationPallet<T>>::finish(RawOrigin::Root.into(), address, data)?;177	} : _(RawOrigin::Signed(pallet_admin.clone()), address)178179	stop_sponsoring_contract {180		let pallet_admin = account::<T::AccountId>("admin", 0, SEED);181		PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;182183		let _ = <T as Config>::Currency::write_balance(&pallet_admin,  Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());184		let address = H160::from_low_u64_be(SEED as u64);185		let data: Vec<u8> = (0..20).collect();186		<EvmMigrationPallet<T>>::begin(RawOrigin::Root.into(), address)?;187		<EvmMigrationPallet<T>>::finish(RawOrigin::Root.into(), address, data)?;188		PromototionPallet::<T>::sponsor_contract(RawOrigin::Signed(pallet_admin.clone()).into(), address)?;189	} : _(RawOrigin::Signed(pallet_admin.clone()), address)190}