git.delta.rocks / unique-network / refs/commits / 2f9c8d434e2f

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, create_data, create_var_data};22use frame_benchmarking::{benchmarks, account};23use up_data_structs::{CollectionMode, MAX_ITEMS_PER_BATCH, CUSTOM_DATA_LIMIT};24use pallet_common::bench_init;25use core::convert::TryInto;26use core::iter::IntoIterator;2728const SEED: u32 = 1;2930fn create_max_item_data<CrossAccountId: Ord>(31	users: impl IntoIterator<Item = (CrossAccountId, u128)>,32) -> CreateRefungibleExData<CrossAccountId> {33	let const_data = create_data::<CUSTOM_DATA_LIMIT>();34	let variable_data = create_data::<CUSTOM_DATA_LIMIT>();35	CreateRefungibleExData {36		const_data,37		variable_data,38		users: users39			.into_iter()40			.collect::<BTreeMap<_, _>>()41			.try_into()42			.unwrap(),43	}44}45fn create_max_item<T: Config>(46	collection: &RefungibleHandle<T>,47	sender: &T::CrossAccountId,48	users: impl IntoIterator<Item = (T::CrossAccountId, u128)>,49) -> Result<TokenId, DispatchError> {50	let data: CreateRefungibleExData<T::CrossAccountId> = create_max_item_data(users);51	<Pallet<T>>::create_item(&collection, sender, data)?;52	Ok(TokenId(<TokensMinted<T>>::get(&collection.id)))53}5455fn create_collection<T: Config>(owner: T::AccountId) -> Result<RefungibleHandle<T>, DispatchError> {56	create_collection_raw(57		owner,58		CollectionMode::NFT,59		<Pallet<T>>::init_collection,60		RefungibleHandle::cast,61	)62}63benchmarks! {64	create_item {65		bench_init!{66			owner: sub; collection: collection(owner);67			sender: cross_from_sub(owner); to: cross_sub;68		};69	}: {create_max_item(&collection, &sender, [(to.clone(), 200)])?}7071	create_multiple_items {72		let b in 0..MAX_ITEMS_PER_BATCH;73		bench_init!{74			owner: sub; collection: collection(owner);75			sender: cross_from_sub(owner); to: cross_sub;76		};77		let data = (0..b).map(|_| create_max_item_data([(to.clone(), 200)])).collect();78	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data)?}7980	create_multiple_items_ex_multiple_items {81		let b in 0..MAX_ITEMS_PER_BATCH;82		bench_init!{83			owner: sub; collection: collection(owner);84			sender: cross_from_sub(owner);85		};86		let data = (0..b).map(|t| {87			bench_init!(to: cross_sub(t););88			create_max_item_data([(to, 200)])89		}).collect();90	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data)?}9192	create_multiple_items_ex_multiple_owners {93		let b in 0..MAX_ITEMS_PER_BATCH;94		bench_init!{95			owner: sub; collection: collection(owner);96			sender: cross_from_sub(owner);97		};98		let data = vec![create_max_item_data((0..b).map(|u| {99			bench_init!(to: cross_sub(u););100			(to, 200)101		}))].try_into().unwrap();102	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data)?}103104	// Other user left, token data is kept105	burn_item_partial {106		bench_init!{107			owner: sub; collection: collection(owner);108			sender: cross_from_sub(owner); burner: cross_sub; another_owner: cross_sub;109		};110		let item = create_max_item(&collection, &sender, [(burner.clone(), 200), (another_owner, 200)])?;111	}: {<Pallet<T>>::burn(&collection, &burner, item, 200)?}112	// No users remaining, token is destroyed113	burn_item_fully {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)])?;119	}: {<Pallet<T>>::burn(&collection, &burner, item, 200)?}120121	transfer_normal {122		bench_init!{123			owner: sub; collection: collection(owner);124			sender: cross_from_sub(owner); receiver: cross_sub;125		};126		let item = create_max_item(&collection, &sender, [(sender.clone(), 200), (receiver.clone(), 200)])?;127	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 100)?}128	// Target account is created129	transfer_creating {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)])?;135	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 100)?}136	// Source account is destroyed137	transfer_removing {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), (receiver.clone(), 200)])?;143	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 200)?}144	// Source account destroyed, target created145	transfer_creating_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)])?;151	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, 200)?}152153	approve {154		bench_init!{155			owner: sub; collection: collection(owner);156			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;157		};158		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;159	}: {<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?}160161	transfer_from_normal {162		bench_init!{163			owner: sub; collection: collection(owner);164			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;165		};166		let item = create_max_item(&collection, &owner, [(sender.clone(), 200), (receiver.clone(), 200)])?;167		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?;168	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 100)?}169	// Target account is created170	transfer_from_creating {171		bench_init!{172			owner: sub; collection: collection(owner);173			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;174		};175		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;176		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 100)?;177	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 100)?}178	// Source account is destroyed179	transfer_from_removing {180		bench_init!{181			owner: sub; collection: collection(owner);182			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;183		};184		let item = create_max_item(&collection, &owner, [(sender.clone(), 200), (receiver.clone(), 200)])?;185		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 200)?;186	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 200)?}187	// Source account destroyed, target created188	transfer_from_creating_removing {189		bench_init!{190			owner: sub; collection: collection(owner);191			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;192		};193		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;194		<Pallet<T>>::set_allowance(&collection, &sender, &spender, item, 200)?;195	}: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, 200)?}196197	// Both source account and token is destroyed198	burn_from {199		bench_init!{200			owner: sub; collection: collection(owner);201			owner: cross_from_sub; sender: cross_sub; burner: cross_sub;202		};203		let item = create_max_item(&collection, &owner, [(sender.clone(), 200)])?;204		<Pallet<T>>::set_allowance(&collection, &sender, &burner, item, 200)?;205	}: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item, 200)?}206207	set_variable_metadata {208		let b in 0..CUSTOM_DATA_LIMIT;209		bench_init!{210			owner: sub; collection: collection(owner);211			sender: cross_from_sub(owner);212		};213		let item = create_max_item(&collection, &sender, [(sender.clone(), 200)])?;214		let data = create_var_data(b).try_into().unwrap();215	}: {<Pallet<T>>::set_variable_metadata(&collection, &sender, item, data)?}216}