git.delta.rocks / unique-network / refs/commits / 45f4f7d0be2d

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 sp_std::prelude::*;21use pallet_common::benchmarking::{create_collection_raw, property_key, property_value, create_data};22use frame_benchmarking::{benchmarks, account};23use up_data_structs::{24	CollectionMode, MAX_ITEMS_PER_BATCH, MAX_PROPERTIES_PER_ITEM, CUSTOM_DATA_LIMIT,25	budget::Unlimited,26};27use pallet_common::bench_init;28use core::convert::TryInto;29use core::iter::IntoIterator;3031const SEED: u32 = 1;3233fn create_max_item_data<CrossAccountId: Ord>(34	users: impl IntoIterator<Item = (CrossAccountId, u128)>,35) -> CreateRefungibleExData<CrossAccountId> {36	let const_data = create_data::<CUSTOM_DATA_LIMIT>();37	CreateRefungibleExData {38		const_data,39		users: users40			.into_iter()41			.collect::<BTreeMap<_, _>>()42			.try_into()43			.unwrap(),44		properties: Default::default(),45	}46}47fn create_max_item<T: Config>(48	collection: &RefungibleHandle<T>,49	sender: &T::CrossAccountId,50	users: impl IntoIterator<Item = (T::CrossAccountId, u128)>,51) -> Result<TokenId, DispatchError> {52	let data: CreateRefungibleExData<T::CrossAccountId> = create_max_item_data(users);53	<Pallet<T>>::create_item(&collection, sender, data, &Unlimited)?;54	Ok(TokenId(<TokensMinted<T>>::get(&collection.id)))55}5657fn create_collection<T: Config>(58	owner: T::CrossAccountId,59) -> Result<RefungibleHandle<T>, DispatchError> {60	create_collection_raw(61		owner,62		CollectionMode::NFT,63		<Pallet<T>>::init_collection,64		RefungibleHandle::cast,65	)66}67benchmarks! {68	create_item {69		bench_init!{70			owner: sub; collection: collection(owner);71			sender: cross_from_sub(owner); to: cross_sub;72		};73	}: {create_max_item(&collection, &sender, [(to.clone(), 200)])?}7475	create_multiple_items {76		let b in 0..MAX_ITEMS_PER_BATCH;77		bench_init!{78			owner: sub; collection: collection(owner);79			sender: cross_from_sub(owner); to: cross_sub;80		};81		let data = (0..b).map(|_| create_max_item_data([(to.clone(), 200)])).collect();82	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}8384	create_multiple_items_ex_multiple_items {85		let b in 0..MAX_ITEMS_PER_BATCH;86		bench_init!{87			owner: sub; collection: collection(owner);88			sender: cross_from_sub(owner);89		};90		let data = (0..b).map(|t| {91			bench_init!(to: cross_sub(t););92			create_max_item_data([(to, 200)])93		}).collect();94	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}9596	create_multiple_items_ex_multiple_owners {97		let b in 0..MAX_ITEMS_PER_BATCH;98		bench_init!{99			owner: sub; collection: collection(owner);100			sender: cross_from_sub(owner);101		};102		let data = vec![create_max_item_data((0..b).map(|u| {103			bench_init!(to: cross_sub(u););104			(to, 200)105		}))].try_into().unwrap();106	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}107108	// Other user left, token data is kept109	burn_item_partial {110		bench_init!{111			owner: sub; collection: collection(owner);112			sender: cross_from_sub(owner); burner: cross_sub; another_owner: cross_sub;113		};114		let item = create_max_item(&collection, &sender, [(burner.clone(), 200), (another_owner, 200)])?;115	}: {<Pallet<T>>::burn(&collection, &burner, item, 200)?}116	// No users remaining, token is destroyed117	burn_item_fully {118		bench_init!{119			owner: sub; collection: collection(owner);120			sender: cross_from_sub(owner); burner: cross_sub; another_owner: cross_sub;121		};122		let item = create_max_item(&collection, &sender, [(burner.clone(), 200)])?;123	}: {<Pallet<T>>::burn(&collection, &burner, item, 200)?}124125	transfer_normal {126		bench_init!{127			owner: sub; collection: collection(owner);128			sender: cross_from_sub(owner); receiver: cross_sub;129		};130		let item = create_max_item(&collection, &sender, [(sender.clone(), 200), (receiver.clone(), 200)])?;131	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 100, &Unlimited)?}132	// Target account is created133	transfer_creating {134		bench_init!{135			owner: sub; collection: collection(owner);136			sender: cross_from_sub(owner); receiver: cross_sub;137		};138		let item = create_max_item(&collection, &sender, [(sender.clone(), 200)])?;139	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 100, &Unlimited)?}140	// Source account is destroyed141	transfer_removing {142		bench_init!{143			owner: sub; collection: collection(owner);144			sender: cross_from_sub(owner); receiver: cross_sub;145		};146		let item = create_max_item(&collection, &sender, [(sender.clone(), 200), (receiver.clone(), 200)])?;147	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 200, &Unlimited)?}148	// Source account destroyed, target created149	transfer_creating_removing {150		bench_init!{151			owner: sub; collection: collection(owner);152			sender: cross_from_sub(owner); receiver: cross_sub;153		};154		let item = create_max_item(&collection, &sender, [(sender.clone(), 200)])?;155	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 200, &Unlimited)?}156157	approve {158		bench_init!{159			owner: sub; collection: collection(owner);160			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;161		};162		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;163	}: {<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?}164165	transfer_from_normal {166		bench_init!{167			owner: sub; collection: collection(owner);168			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;169		};170		let item = create_max_item(&collection, &owner, [(sender.clone(), 200), (receiver.clone(), 200)])?;171		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?;172	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 100, &Unlimited)?}173	// Target account is created174	transfer_from_creating {175		bench_init!{176			owner: sub; collection: collection(owner);177			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;178		};179		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;180		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?;181	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 100, &Unlimited)?}182	// Source account is destroyed183	transfer_from_removing {184		bench_init!{185			owner: sub; collection: collection(owner);186			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;187		};188		let item = create_max_item(&collection, &owner, [(sender.clone(), 200), (receiver.clone(), 200)])?;189		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 200)?;190	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 200, &Unlimited)?}191	// Source account destroyed, target created192	transfer_from_creating_removing {193		bench_init!{194			owner: sub; collection: collection(owner);195			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;196		};197		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;198		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 200)?;199	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 200, &Unlimited)?}200201	// Both source account and token is destroyed202	burn_from {203		bench_init!{204			owner: sub; collection: collection(owner);205			owner: cross_from_sub; sender: cross_sub; burner: cross_sub;206		};207		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;208		<Pallet<T>>::set_allowance(&collection, &sender, &burner, item, 200)?;209	}: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item, 200, &Unlimited)?}210211	set_token_property_permissions {212		let b in 0..MAX_PROPERTIES_PER_ITEM;213		bench_init!{214			owner: sub; collection: collection(owner);215			owner: cross_from_sub;216		};217		let perms = (0..b).map(|k| PropertyKeyPermission {218			key: property_key(k as usize),219			permission: PropertyPermission {220				mutable: false,221				collection_admin: false,222				token_owner: false,223			},224		}).collect::<Vec<_>>();225	}: {<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?}226227	set_token_properties {228		let b in 0..MAX_PROPERTIES_PER_ITEM;229		bench_init!{230			owner: sub; collection: collection(owner);231			owner: cross_from_sub;232		};233		let perms = (0..b).map(|k| PropertyKeyPermission {234			key: property_key(k as usize),235			permission: PropertyPermission {236				mutable: false,237				collection_admin: true,238				token_owner: true,239			},240		}).collect::<Vec<_>>();241		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;242		let props = (0..b).map(|k| Property {243			key: property_key(k as usize),244			value: property_value(),245		}).collect::<Vec<_>>();246		let item = create_max_item(&collection, &owner, [(owner.clone(), 200)])?;247	}: {<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), false, &Unlimited)?}248249	delete_token_properties {250		let b in 0..MAX_PROPERTIES_PER_ITEM;251		bench_init!{252			owner: sub; collection: collection(owner);253			owner: cross_from_sub;254		};255		let perms = (0..b).map(|k| PropertyKeyPermission {256			key: property_key(k as usize),257			permission: PropertyPermission {258				mutable: true,259				collection_admin: true,260				token_owner: true,261			},262		}).collect::<Vec<_>>();263		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;264		let props = (0..b).map(|k| Property {265			key: property_key(k as usize),266			value: property_value(),267		}).collect::<Vec<_>>();268		let item = create_max_item(&collection, &owner, [(owner.clone(), 200)])?;269		<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), false, &Unlimited)?;270		let to_delete = (0..b).map(|k| property_key(k as usize)).collect::<Vec<_>>();271	}: {<Pallet<T>>::delete_token_properties(&collection, &owner, item, to_delete.into_iter(), &Unlimited)?}272273	repartition_item {274		bench_init!{275			owner: sub; collection: collection(owner);276			sender: cross_from_sub(owner); owner: cross_sub;277		};278		let item = create_max_item(&collection, &sender, [(owner.clone(), 100)])?;279	}: {<Pallet<T>>::repartition(&collection, &owner, item, 200)?}280}