git.delta.rocks / unique-network / refs/commits / 4f0a4460dc55

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_collection<T: Config>(owner: &T::AccountId) -> DispatchResult {57	<T as pallet_common::Config>::Currency::deposit_creating(58		owner,59		T::CollectionCreationPrice::get(),60	);6162	let metadata = create_data();63	let max = None;64	let symbol = create_data();6566	<Pallet<T>>::create_collection(67		RawOrigin::Signed(owner.clone()).into(),68		metadata,69		max,70		symbol,71	)72}7374fn create_max_nft<T: Config>(75	owner: &T::AccountId,76	collection_id: RmrkCollectionId,77) -> DispatchResult {78	let royalty_recipient = Some(owner.clone());79	let royalty_amount = Some(Permill::from_percent(25));80	let metadata = create_data();81	let transferable = true;8283	<Pallet<T>>::mint_nft(84		RawOrigin::Signed(owner.clone()).into(),85		owner.clone(),86		collection_id,87		royalty_recipient,88		royalty_amount,89		metadata,90		transferable,91	)92}9394struct NftBuilder {95	collection_id: RmrkCollectionId,96	current_nft_id: RmrkNftId,97}9899impl NftBuilder {100	fn new(collection_id: RmrkCollectionId) -> Self {101		Self {102			collection_id,103			current_nft_id: 0,104		}105	}106107	fn build<T: Config>(&mut self, owner: &T::AccountId) -> Result<RmrkNftId, DispatchError> {108		create_max_nft::<T>(owner, self.collection_id)?;109		self.current_nft_id += 1;110111		Ok(self.current_nft_id)112	}113114	fn build_tower<T: Config>(115		&mut self,116		owner: &T::AccountId,117		height: u32,118	) -> Result<RmrkNftId, DispatchError> {119		self.build::<T>(owner)?;120121		let mut prev_nft_id = self.current_nft_id;122123		for _ in 0..height {124			self.build::<T>(owner)?;125126			let new_owner =127				<RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(128					self.collection_id,129					prev_nft_id,130				);131132			<Pallet<T>>::send(133				RawOrigin::Signed(owner.clone()).into(),134				self.collection_id,135				self.current_nft_id,136				new_owner,137			)?;138139			prev_nft_id = self.current_nft_id;140		}141142		let deepest_nft_id = self.current_nft_id;143144		Ok(deepest_nft_id)145	}146}147148benchmarks! {149	create_collection {150		let caller = account("caller", 0, SEED);151		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());152		let metadata = create_data();153		let max = None;154		let symbol = create_data();155	}: _(RawOrigin::Signed(caller), metadata, max, symbol)156157	destroy_collection {158		let caller = account("caller", 0, SEED);159160		create_max_collection::<T>(&caller)?;161		let collection_id = 0;162	}: _(RawOrigin::Signed(caller), collection_id)163164	change_collection_issuer {165		let caller: T::AccountId = account("caller", 0, SEED);166167		create_max_collection::<T>(&caller)?;168		let collection_id = 0;169170		let new_owner: T::AccountId = account("new_owner", 0, SEED);171172		let new_owner_source = T::Lookup::unlookup(new_owner);173	}: _(RawOrigin::Signed(caller), collection_id, new_owner_source)174175	lock_collection {176		let caller: T::AccountId = account("caller", 0, SEED);177178		create_max_collection::<T>(&caller)?;179		let collection_id = 0;180	}: _(RawOrigin::Signed(caller), collection_id)181182	mint_nft {183		let caller: T::AccountId = account("caller", 0, SEED);184185		create_max_collection::<T>(&caller)?;186		let collection_id = 0;187		let owner = caller.clone();188189		let royalty_recipient = Some(caller.clone());190		let royalty_amount = Some(Permill::from_percent(25));191		let metadata = create_data();192		let transferable = true;193	}:  _(194		RawOrigin::Signed(caller),195		owner,196		collection_id,197		royalty_recipient,198		royalty_amount,199		metadata,200		transferable201	)202203	send {204		let caller: T::AccountId = account("caller", 0, SEED);205		create_max_collection::<T>(&caller)?;206		let collection_id = 0;207208		let mut nft_builder = NftBuilder::new(collection_id);209		let src_nft_id = nft_builder.build::<T>(&caller)?;210		let target_nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 2)?;211	}: _(212		RawOrigin::Signed(caller),213		collection_id,214		src_nft_id,215		<RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(collection_id, target_nft_id)216	)217218	accept_nft {219		let caller: T::AccountId = account("caller", 0, SEED);220		let sender: T::AccountId = account("sender", 0, SEED);221222		create_max_collection::<T>(&sender)?;223		let src_collection_id = 0;224225		create_max_collection::<T>(&caller)?;226		let target_collection_id = 1;227228		let mut src_nft_builder = NftBuilder::new(src_collection_id);229		let src_nft_id = src_nft_builder.build::<T>(&sender)?;230231		let mut target_nft_builder = NftBuilder::new(target_collection_id);232		let fake_target_nft_id = target_nft_builder.build::<T>(&caller)?;233		let target_nft_id = target_nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;234235		let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(236			target_collection_id,237			fake_target_nft_id238		);239240		let actual_new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(241			target_collection_id,242			target_nft_id243		);244245		<Pallet<T>>::send(246			RawOrigin::Signed(sender.clone()).into(),247			src_collection_id,248			src_nft_id,249			new_owner,250		)?;251	}: _(252		RawOrigin::Signed(caller),253		src_collection_id,254		src_nft_id,255		actual_new_owner256	)257258	set_property {259		let caller: T::AccountId = account("caller", 0, SEED);260		create_max_collection::<T>(&caller)?;261		let collection_id = 0;262263		let mut nft_builder = NftBuilder::new(collection_id);264		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;265266		let key = create_data();267		let value = create_data();268	}: _(269		RawOrigin::Signed(caller),270		collection_id,271		Some(nft_id),272		key,273		value274	)275276	set_priority {277		let caller: T::AccountId = account("caller", 0, SEED);278		create_max_collection::<T>(&caller)?;279		let collection_id = 0;280281		let mut nft_builder = NftBuilder::new(collection_id);282		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;283		let priorities = create_u32_array();284	}: _(285		RawOrigin::Signed(caller),286		collection_id,287		nft_id,288		priorities289	)290291	add_basic_resource {292		let caller: T::AccountId = account("caller", 0, SEED);293		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());294295		create_max_collection::<T>(&caller)?;296		let collection_id = 0;297298		let mut nft_builder = NftBuilder::new(collection_id);299		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;300		let resource = create_basic_resource();301	}: _(302		RawOrigin::Signed(caller),303		collection_id,304		nft_id,305		resource306	)307308	add_composable_resource {309		let caller: T::AccountId = account("caller", 0, SEED);310		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());311312		create_max_collection::<T>(&caller)?;313		let collection_id = 0;314315		let mut nft_builder = NftBuilder::new(collection_id);316		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;317		let resource = create_composable_resource();318	}: _(319		RawOrigin::Signed(caller),320		collection_id,321		nft_id,322		resource323	)324325	add_slot_resource {326		let caller: T::AccountId = account("caller", 0, SEED);327		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());328329		create_max_collection::<T>(&caller)?;330		let collection_id = 0;331332		let mut nft_builder = NftBuilder::new(collection_id);333		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;334		let resource = create_slot_resource();335	}: _(336		RawOrigin::Signed(caller),337		collection_id,338		nft_id,339		resource340	)341342	remove_resource {343		let caller: T::AccountId = account("caller", 0, SEED);344		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());345346		create_max_collection::<T>(&caller)?;347		let collection_id = 0;348349		let mut nft_builder = NftBuilder::new(collection_id);350		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;351		let resource = create_basic_resource();352353		<Pallet<T>>::add_basic_resource(354			RawOrigin::Signed(caller.clone()).into(),355			collection_id,356			nft_id,357			resource358		)?;359360		let resource_id = 1;361	}: _(362		RawOrigin::Signed(caller),363		collection_id,364		nft_id,365		resource_id366	)367368	accept_resource {369		let caller: T::AccountId = account("caller", 0, SEED);370		let admin: T::AccountId = account("admin", 0, SEED);371372		<T as pallet_common::Config>::Currency::deposit_creating(&admin, T::CollectionCreationPrice::get());373374		create_max_collection::<T>(&admin)?;375		let collection_id = 0;376377		let mut nft_builder = NftBuilder::new(collection_id);378		let root_nft_id = 1;379		let nested_nft_id = nft_builder.build_tower::<T>(&admin, NESTING_BUDGET - 1)?;380		let resource = create_basic_resource();381382		let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::AccountId(caller.clone());383384		<Pallet<T>>::send(385			RawOrigin::Signed(admin.clone()).into(),386			collection_id,387			root_nft_id,388			new_owner,389		)?;390391		<Pallet<T>>::add_basic_resource(392			RawOrigin::Signed(admin.clone()).into(),393			collection_id,394			nested_nft_id,395			resource396		)?;397398		let resource_id = 1;399	}: _(400		RawOrigin::Signed(caller),401		collection_id,402		nested_nft_id,403		resource_id404	)405406	accept_resource_removal {407		let caller: T::AccountId = account("caller", 0, SEED);408		let admin: T::AccountId = account("admin", 0, SEED);409410		<T as pallet_common::Config>::Currency::deposit_creating(&admin, T::CollectionCreationPrice::get());411412		create_max_collection::<T>(&admin)?;413		let collection_id = 0;414415		let mut nft_builder = NftBuilder::new(collection_id);416		let root_nft_id = 1;417		let nested_nft_id = nft_builder.build_tower::<T>(&admin, NESTING_BUDGET - 1)?;418		let resource = create_basic_resource();419420		<Pallet<T>>::add_basic_resource(421			RawOrigin::Signed(admin.clone()).into(),422			collection_id,423			nested_nft_id,424			resource425		)?;426427		let resource_id = 1;428429		let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::AccountId(caller.clone());430431		<Pallet<T>>::send(432			RawOrigin::Signed(admin.clone()).into(),433			collection_id,434			root_nft_id,435			new_owner,436		)?;437438		<Pallet<T>>::remove_resource(439			RawOrigin::Signed(admin).into(),440			collection_id,441			nested_nft_id,442			resource_id443		)?;444	}: _(445		RawOrigin::Signed(caller),446		collection_id,447		nested_nft_id,448		resource_id449	)450}