`ac9b5eeb31df`

source

pallets/refungible/src/benchmarking.rs11.7 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/>.1617use super::*;18use crate::{Pallet, Config, RefungibleHandle};1920use core::convert::TryInto;21use core::iter::IntoIterator;22use frame_benchmarking::{benchmarks, account};23use pallet_common::{24	bench_init,25	benchmarking::{create_collection_raw, property_key, property_value},26};27use sp_std::prelude::*;28use up_data_structs::{29	CollectionMode, MAX_ITEMS_PER_BATCH, MAX_PROPERTIES_PER_ITEM, budget::Unlimited,30	PropertyPermission,31};3233const SEED: u32 = 1;3435fn create_max_item_data<T: Config>(36	users: impl IntoIterator<Item = (T::CrossAccountId, u128)>,37) -> CreateItemData<T> {38	CreateItemData::<T> {39		users: users40			.into_iter()41			.collect::<BTreeMap<_, _>>()42			.try_into()43			.unwrap(),44		properties: Default::default(),45	}46}4748fn create_max_item<T: Config>(49	collection: &RefungibleHandle<T>,50	sender: &T::CrossAccountId,51	users: impl IntoIterator<Item = (T::CrossAccountId, u128)>,52) -> Result<TokenId, DispatchError> {53	let data: CreateItemData<T> = create_max_item_data::<T>(users);54	<Pallet<T>>::create_item(&collection, sender, data, &Unlimited)?;55	Ok(TokenId(<TokensMinted<T>>::get(&collection.id)))56}5758fn create_collection<T: Config>(59	owner: T::CrossAccountId,60) -> Result<RefungibleHandle<T>, DispatchError> {61	create_collection_raw(62		owner,63		CollectionMode::ReFungible,64		|owner: T::CrossAccountId, data| {65			<Pallet<T>>::init_collection(owner.clone(), owner, data, Default::default())66		},67		RefungibleHandle::cast,68	)69}7071benchmarks! {72	create_item {73		bench_init!{74			owner: sub; collection: collection(owner);75			sender: cross_from_sub(owner); to: cross_sub;76		};77	}: {create_max_item(&collection, &sender, [(to.clone(), 200)])?}7879	create_multiple_items {80		let b in 0..MAX_ITEMS_PER_BATCH;81		bench_init!{82			owner: sub; collection: collection(owner);83			sender: cross_from_sub(owner); to: cross_sub;84		};85		let data = (0..b).map(|_| create_max_item_data::<T>([(to.clone(), 200)])).collect();86	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}8788	create_multiple_items_ex_multiple_items {89		let b in 0..MAX_ITEMS_PER_BATCH;90		bench_init!{91			owner: sub; collection: collection(owner);92			sender: cross_from_sub(owner);93		};94		let data = (0..b).map(|t| {95			bench_init!(to: cross_sub(t););96			create_max_item_data::<T>([(to, 200)])97		}).collect();98	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}99100	create_multiple_items_ex_multiple_owners {101		let b in 0..MAX_ITEMS_PER_BATCH;102		bench_init!{103			owner: sub; collection: collection(owner);104			sender: cross_from_sub(owner);105		};106		let data = vec![create_max_item_data::<T>((0..b).map(|u| {107			bench_init!(to: cross_sub(u););108			(to, 200)109		}))].try_into().unwrap();110	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}111112	// Other user left, token data is kept113	burn_item_partial {114		bench_init!{115			owner: sub; collection: collection(owner);116			sender: cross_from_sub(owner); burner: cross_sub; another_owner: cross_sub;117		};118		let item = create_max_item(&collection, &sender, [(burner.clone(), 200), (another_owner, 200)])?;119	}: {<Pallet<T>>::burn(&collection, &burner, item, 200)?}120	// No users remaining, token is destroyed121	burn_item_fully {122		bench_init!{123			owner: sub; collection: collection(owner);124			sender: cross_from_sub(owner); burner: cross_sub; another_owner: cross_sub;125		};126		let item = create_max_item(&collection, &sender, [(burner.clone(), 200)])?;127	}: {<Pallet<T>>::burn(&collection, &burner, item, 200)?}128129	transfer_normal {130		bench_init!{131			owner: sub; collection: collection(owner);132			sender: cross_from_sub(owner); receiver: cross_sub;133		};134		let item = create_max_item(&collection, &sender, [(sender.clone(), 200), (receiver.clone(), 200)])?;135	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 100, &Unlimited)?}136	// Target account is created137	transfer_creating {138		bench_init!{139			owner: sub; collection: collection(owner);140			sender: cross_from_sub(owner); receiver: cross_sub;141		};142		let item = create_max_item(&collection, &sender, [(sender.clone(), 200)])?;143	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 100, &Unlimited)?}144	// Source account is destroyed145	transfer_removing {146		bench_init!{147			owner: sub; collection: collection(owner);148			sender: cross_from_sub(owner); receiver: cross_sub;149		};150		let item = create_max_item(&collection, &sender, [(sender.clone(), 200), (receiver.clone(), 200)])?;151	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 200, &Unlimited)?}152	// Source account destroyed, target created153	transfer_creating_removing {154		bench_init!{155			owner: sub; collection: collection(owner);156			sender: cross_from_sub(owner); receiver: cross_sub;157		};158		let item = create_max_item(&collection, &sender, [(sender.clone(), 200)])?;159	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 200, &Unlimited)?}160161	approve {162		bench_init!{163			owner: sub; collection: collection(owner);164			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;165		};166		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;167	}: {<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?}168169	approve_from {170		bench_init!{171			owner: sub; collection: collection(owner);172			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;173		};174		let owner_eth = T::CrossAccountId::from_eth(*sender.as_eth());175		let item = create_max_item(&collection, &owner, [(owner_eth.clone(), 200)])?;176	}: {<Pallet<T>>::set_allowance_from(&collection, &sender, &owner_eth, &spender, item, 100)?}177178	transfer_from_normal {179		bench_init!{180			owner: sub; collection: collection(owner);181			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;182		};183		let item = create_max_item(&collection, &owner, [(sender.clone(), 200), (receiver.clone(), 200)])?;184		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?;185	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 100, &Unlimited)?}186	// Target account is created187	transfer_from_creating {188		bench_init!{189			owner: sub; collection: collection(owner);190			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;191		};192		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;193		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?;194	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 100, &Unlimited)?}195	// Source account is destroyed196	transfer_from_removing {197		bench_init!{198			owner: sub; collection: collection(owner);199			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;200		};201		let item = create_max_item(&collection, &owner, [(sender.clone(), 200), (receiver.clone(), 200)])?;202		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 200)?;203	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 200, &Unlimited)?}204	// Source account destroyed, target created205	transfer_from_creating_removing {206		bench_init!{207			owner: sub; collection: collection(owner);208			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;209		};210		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;211		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 200)?;212	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 200, &Unlimited)?}213214	// Both source account and token is destroyed215	burn_from {216		bench_init!{217			owner: sub; collection: collection(owner);218			owner: cross_from_sub; sender: cross_sub; burner: cross_sub;219		};220		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;221		<Pallet<T>>::set_allowance(&collection, &sender, &burner, item, 200)?;222	}: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item, 200, &Unlimited)?}223224	set_token_property_permissions {225		let b in 0..MAX_PROPERTIES_PER_ITEM;226		bench_init!{227			owner: sub; collection: collection(owner);228			owner: cross_from_sub;229		};230		let perms = (0..b).map(|k| PropertyKeyPermission {231			key: property_key(k as usize),232			permission: PropertyPermission {233				mutable: false,234				collection_admin: false,235				token_owner: false,236			},237		}).collect::<Vec<_>>();238	}: {<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?}239240	set_token_properties {241		let b in 0..MAX_PROPERTIES_PER_ITEM;242		bench_init!{243			owner: sub; collection: collection(owner);244			owner: cross_from_sub;245		};246		let perms = (0..b).map(|k| PropertyKeyPermission {247			key: property_key(k as usize),248			permission: PropertyPermission {249				mutable: false,250				collection_admin: true,251				token_owner: true,252			},253		}).collect::<Vec<_>>();254		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;255		let props = (0..b).map(|k| Property {256			key: property_key(k as usize),257			value: property_value(),258		}).collect::<Vec<_>>();259		let item = create_max_item(&collection, &owner, [(owner.clone(), 200)])?;260	}: {<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), false, &Unlimited)?}261262	delete_token_properties {263		let b in 0..MAX_PROPERTIES_PER_ITEM;264		bench_init!{265			owner: sub; collection: collection(owner);266			owner: cross_from_sub;267		};268		let perms = (0..b).map(|k| PropertyKeyPermission {269			key: property_key(k as usize),270			permission: PropertyPermission {271				mutable: true,272				collection_admin: true,273				token_owner: true,274			},275		}).collect::<Vec<_>>();276		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;277		let props = (0..b).map(|k| Property {278			key: property_key(k as usize),279			value: property_value(),280		}).collect::<Vec<_>>();281		let item = create_max_item(&collection, &owner, [(owner.clone(), 200)])?;282		<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), false, &Unlimited)?;283		let to_delete = (0..b).map(|k| property_key(k as usize)).collect::<Vec<_>>();284	}: {<Pallet<T>>::delete_token_properties(&collection, &owner, item, to_delete.into_iter(), &Unlimited)?}285286	repartition_item {287		bench_init!{288			owner: sub; collection: collection(owner);289			sender: cross_from_sub(owner); owner: cross_sub;290		};291		let item = create_max_item(&collection, &sender, [(owner.clone(), 100)])?;292	}: {<Pallet<T>>::repartition(&collection, &owner, item, 200)?}293294	token_owner {295		bench_init!{296			owner: sub; collection: collection(owner);297			sender: cross_from_sub(owner); owner: cross_sub;298		};299		let item = create_max_item(&collection, &sender, [(owner.clone(), 100)])?;300	}: {<Pallet<T>>::token_owner(collection.id, item).unwrap()}301302	set_allowance_for_all {303		bench_init!{304			owner: sub; collection: collection(owner); owner: cross_from_sub;305			operator: cross_sub;306		};307	}: {<Pallet<T>>::set_allowance_for_all(&collection, &owner, &operator, true)?}308309	allowance_for_all {310		bench_init!{311			owner: sub; collection: collection(owner); owner: cross_from_sub;312			operator: cross_sub;313		};314	}: {<Pallet<T>>::allowance_for_all(&collection, &owner, &operator)}315316	repair_item {317		bench_init!{318			owner: sub; collection: collection(owner);319			owner: cross_from_sub;320		};321		let item = create_max_item(&collection, &owner, [(owner.clone(), 100)])?;322	}: {<Pallet<T>>::repair_item(&collection, item)?}323}