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(owner, T::CollectionCreationPrice::get());5859	let metadata = create_data();60	let max = None;61	let symbol = create_data();6263	<Pallet<T>>::create_collection(RawOrigin::Signed(owner.clone()).into(), metadata, max, symbol)64}6566fn create_max_nft<T: Config>(owner: &T::AccountId, collection_id: RmrkCollectionId) -> DispatchResult {67	let royalty_recipient = Some(owner.clone());68	let royalty_amount = Some(Permill::from_percent(25));69	let metadata = create_data();70	let transferable = true;7172	<Pallet<T>>::mint_nft(73		RawOrigin::Signed(owner.clone()).into(),74		owner.clone(),75		collection_id,76		royalty_recipient,77		royalty_amount,78		metadata,79		transferable80	)81}8283struct NftBuilder {84	collection_id: RmrkCollectionId,85	current_nft_id: RmrkNftId86}8788impl NftBuilder {89	fn new(collection_id: RmrkCollectionId) -> Self {90		Self {91			collection_id,92			current_nft_id: 093		}94	}9596	fn build<T: Config>(&mut self, owner: &T::AccountId) -> Result<RmrkNftId, DispatchError> {97		create_max_nft::<T>(owner, self.collection_id)?;98		self.current_nft_id += 1;99100		Ok(self.current_nft_id)101	}102103	fn build_tower<T: Config>(&mut self, owner: &T::AccountId, height: u32) -> Result<RmrkNftId, DispatchError> {104		self.build::<T>(owner)?;105106		let mut prev_nft_id = self.current_nft_id;107108		for _ in 0..height {109			self.build::<T>(owner)?;110			111			let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(112				self.collection_id,113				prev_nft_id114			);115116			<Pallet<T>>::send(117				RawOrigin::Signed(owner.clone()).into(),118				self.collection_id,119				self.current_nft_id,120				new_owner,121			)?;122123			prev_nft_id = self.current_nft_id;124		}125126		let deepest_nft_id = self.current_nft_id;127128		Ok(deepest_nft_id)129	}130}131132benchmarks! {133	create_collection {134		let caller = account("caller", 0, SEED);135		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());136		let metadata = create_data();137		let max = None;138		let symbol = create_data();139	}: _(RawOrigin::Signed(caller), metadata, max, symbol)140141	destroy_collection {142		let caller = account("caller", 0, SEED);143		144		create_max_collection::<T>(&caller)?;145		let collection_id = 0;146	}: _(RawOrigin::Signed(caller), collection_id)147148	change_collection_issuer {149		let caller: T::AccountId = account("caller", 0, SEED);150151		create_max_collection::<T>(&caller)?;152		let collection_id = 0;153154		let new_owner: T::AccountId = account("new_owner", 0, SEED);155156		let new_owner_source = T::Lookup::unlookup(new_owner);157	}: _(RawOrigin::Signed(caller), collection_id, new_owner_source)158159	lock_collection {160		let caller: T::AccountId = account("caller", 0, SEED);161162		create_max_collection::<T>(&caller)?;163		let collection_id = 0;164	}: _(RawOrigin::Signed(caller), collection_id)165166	mint_nft {167		let caller: T::AccountId = account("caller", 0, SEED);168169		create_max_collection::<T>(&caller)?;170		let collection_id = 0;171		let owner = caller.clone();172		173		let royalty_recipient = Some(caller.clone());174		let royalty_amount = Some(Permill::from_percent(25));175		let metadata = create_data();176		let transferable = true;177	}:  _(178		RawOrigin::Signed(caller),179		owner,180		collection_id,181		royalty_recipient,182		royalty_amount,183		metadata,184		transferable185	)186187	send {188		let caller: T::AccountId = account("caller", 0, SEED);189		create_max_collection::<T>(&caller)?;190		let collection_id = 0;191192		let mut nft_builder = NftBuilder::new(collection_id);193		let src_nft_id = nft_builder.build::<T>(&caller)?;194		let target_nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 2)?;195	}: _(196		RawOrigin::Signed(caller),197		collection_id,198		src_nft_id,199		<RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(collection_id, target_nft_id)200	)201202	accept_nft {203		let caller: T::AccountId = account("caller", 0, SEED);204		let sender: T::AccountId = account("sender", 0, SEED);205206		create_max_collection::<T>(&sender)?;207		let src_collection_id = 0;208209		create_max_collection::<T>(&caller)?;210		let target_collection_id = 1;211212		let mut src_nft_builder = NftBuilder::new(src_collection_id);213		let src_nft_id = src_nft_builder.build::<T>(&sender)?;214215		let mut target_nft_builder = NftBuilder::new(target_collection_id);216		let fake_target_nft_id = target_nft_builder.build::<T>(&caller)?;217		let target_nft_id = target_nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;218219		let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(220			target_collection_id,221			fake_target_nft_id222		);223224		let actual_new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::CollectionAndNftTuple(225			target_collection_id,226			target_nft_id227		);228229		<Pallet<T>>::send(230			RawOrigin::Signed(sender.clone()).into(),231			src_collection_id,232			src_nft_id,233			new_owner,234		)?;235	}: _(236		RawOrigin::Signed(caller),237		src_collection_id,238		src_nft_id,239		actual_new_owner240	)241242	set_property {243		let caller: T::AccountId = account("caller", 0, SEED);244		create_max_collection::<T>(&caller)?;245		let collection_id = 0;246247		let mut nft_builder = NftBuilder::new(collection_id);248		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;249250		let key = create_data();251		let value = create_data();252	}: _(253		RawOrigin::Signed(caller),254		collection_id,255		Some(nft_id),256		key,257		value258	)259260	set_priority {261		let caller: T::AccountId = account("caller", 0, SEED);262		create_max_collection::<T>(&caller)?;263		let collection_id = 0;264265		let mut nft_builder = NftBuilder::new(collection_id);266		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;267		let priorities = create_u32_array();268	}: _(269		RawOrigin::Signed(caller),270		collection_id,271		nft_id,272		priorities273	)274275	add_basic_resource {276		let caller: T::AccountId = account("caller", 0, SEED);277		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());278279		create_max_collection::<T>(&caller)?;280		let collection_id = 0;281282		let mut nft_builder = NftBuilder::new(collection_id);283		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;284		let resource = create_basic_resource();285	}: _(286		RawOrigin::Signed(caller),287		collection_id,288		nft_id,289		resource290	)291292	add_composable_resource {293		let caller: T::AccountId = account("caller", 0, SEED);294		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());295296		create_max_collection::<T>(&caller)?;297		let collection_id = 0;298299		let mut nft_builder = NftBuilder::new(collection_id);300		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;301		let resource = create_composable_resource();302	}: _(303		RawOrigin::Signed(caller),304		collection_id,305		nft_id,306		resource307	)308309	add_slot_resource {310		let caller: T::AccountId = account("caller", 0, SEED);311		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());312313		create_max_collection::<T>(&caller)?;314		let collection_id = 0;315316		let mut nft_builder = NftBuilder::new(collection_id);317		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;318		let resource = create_slot_resource();319	}: _(320		RawOrigin::Signed(caller),321		collection_id,322		nft_id,323		resource324	)325326	remove_resource {327		let caller: T::AccountId = account("caller", 0, SEED);328		<T as pallet_common::Config>::Currency::deposit_creating(&caller, T::CollectionCreationPrice::get());329330		create_max_collection::<T>(&caller)?;331		let collection_id = 0;332333		let mut nft_builder = NftBuilder::new(collection_id);334		let nft_id = nft_builder.build_tower::<T>(&caller, NESTING_BUDGET - 1)?;335		let resource = create_basic_resource();336337		<Pallet<T>>::add_basic_resource(338			RawOrigin::Signed(caller.clone()).into(),339			collection_id,340			nft_id,341			resource342		)?;343344		let resource_id = 1;345	}: _(346		RawOrigin::Signed(caller),347		collection_id,348		nft_id,349		resource_id350	)351352	accept_resource {353		let caller: T::AccountId = account("caller", 0, SEED);354		let admin: T::AccountId = account("admin", 0, SEED);355356		<T as pallet_common::Config>::Currency::deposit_creating(&admin, T::CollectionCreationPrice::get());357358		create_max_collection::<T>(&admin)?;359		let collection_id = 0;360361		let mut nft_builder = NftBuilder::new(collection_id);362		let root_nft_id = 1;363		let nested_nft_id = nft_builder.build_tower::<T>(&admin, NESTING_BUDGET - 1)?;364		let resource = create_basic_resource();365366		let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::AccountId(caller.clone());367368		<Pallet<T>>::send(369			RawOrigin::Signed(admin.clone()).into(),370			collection_id,371			root_nft_id,372			new_owner,373		)?;374375		<Pallet<T>>::add_basic_resource(376			RawOrigin::Signed(admin.clone()).into(),377			collection_id,378			nested_nft_id,379			resource380		)?;381382		let resource_id = 1;383	}: _(384		RawOrigin::Signed(caller),385		collection_id,386		nested_nft_id,387		resource_id388	)389390	accept_resource_removal {391		let caller: T::AccountId = account("caller", 0, SEED);392		let admin: T::AccountId = account("admin", 0, SEED);393394		<T as pallet_common::Config>::Currency::deposit_creating(&admin, T::CollectionCreationPrice::get());395396		create_max_collection::<T>(&admin)?;397		let collection_id = 0;398399		let mut nft_builder = NftBuilder::new(collection_id);400		let root_nft_id = 1;401		let nested_nft_id = nft_builder.build_tower::<T>(&admin, NESTING_BUDGET - 1)?;402		let resource = create_basic_resource();403404		<Pallet<T>>::add_basic_resource(405			RawOrigin::Signed(admin.clone()).into(),406			collection_id,407			nested_nft_id,408			resource409		)?;410411		let resource_id = 1;412413		let new_owner = <RmrkAccountIdOrCollectionNftTuple<T::AccountId>>::AccountId(caller.clone());414415		<Pallet<T>>::send(416			RawOrigin::Signed(admin.clone()).into(),417			collection_id,418			root_nft_id,419			new_owner,420		)?;421422		<Pallet<T>>::remove_resource(423			RawOrigin::Signed(admin).into(),424			collection_id,425			nested_nft_id,426			resource_id427		)?;428	}: _(429		RawOrigin::Signed(caller),430		collection_id,431		nested_nft_id,432		resource_id433	)434}