git.delta.rocks / unique-network / refs/commits / e48f2d1c76b5

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 core::{convert::TryInto, iter::IntoIterator};1819use frame_benchmarking::{account, benchmarks};20use pallet_common::{21	bench_init,22	benchmarking::{23		create_collection_raw, property_key, property_value, load_is_admin_and_property_permissions,24	},25};26use sp_std::prelude::*;27use up_data_structs::{28	budget::Unlimited, CollectionMode, PropertyPermission, MAX_ITEMS_PER_BATCH,29	MAX_PROPERTIES_PER_ITEM,30};3132use super::*;33use crate::{Config, Pallet, RefungibleHandle};3435const SEED: u32 = 1;3637fn create_max_item_data<T: Config>(38	users: impl IntoIterator<Item = (T::CrossAccountId, u128)>,39) -> CreateItemData<T> {40	CreateItemData::<T> {41		users: users42			.into_iter()43			.collect::<BTreeMap<_, _>>()44			.try_into()45			.unwrap(),46		properties: Default::default(),47	}48}4950fn create_max_item<T: Config>(51	collection: &RefungibleHandle<T>,52	sender: &T::CrossAccountId,53	users: impl IntoIterator<Item = (T::CrossAccountId, u128)>,54) -> Result<TokenId, DispatchError> {55	let data: CreateItemData<T> = create_max_item_data::<T>(users);56	<Pallet<T>>::create_item(collection, sender, data, &Unlimited)?;57	Ok(TokenId(<TokensMinted<T>>::get(collection.id)))58}5960fn create_collection<T: Config>(61	owner: T::CrossAccountId,62) -> Result<RefungibleHandle<T>, DispatchError> {63	create_collection_raw(64		owner,65		CollectionMode::ReFungible,66		|owner: T::CrossAccountId, data| <Pallet<T>>::init_collection(owner.clone(), owner, data),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		}))];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(), &Unlimited)?}261262	init_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		};268269		let perms = (0..b).map(|k| PropertyKeyPermission {270			key: property_key(k as usize),271			permission: PropertyPermission {272				mutable: false,273				collection_admin: true,274				token_owner: true,275			},276		}).collect::<Vec<_>>();277		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;278		let props = (0..b).map(|k| Property {279			key: property_key(k as usize),280			value: property_value(),281		}).collect::<Vec<_>>();282		let item = create_max_item(&collection, &owner, [(owner.clone(), 200)])?;283284		let (is_collection_admin, property_permissions) = load_is_admin_and_property_permissions(&collection, &owner);285	}: {286		let mut property_writer = pallet_common::collection_info_loaded_property_writer(287			&collection,288			is_collection_admin,289			property_permissions,290		);291292		property_writer.write_token_properties(293			true,294			item,295			props.into_iter(),296			crate::erc::ERC721TokenEvent::TokenChanged {297				token_id: item.into(),298			}299			.to_log(T::ContractAddress::get()),300		)?301	}302303	delete_token_properties {304		let b in 0..MAX_PROPERTIES_PER_ITEM;305		bench_init!{306			owner: sub; collection: collection(owner);307			owner: cross_from_sub;308		};309		let perms = (0..b).map(|k| PropertyKeyPermission {310			key: property_key(k as usize),311			permission: PropertyPermission {312				mutable: true,313				collection_admin: true,314				token_owner: true,315			},316		}).collect::<Vec<_>>();317		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;318		let props = (0..b).map(|k| Property {319			key: property_key(k as usize),320			value: property_value(),321		}).collect::<Vec<_>>();322		let item = create_max_item(&collection, &owner, [(owner.clone(), 200)])?;323		<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), &Unlimited)?;324		let to_delete = (0..b).map(|k| property_key(k as usize)).collect::<Vec<_>>();325	}: {<Pallet<T>>::delete_token_properties(&collection, &owner, item, to_delete.into_iter(), &Unlimited)?}326327	repartition_item {328		bench_init!{329			owner: sub; collection: collection(owner);330			sender: cross_from_sub(owner); owner: cross_sub;331		};332		let item = create_max_item(&collection, &sender, [(owner.clone(), 100)])?;333	}: {<Pallet<T>>::repartition(&collection, &owner, item, 200)?}334335	token_owner {336		bench_init!{337			owner: sub; collection: collection(owner);338			sender: cross_from_sub(owner); owner: cross_sub;339		};340		let item = create_max_item(&collection, &sender, [(owner, 100)])?;341	}: {<Pallet<T>>::token_owner(collection.id, item).unwrap()}342343	set_allowance_for_all {344		bench_init!{345			owner: sub; collection: collection(owner); owner: cross_from_sub;346			operator: cross_sub;347		};348	}: {<Pallet<T>>::set_allowance_for_all(&collection, &owner, &operator, true)?}349350	allowance_for_all {351		bench_init!{352			owner: sub; collection: collection(owner); owner: cross_from_sub;353			operator: cross_sub;354		};355	}: {<Pallet<T>>::allowance_for_all(&collection, &owner, &operator)}356357	repair_item {358		bench_init!{359			owner: sub; collection: collection(owner);360			owner: cross_from_sub;361		};362		let item = create_max_item(&collection, &owner, [(owner.clone(), 100)])?;363	}: {<Pallet<T>>::repair_item(&collection, item)?}364}