1234567891011121314151617use super::*;18use crate::{Pallet, Config, FungibleHandle};1920use sp_std::prelude::*;21use pallet_common::benchmarking::create_collection_raw;22use frame_benchmarking::{benchmarks, account};23use up_data_structs::{CollectionMode, MAX_ITEMS_PER_BATCH};24use pallet_common::bench_init;2526const SEED: u32 = 1;2728fn create_collection<T: Config>(owner: T::AccountId) -> Result<FungibleHandle<T>, DispatchError> {29 create_collection_raw(30 owner,31 CollectionMode::Fungible(0),32 <Pallet<T>>::init_collection,33 FungibleHandle::cast,34 )35}3637benchmarks! {38 create_item {39 bench_init!{40 owner: sub; collection: collection(owner);41 sender: cross_from_sub(owner); to: cross_sub;42 };43 }: {<Pallet<T>>::create_item(&collection, &sender, (to, 200))?}4445 create_multiple_items_ex {46 let b in 0..MAX_ITEMS_PER_BATCH;47 bench_init!{48 owner: sub; collection: collection(owner);49 sender: cross_from_sub(owner);50 };51 let data = (0..b).map(|i| {52 bench_init!(to: cross_sub(i););53 (to, 200)54 }).collect::<BTreeMap<_, _>>().try_into().unwrap();55 }: {<Pallet<T>>::create_multiple_items(&collection, &sender, data)}5657 burn_item {58 bench_init!{59 owner: sub; collection: collection(owner);60 owner: cross_from_sub; burner: cross_sub;61 };62 <Pallet<T>>::create_item(&collection, &owner, (burner.clone(), 200))?;63 }: {<Pallet<T>>::burn(&collection, &burner, 100)?}6465 transfer {66 bench_init!{67 owner: sub; collection: collection(owner);68 owner: cross_from_sub; sender: cross_sub; to: cross_sub;69 };70 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200))?;71 }: {<Pallet<T>>::transfer(&collection, &sender, &to, 200)?}7273 approve {74 bench_init!{75 owner: sub; collection: collection(owner);76 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;77 };78 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200))?;79 }: {<Pallet<T>>::set_allowance(&collection, &sender, &spender, 100)?}8081 transfer_from {82 bench_init!{83 owner: sub; collection: collection(owner);84 owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;85 };86 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200))?;87 <Pallet<T>>::set_allowance(&collection, &sender, &spender, 200)?;88 }: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, 100)?}8990 burn_from {91 bench_init!{92 owner: sub; collection: collection(owner);93 owner: cross_from_sub; sender: cross_sub; burner: cross_sub;94 };95 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200))?;96 <Pallet<T>>::set_allowance(&collection, &sender, &burner, 200)?;97 }: {<Pallet<T>>::burn_from(&collection, &burner, &sender, 100)?}98}