1use sp_std::vec;23use frame_benchmarking::{benchmarks, account};4use frame_system::RawOrigin;5use frame_support::{6 traits::{Currency, Get},7 BoundedVec,8};9use sp_runtime::Permill;1011use up_data_structs::*;1213use super::*;1415const SEED: u32 = 1;1617fn create_data<S: Get<u32>>() -> BoundedVec<u8, S> {18 vec![b'A'; S::get() as usize].try_into().expect("size == S")19}2021fn create_u32_array<S: Get<u32>>() -> BoundedVec<u32, S> {22 vec![0; S::get() as usize].try_into().expect("size == S")23}2425fn create_basic_resource() -> RmrkBasicResource {26 RmrkBasicResource {27 src: Some(create_data()),28 metadata: Some(create_data()),29 license: Some(create_data()),30 thumb: Some(create_data()),31 }32}3334fn create_composable_resource() -> RmrkComposableResource {35 RmrkComposableResource {36 parts: create_u32_array(),37 base: 100,38 src: Some(create_data()),39 metadata: Some(create_data()),40 license: Some(create_data()),41 thumb: Some(create_data()),42 }43}4445fn create_slot_resource() -> RmrkSlotResource {46 RmrkSlotResource {47 base: 100,48 slot: 200,49 src: Some(create_data()),50 metadata: Some(create_data()),51 license: Some(create_data()),52 thumb: Some(create_data()),53 }54}5556fn create_max_resource_types_array<S: Get<u32>>(num: usize) -> BoundedVec<RmrkResourceTypes, S> {57 vec![RmrkResourceTypes::Composable(create_composable_resource()); num]58 .try_into()59 .expect("num <= S")60}6162fn create_max_collection<T: Config>(owner: &T::AccountId) -> DispatchResult {63 <T as pallet_common::Config>::Currency::deposit_creating(64 owner,65 T::CollectionCreationPrice::get(),66 );6768 let metadata = create_data();69 let max = None;70 let symbol = create_data();7172 <Pallet<T>>::create_collection(73 RawOrigin::Signed(owner.clone()).into(),74 metadata,75 max,76 symbol,77 )78}7980fn create_nft<T: Config>(owner: &T::AccountId, collection_id: RmrkCollectionId) -> DispatchResult {81 let royalty_recipient = Some(owner.clone());82 let royalty_amount = Some(Permill::from_percent(25));83 let metadata = create_data();84 let transferable = true;8586 <Pallet<T>>::mint_nft(87 RawOrigin::Signed(owner.clone()).into(),88 owner.clone(),89 collection_id,90 royalty_recipient,91 royalty_amount,92 metadata,93 transferable,94 None,95 )96}9798struct NftBuilder {99 collection_id: RmrkCollectionId,100 current_nft_id: RmrkNftId,101}102103impl NftBuilder {104 fn new(collection_id: RmrkCollectionId) -> Self {105 Self {106 collection_id,107 current_nft_id: 0,108 }109 }110111 fn current_nft_id(&self) -> RmrkNftId {112 self.current_nft_id113 }114115 fn build<T: Config>(&mut self, owner: &T::AccountId) -> Result<RmrkNftId, DispatchError> {116 create_nft::<T>(owner, self.collection_id)?;117 self.current_nft_id += 1;118119 Ok(self.current_nft_id)120 }121122 fn build_tower<T: Config>(123 &mut self,124 owner: &T::AccountId,125 height: u32,126 ) -> Result<(RmrkNftId, RmrkNftId), DispatchError> {127 self.build::<T>(owner)?;128129 let root_nft_id = self.current_nft_id;130 let mut prev_nft_id = root_nft_id;131132 for _ in 0..height {133 self.build::<T>(owner)?;134135 let new_owner =136 <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(137 self.collection_id,138 prev_nft_id,139 );140141 <Pallet<T>>::send(142 RawOrigin::Signed(owner.clone()).into(),143 self.collection_id,144 self.current_nft_id,145 new_owner,146 )?;147148 prev_nft_id = self.current_nft_id;149 }150151 let deepest_nft_id = self.current_nft_id;152153 Ok((root_nft_id, deepest_nft_id))154 }155156 fn build_wide_tree<T: Config>(157 &mut self,158 owner: &T::AccountId,159 width: u32,160 ) -> Result<RmrkNftId, DispatchError> {161 self.build::<T>(owner)?;162163 let root_nft_id = self.current_nft_id;164165 let root_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(166 self.collection_id,167 root_nft_id,168 );169170 for _ in 0..width {171 self.build::<T>(owner)?;172173 <Pallet<T>>::send(174 RawOrigin::Signed(owner.clone()).into(),175 self.collection_id,176 self.current_nft_id,177 root_owner.clone(),178 )?;179 }180181 Ok(root_nft_id)182 }183}184185benchmarks! {186 create_collection {187 let caller = account("caller", 0, SEED);188 <T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());189 let metadata = create_data();190 let max = None;191 let symbol = create_data();192 }: _(RawOrigin::Signed(caller), metadata, max, symbol)193194 destroy_collection {195 let caller = account("caller", 0, SEED);196197 create_max_collection::<T>(&caller)?;198 let collection_id = 0;199 }: _(RawOrigin::Signed(caller), collection_id)200201 change_collection_issuer {202 let caller: T::AccountId = account("caller", 0, SEED);203204 create_max_collection::<T>(&caller)?;205 let collection_id = 0;206207 let new_owner: T::AccountId = account("new_owner", 0, SEED);208209 let new_owner_source = T::Lookup::unlookup(new_owner);210 }: _(RawOrigin::Signed(caller), collection_id, new_owner_source)211212 lock_collection {213 let caller: T::AccountId = account("caller", 0, SEED);214215 create_max_collection::<T>(&caller)?;216 let collection_id = 0;217 }: _(RawOrigin::Signed(caller), collection_id)218219 mint_nft {220 let b in 0..100;221222 let caller: T::AccountId = account("caller", 0, SEED);223 <T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());224225 create_max_collection::<T>(&caller)?;226 let collection_id = 0;227 let owner = caller.clone();228229 let royalty_recipient = Some(caller.clone());230 let royalty_amount = Some(Permill::from_percent(25));231 let metadata = create_data();232 let transferable = true;233 }: _(234 RawOrigin::Signed(caller),235 owner,236 collection_id,237 royalty_recipient,238 royalty_amount,239 metadata,240 transferable,241 Some(create_max_resource_types_array(b as usize))242 )243244 burn_nft {245 let b in 0..200;246247 let caller: T::AccountId = account("caller", 0, SEED);248 create_max_collection::<T>(&caller)?;249 let collection_id = 0;250251 let mut nft_builder = NftBuilder::new(collection_id);252 let root_nft_id = nft_builder.build_wide_tree::<T>(&caller, b)?;253 let max_burns = b + 1;254 }: _(255 RawOrigin::Signed(caller),256 collection_id,257 root_nft_id,258 max_burns259 )260261 send {262 let caller: T::AccountId = account("caller", 0, SEED);263 create_max_collection::<T>(&caller)?;264 let collection_id = 0;265266 let mut nft_builder = NftBuilder::new(collection_id);267 let src_nft_id = nft_builder.build::<T>(&caller)?;268 let (_, target_nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 2)?;269 }: _(270 RawOrigin::Signed(caller),271 collection_id,272 src_nft_id,273 <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(collection_id, target_nft_id)274 )275276 accept_nft {277 let caller: T::AccountId = account("caller", 0, SEED);278 let sender: T::AccountId = account("sender", 0, SEED);279280 create_max_collection::<T>(&sender)?;281 let src_collection_id = 0;282283 create_max_collection::<T>(&caller)?;284 let target_collection_id = 1;285286 let mut src_nft_builder = NftBuilder::new(src_collection_id);287 let src_nft_id = src_nft_builder.build::<T>(&sender)?;288289 let mut target_nft_builder = NftBuilder::new(target_collection_id);290 let fake_target_nft_id = target_nft_builder.build::<T>(&caller)?;291 let (_, target_nft_id) = target_nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;292293 let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(294 target_collection_id,295 fake_target_nft_id296 );297298 let actual_new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(299 target_collection_id,300 target_nft_id301 );302303 <Pallet<T>>::send(304 RawOrigin::Signed(sender.clone()).into(),305 src_collection_id,306 src_nft_id,307 new_owner,308 )?;309 }: _(310 RawOrigin::Signed(caller),311 src_collection_id,312 src_nft_id,313 actual_new_owner314 )315316 reject_nft {317 let caller: T::AccountId = account("caller", 0, SEED);318 let sender: T::AccountId = account("sender", 0, SEED);319320 create_max_collection::<T>(&sender)?;321 let src_collection_id = 0;322323 create_max_collection::<T>(&caller)?;324 let target_collection_id = 1;325326 let mut src_nft_builder = NftBuilder::new(src_collection_id);327 let (src_root_nft_id, _) = src_nft_builder.build_tower::<T>(&sender, NESTING_BUDGET - 1)?;328329 let mut target_nft_builder = NftBuilder::new(target_collection_id);330 let target_nft_id = target_nft_builder.build::<T>(&caller)?;331332 let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(333 target_collection_id,334 target_nft_id335 );336337 <Pallet<T>>::send(338 RawOrigin::Signed(sender.clone()).into(),339 src_collection_id,340 src_root_nft_id,341 new_owner,342 )?;343 }: _(344 RawOrigin::Signed(caller),345 src_collection_id,346 src_root_nft_id347 )348349 set_property {350 let caller: T::AccountId = account("caller", 0, SEED);351 create_max_collection::<T>(&caller)?;352 let collection_id = 0;353354 let mut nft_builder = NftBuilder::new(collection_id);355 let (_, nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;356357 let key = create_data();358 let value = create_data();359 }: _(360 RawOrigin::Signed(caller),361 collection_id,362 Some(nft_id),363 key,364 value365 )366367 set_priority {368 let caller: T::AccountId = account("caller", 0, SEED);369 create_max_collection::<T>(&caller)?;370 let collection_id = 0;371372 let mut nft_builder = NftBuilder::new(collection_id);373 let (_, nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;374 let priorities = create_u32_array();375 }: _(376 RawOrigin::Signed(caller),377 collection_id,378 nft_id,379 priorities380 )381382 add_basic_resource {383 let caller: T::AccountId = account("caller", 0, SEED);384 <T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());385386 create_max_collection::<T>(&caller)?;387 let collection_id = 0;388389 let mut nft_builder = NftBuilder::new(collection_id);390 let (_, nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;391 let resource = create_basic_resource();392 }: _(393 RawOrigin::Signed(caller),394 collection_id,395 nft_id,396 resource397 )398399 add_composable_resource {400 let caller: T::AccountId = account("caller", 0, SEED);401 <T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());402403 create_max_collection::<T>(&caller)?;404 let collection_id = 0;405406 let mut nft_builder = NftBuilder::new(collection_id);407 let (_, nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;408 let resource = create_composable_resource();409 }: _(410 RawOrigin::Signed(caller),411 collection_id,412 nft_id,413 resource414 )415416 add_slot_resource {417 let caller: T::AccountId = account("caller", 0, SEED);418 <T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());419420 create_max_collection::<T>(&caller)?;421 let collection_id = 0;422423 let mut nft_builder = NftBuilder::new(collection_id);424 let (_, nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;425 let resource = create_slot_resource();426 }: _(427 RawOrigin::Signed(caller),428 collection_id,429 nft_id,430 resource431 )432433 remove_resource {434 let caller: T::AccountId = account("caller", 0, SEED);435 <T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());436437 create_max_collection::<T>(&caller)?;438 let collection_id = 0;439440 let mut nft_builder = NftBuilder::new(collection_id);441 let (_, nft_id) = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;442 let resource = create_basic_resource();443444 <Pallet<T>>::add_basic_resource(445 RawOrigin::Signed(caller.clone()).into(),446 collection_id,447 nft_id,448 resource449 )?;450451 let resource_id = 1;452 }: _(453 RawOrigin::Signed(caller),454 collection_id,455 nft_id,456 resource_id457 )458459 accept_resource {460 let caller: T::AccountId = account("caller", 0, SEED);461 let admin: T::AccountId = account("admin", 0, SEED);462463 <T as pallet_common::Config>::Currency::deposit_creating(&admin, T::CollectionCreationPrice::get());464465 create_max_collection::<T>(&admin)?;466 let collection_id = 0;467468 let mut nft_builder = NftBuilder::new(collection_id);469 let root_nft_id = 1;470 let (_, nested_nft_id) = nft_builder.build_tower::<T>(&admin, NESTING_BUDGET - 1)?;471 let resource = create_basic_resource();472473 let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::AccountId(caller.clone());474475 <Pallet<T>>::send(476 RawOrigin::Signed(admin.clone()).into(),477 collection_id,478 root_nft_id,479 new_owner,480 )?;481482 <Pallet<T>>::add_basic_resource(483 RawOrigin::Signed(admin.clone()).into(),484 collection_id,485 nested_nft_id,486 resource487 )?;488489 let resource_id = 1;490 }: _(491 RawOrigin::Signed(caller),492 collection_id,493 nested_nft_id,494 resource_id495 )496497 accept_resource_removal {498 let caller: T::AccountId = account("caller", 0, SEED);499 let admin: T::AccountId = account("admin", 0, SEED);500501 <T as pallet_common::Config>::Currency::deposit_creating(&admin, T::CollectionCreationPrice::get());502503 create_max_collection::<T>(&admin)?;504 let collection_id = 0;505506 let mut nft_builder = NftBuilder::new(collection_id);507 let root_nft_id = 1;508 let (_, nested_nft_id) = nft_builder.build_tower::<T>(&admin, NESTING_BUDGET - 1)?;509 let resource = create_basic_resource();510511 <Pallet<T>>::add_basic_resource(512 RawOrigin::Signed(admin.clone()).into(),513 collection_id,514 nested_nft_id,515 resource516 )?;517518 let resource_id = 1;519520 let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::AccountId(caller.clone());521522 <Pallet<T>>::send(523 RawOrigin::Signed(admin.clone()).into(),524 collection_id,525 root_nft_id,526 new_owner,527 )?;528529 <Pallet<T>>::remove_resource(530 RawOrigin::Signed(admin).into(),531 collection_id,532 nested_nft_id,533 resource_id534 )?;535 }: _(536 RawOrigin::Signed(caller),537 collection_id,538 nested_nft_id,539 resource_id540 )541}