git.delta.rocks / unique-network / refs/commits / 764dfd599fff

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, NonfungibleHandle};1920use sp_std::prelude::*;21use pallet_common::benchmarking::{create_collection_raw, create_data, property_key, property_value};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;2829const SEED: u32 = 1;3031fn create_max_item_data<T: Config>(owner: T::CrossAccountId) -> CreateItemData<T> {32	CreateItemData::<T> {33		owner,34		properties: Default::default(),35	}36}37fn create_max_item<T: Config>(38	collection: &NonfungibleHandle<T>,39	sender: &T::CrossAccountId,40	owner: T::CrossAccountId,41) -> Result<TokenId, DispatchError> {42	<Pallet<T>>::create_item(43		&collection,44		sender,45		create_max_item_data::<T>(owner),46		&Unlimited,47	)?;48	Ok(TokenId(<TokensMinted<T>>::get(&collection.id)))49}5051fn create_collection<T: Config>(52	owner: T::AccountId,53) -> Result<NonfungibleHandle<T>, DispatchError> {54	create_collection_raw(55		owner,56		CollectionMode::NFT,57		<Pallet<T>>::init_collection,58		NonfungibleHandle::cast,59	)60}6162benchmarks! {63	create_item {64		bench_init!{65			owner: sub; collection: collection(owner);66			sender: cross_from_sub(owner); to: cross_sub;67		};68	}: {create_max_item(&collection, &sender, to.clone())?}6970	create_multiple_items {71		let b in 0..MAX_ITEMS_PER_BATCH;72		bench_init!{73			owner: sub; collection: collection(owner);74			sender: cross_from_sub(owner); to: cross_sub;75		};76		let data = (0..b).map(|_| create_max_item_data::<T>(to.clone())).collect();77	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}7879	create_multiple_items_ex {80		let b in 0..MAX_ITEMS_PER_BATCH;81		bench_init!{82			owner: sub; collection: collection(owner);83			sender: cross_from_sub(owner);84		};85		let data = (0..b).map(|i| {86			bench_init!(to: cross_sub(i););87			create_max_item_data::<T>(to)88		}).collect();89	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}9091	burn_item {92		bench_init!{93			owner: sub; collection: collection(owner);94			sender: cross_from_sub(owner); burner: cross_sub;95		};96		let item = create_max_item(&collection, &sender, burner.clone())?;97	}: {<Pallet<T>>::burn(&collection, &burner, item)?}9899	transfer {100		bench_init!{101			owner: sub; collection: collection(owner);102			owner: cross_from_sub; sender: cross_sub; receiver: cross_sub;103		};104		let item = create_max_item(&collection, &owner, sender.clone())?;105	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, &Unlimited)?}106107	approve {108		bench_init!{109			owner: sub; collection: collection(owner);110			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;111		};112		let item = create_max_item(&collection, &owner, sender.clone())?;113	}: {<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?}114115	transfer_from {116		bench_init!{117			owner: sub; collection: collection(owner);118			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;119		};120		let item = create_max_item(&collection, &owner, sender.clone())?;121		<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?;122	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, &Unlimited)?}123124	burn_from {125		bench_init!{126			owner: sub; collection: collection(owner);127			owner: cross_from_sub; sender: cross_sub; burner: cross_sub;128		};129		let item = create_max_item(&collection, &owner, sender.clone())?;130		<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&burner))?;131	}: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item, &Unlimited)?}132133	set_property_permissions {134		let b in 0..MAX_PROPERTIES_PER_ITEM;135		bench_init!{136			owner: sub; collection: collection(owner);137			owner: cross_from_sub;138		};139		let perms = (0..b).map(|k| PropertyKeyPermission {140			key: property_key(k as usize),141			permission: PropertyPermission {142				mutable: false,143				collection_admin: false,144				token_owner: false,145			},146		}).collect::<Vec<_>>();147	}: {<Pallet<T>>::set_property_permissions(&collection, &owner, perms)?}148149	set_token_properties {150		let b in 0..MAX_PROPERTIES_PER_ITEM;151		bench_init!{152			owner: sub; collection: collection(owner);153			owner: cross_from_sub;154		};155		let perms = (0..b).map(|k| PropertyKeyPermission {156			key: property_key(k as usize),157			permission: PropertyPermission {158				mutable: false,159				collection_admin: true,160				token_owner: true,161			},162		}).collect::<Vec<_>>();163		<Pallet<T>>::set_property_permissions(&collection, &owner, perms)?;164		let props = (0..b).map(|k| Property {165			key: property_key(k as usize),166			value: property_value(),167		}).collect::<Vec<_>>();168		let item = create_max_item(&collection, &owner, owner.clone())?;169	}: {<Pallet<T>>::set_token_properties(&collection, &owner, item, props)?}170171	delete_token_properties {172		let b in 0..MAX_PROPERTIES_PER_ITEM;173		bench_init!{174			owner: sub; collection: collection(owner);175			owner: cross_from_sub;176		};177		let perms = (0..b).map(|k| PropertyKeyPermission {178			key: property_key(k as usize),179			permission: PropertyPermission {180				mutable: true,181				collection_admin: true,182				token_owner: true,183			},184		}).collect::<Vec<_>>();185		<Pallet<T>>::set_property_permissions(&collection, &owner, perms)?;186		let props = (0..b).map(|k| Property {187			key: property_key(k as usize),188			value: property_value(),189		}).collect::<Vec<_>>();190		let item = create_max_item(&collection, &owner, owner.clone())?;191		<Pallet<T>>::set_token_properties(&collection, &owner, item, props)?;192		let to_delete = (0..b).map(|k| property_key(k as usize)).collect::<Vec<_>>();193	}: {<Pallet<T>>::delete_token_properties(&collection, &owner, item, to_delete)?}194}