1use super::*;2use crate::{Pallet, Config, NonfungibleHandle};34use sp_std::prelude::*;5use pallet_common::benchmarking::{create_collection_raw, create_data};6use frame_benchmarking::{benchmarks, account};7use up_data_structs::{CollectionMode, MAX_ITEMS_PER_BATCH};8use pallet_common::bench_init;9use core::convert::TryInto;1011const SEED: u32 = 1;1213fn create_max_item_data<T: Config>(owner: T::CrossAccountId) -> CreateItemData<T> {14 let const_data = create_data(CUSTOM_DATA_LIMIT as usize).try_into().unwrap();15 let variable_data = create_data(CUSTOM_DATA_LIMIT as usize).try_into().unwrap();16 CreateItemData {17 const_data,18 variable_data,19 owner,20 }21}22fn create_max_item<T: Config>(23 collection: &NonfungibleHandle<T>,24 sender: &T::CrossAccountId,25 owner: T::CrossAccountId,26) -> Result<TokenId, DispatchError> {27 <Pallet<T>>::create_item(&collection, sender, create_max_item_data(owner))?;28 Ok(TokenId(<TokensMinted<T>>::get(&collection.id)))29}3031fn create_collection<T: Config>(32 owner: T::AccountId,33) -> Result<NonfungibleHandle<T>, DispatchError> {34 create_collection_raw(35 owner,36 CollectionMode::NFT,37 <Pallet<T>>::init_collection,38 NonfungibleHandle::cast,39 )40}4142benchmarks! {43 create_item {44 bench_init!{45 owner: sub; collection: collection(owner);46 sender: cross_from_sub(owner); to: cross_sub;47 };48 }: {create_max_item(&collection, &sender, to.clone())?}4950 create_multiple_items {51 let b in 0..MAX_ITEMS_PER_BATCH;52 bench_init!{53 owner: sub; collection: collection(owner);54 sender: cross_from_sub(owner); to: cross_sub;55 };56 let data = (0..b).map(|_| create_max_item_data(to.clone())).collect();57 }: {<Pallet<T>>::create_multiple_items(&collection, &sender, data)?}5859 burn_item {60 bench_init!{61 owner: sub; collection: collection(owner);62 sender: cross_from_sub(owner); burner: cross_sub;63 };64 let item = create_max_item(&collection, &sender, burner.clone())?;65 }: {<Pallet<T>>::burn(&collection, &burner, item)?}6667 transfer {68 bench_init!{69 owner: sub; collection: collection(owner);70 owner: cross_from_sub; sender: cross_sub; receiver: cross_sub;71 };72 let item = create_max_item(&collection, &owner, sender.clone())?;73 }: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item)?}7475 approve {76 bench_init!{77 owner: sub; collection: collection(owner);78 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;79 };80 let item = create_max_item(&collection, &owner, sender.clone())?;81 }: {<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?}8283 transfer_from {84 bench_init!{85 owner: sub; collection: collection(owner);86 owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;87 };88 let item = create_max_item(&collection, &owner, sender.clone())?;89 <Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?;90 }: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item)?}9192 burn_from {93 bench_init!{94 owner: sub; collection: collection(owner);95 owner: cross_from_sub; sender: cross_sub; burner: cross_sub;96 };97 let item = create_max_item(&collection, &owner, sender.clone())?;98 <Pallet<T>>::set_allowance(&collection, &sender, item, Some(&burner))?;99 }: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item)?}100101 set_variable_metadata {102 let b in 0..CUSTOM_DATA_LIMIT;103 bench_init!{104 owner: sub; collection: collection(owner);105 owner: cross_from_sub; sender: cross_sub;106 };107 let item = create_max_item(&collection, &owner, sender.clone())?;108 let data = create_data(b as usize);109 }: {<Pallet<T>>::set_variable_metadata(&collection, &sender, item, data)?}110}