git.delta.rocks / unique-network / refs/commits / ccbb0c2b6b71

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}