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, budget::Unlimited};24use pallet_common::bench_init;2526const SEED: u32 = 1;2728fn create_collection<T: Config>(29 owner: T::CrossAccountId,30) -> Result<FungibleHandle<T>, DispatchError> {31 create_collection_raw(32 owner,33 CollectionMode::Fungible(0),34 |owner: T::CrossAccountId, data| <Pallet<T>>::init_collection(owner.clone(), owner, data),35 FungibleHandle::cast,36 )37}3839benchmarks! {40 create_item {41 bench_init!{42 owner: sub; collection: collection(owner);43 sender: cross_from_sub(owner); to: cross_sub;44 };45 }: {<Pallet<T>>::create_item(&collection, &sender, (to, 200), &Unlimited)?}4647 create_multiple_items_ex {48 let b in 0..MAX_ITEMS_PER_BATCH;49 bench_init!{50 owner: sub; collection: collection(owner);51 sender: cross_from_sub(owner);52 };53 let data = (0..b).map(|i| {54 bench_init!(to: cross_sub(i););55 (to, 200)56 }).collect::<BTreeMap<_, _>>();57 }: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}5859 burn_item {60 bench_init!{61 owner: sub; collection: collection(owner);62 owner: cross_from_sub; burner: cross_sub;63 };64 <Pallet<T>>::create_item(&collection, &owner, (burner.clone(), 200), &Unlimited)?;65 }: {<Pallet<T>>::burn(&collection, &burner, 100)?}6667 transfer_raw {68 bench_init!{69 owner: sub; collection: collection(owner);70 owner: cross_from_sub; sender: cross_sub; to: cross_sub;71 };72 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200), &Unlimited)?;73 }: {<Pallet<T>>::transfer(&collection, &sender, &to, 200, &Unlimited)?}7475 approve {76 bench_init!{77 owner: sub; collection: collection(owner);78 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;79 };80 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200), &Unlimited)?;81 }: {<Pallet<T>>::set_allowance(&collection, &sender, &spender, 100)?}8283 approve_from {84 bench_init!{85 owner: sub; collection: collection(owner);86 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;8788 };89 let owner_eth = T::CrossAccountId::from_eth(*sender.as_eth());90 <Pallet<T>>::create_item(&collection, &owner, (owner_eth.clone(), 200), &Unlimited)?;91 }: {<Pallet<T>>::set_allowance_from(&collection, &sender, &owner_eth, &spender, 100)?}9293 check_allowed_raw {94 bench_init!{95 owner: sub; collection: collection(owner);96 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;97 };98 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200), &Unlimited)?;99 <Pallet<T>>::set_allowance(&collection, &sender, &spender, 200)?;100 }: {<Pallet<T>>::check_allowed(&collection, &spender, &sender, 200, &Unlimited)?;}101102 set_allowance_unchecked_raw {103 bench_init!{104 owner: sub; collection: collection(owner);105 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;106 };107 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200), &Unlimited)?;108 }: {<Pallet<T>>::set_allowance_unchecked(&collection, &sender, &spender, 200);}109110 burn_from {111 bench_init!{112 owner: sub; collection: collection(owner);113 owner: cross_from_sub; sender: cross_sub; burner: cross_sub;114 };115 <Pallet<T>>::create_item(&collection, &owner, (sender.clone(), 200), &Unlimited)?;116 <Pallet<T>>::set_allowance(&collection, &sender, &burner, 200)?;117 }: {<Pallet<T>>::burn_from(&collection, &burner, &sender, 100, &Unlimited)?}118}