1234567891011121314151617use super::*;18use crate::{Pallet, Config, NonfungibleHandle};1920use sp_std::prelude::*;21use pallet_common::benchmarking::{create_collection_raw, property_key, property_value};22use frame_benchmarking::{benchmarks, account};23use up_data_structs::{CollectionMode, MAX_ITEMS_PER_BATCH, MAX_PROPERTIES_PER_ITEM, budget::Unlimited};24use pallet_common::bench_init;2526const SEED: u32 = 1;2728fn create_max_item_data<T: Config>(owner: T::CrossAccountId) -> CreateItemData<T> {29 CreateItemData::<T> {30 owner,31 properties: Default::default(),32 }33}34fn create_max_item<T: Config>(35 collection: &NonfungibleHandle<T>,36 sender: &T::CrossAccountId,37 owner: T::CrossAccountId,38) -> Result<TokenId, DispatchError> {39 <Pallet<T>>::create_item(40 &collection,41 sender,42 create_max_item_data::<T>(owner),43 &Unlimited,44 )?;45 Ok(TokenId(<TokensMinted<T>>::get(&collection.id)))46}4748fn create_collection<T: Config>(49 owner: T::CrossAccountId,50) -> Result<NonfungibleHandle<T>, DispatchError> {51 create_collection_raw(52 owner,53 CollectionMode::NFT,54 |owner, data| <Pallet<T>>::init_collection(owner, data, true),55 NonfungibleHandle::cast,56 )57}5859benchmarks! {60 create_item {61 bench_init!{62 owner: sub; collection: collection(owner);63 sender: cross_from_sub(owner); to: cross_sub;64 };65 }: {create_max_item(&collection, &sender, to.clone())?}6667 create_multiple_items {68 let b in 0..MAX_ITEMS_PER_BATCH;69 bench_init!{70 owner: sub; collection: collection(owner);71 sender: cross_from_sub(owner); to: cross_sub;72 };73 let data = (0..b).map(|_| create_max_item_data::<T>(to.clone())).collect();74 }: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}7576 create_multiple_items_ex {77 let b in 0..MAX_ITEMS_PER_BATCH;78 bench_init!{79 owner: sub; collection: collection(owner);80 sender: cross_from_sub(owner);81 };82 let data = (0..b).map(|i| {83 bench_init!(to: cross_sub(i););84 create_max_item_data::<T>(to)85 }).collect();86 }: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}8788 burn_item {89 bench_init!{90 owner: sub; collection: collection(owner);91 sender: cross_from_sub(owner); burner: cross_sub;92 };93 let item = create_max_item(&collection, &sender, burner.clone())?;94 }: {<Pallet<T>>::burn(&collection, &burner, item)?}9596 burn_recursively_self_raw {97 bench_init!{98 owner: sub; collection: collection(owner);99 sender: cross_from_sub(owner); burner: cross_sub;100 };101 let item = create_max_item(&collection, &sender, burner.clone())?;102 }: {<Pallet<T>>::burn_recursively(&collection, &burner, item, &Unlimited, &Unlimited)?}103104 burn_recursively_breadth_plus_self_plus_self_per_each_raw {105 let b in 0..200;106 bench_init!{107 owner: sub; collection: collection(owner);108 sender: cross_from_sub(owner); burner: cross_sub;109 };110 let item = create_max_item(&collection, &sender, burner.clone())?;111 for i in 0..b {112 create_max_item(&collection, &sender, T::CrossTokenAddressMapping::token_to_address(collection.id, item))?;113 }114 }: {<Pallet<T>>::burn_recursively(&collection, &burner, item, &Unlimited, &Unlimited)?}115116 transfer {117 bench_init!{118 owner: sub; collection: collection(owner);119 owner: cross_from_sub; sender: cross_sub; receiver: cross_sub;120 };121 let item = create_max_item(&collection, &owner, sender.clone())?;122 }: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, &Unlimited)?}123124 approve {125 bench_init!{126 owner: sub; collection: collection(owner);127 owner: cross_from_sub; sender: cross_sub; spender: cross_sub;128 };129 let item = create_max_item(&collection, &owner, sender.clone())?;130 }: {<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?}131132 transfer_from {133 bench_init!{134 owner: sub; collection: collection(owner);135 owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;136 };137 let item = create_max_item(&collection, &owner, sender.clone())?;138 <Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?;139 }: {<Pallet<T>>::transfer_from(&collection, &spender, &sender, &receiver, item, &Unlimited)?}140141 burn_from {142 bench_init!{143 owner: sub; collection: collection(owner);144 owner: cross_from_sub; sender: cross_sub; burner: cross_sub;145 };146 let item = create_max_item(&collection, &owner, sender.clone())?;147 <Pallet<T>>::set_allowance(&collection, &sender, item, Some(&burner))?;148 }: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item, &Unlimited)?}149150 set_token_property_permissions {151 let b in 0..MAX_PROPERTIES_PER_ITEM;152 bench_init!{153 owner: sub; collection: collection(owner);154 owner: cross_from_sub;155 };156 let perms = (0..b).map(|k| PropertyKeyPermission {157 key: property_key(k as usize),158 permission: PropertyPermission {159 mutable: false,160 collection_admin: false,161 token_owner: false,162 },163 }).collect::<Vec<_>>();164 }: {<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?}165166 set_token_properties {167 let b in 0..MAX_PROPERTIES_PER_ITEM;168 bench_init!{169 owner: sub; collection: collection(owner);170 owner: cross_from_sub;171 };172 let perms = (0..b).map(|k| PropertyKeyPermission {173 key: property_key(k as usize),174 permission: PropertyPermission {175 mutable: false,176 collection_admin: true,177 token_owner: true,178 },179 }).collect::<Vec<_>>();180 <Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;181 let props = (0..b).map(|k| Property {182 key: property_key(k as usize),183 value: property_value(),184 }).collect::<Vec<_>>();185 let item = create_max_item(&collection, &owner, owner.clone())?;186 }: {<Pallet<T>>::set_token_properties(&collection, &owner, item, props, false, &Unlimited)?}187188 delete_token_properties {189 let b in 0..MAX_PROPERTIES_PER_ITEM;190 bench_init!{191 owner: sub; collection: collection(owner);192 owner: cross_from_sub;193 };194 let perms = (0..b).map(|k| PropertyKeyPermission {195 key: property_key(k as usize),196 permission: PropertyPermission {197 mutable: true,198 collection_admin: true,199 token_owner: true,200 },201 }).collect::<Vec<_>>();202 <Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;203 let props = (0..b).map(|k| Property {204 key: property_key(k as usize),205 value: property_value(),206 }).collect::<Vec<_>>();207 let item = create_max_item(&collection, &owner, owner.clone())?;208 <Pallet<T>>::set_token_properties(&collection, &owner, item, props, false, &Unlimited)?;209 let to_delete = (0..b).map(|k| property_key(k as usize)).collect::<Vec<_>>();210 }: {<Pallet<T>>::delete_token_properties(&collection, &owner, item, to_delete, &Unlimited)?}211}