Merge pull request #229 from hashed-io/fruniques/update_tests

Update tests

Author: tlacloc

pallets/fruniques/README.md

@@ -2,7 +2,7 @@
 ### **FR**actional **Uniques**
 > This is WIP - just being spec'd out
 
-This pallet is being developed **tightly coupled** to both [`pallet_assets`](https://paritytech.github.io/substrate/latest/pallet_assets/)  and [`pallet_uniques`](https://paritytech.github.io/substrate/latest/pallet_uniques/index.html). These are the default [Statemine](https://github.com/paritytech/cumulus/tree/master/polkadot-parachains/statemine) pallets for fungible and non-fungible tokens. 
+This pallet is being developed **tightly coupled** to both [`pallet_assets`](https://paritytech.github.io/substrate/latest/pallet_assets/)  and [`pallet_uniques`](https://paritytech.github.io/substrate/latest/pallet_uniques/index.html). These are the default [Statemine](https://github.com/paritytech/cumulus/tree/master/polkadot-parachains/statemine) pallets for fungible and non-fungible tokens.
 
 A Frunique is a type of Non-Fungible Token (NFT)
 
@@ -19,20 +19,20 @@ This pallet provides functionality that allows NFT holders to mint a fungible to
 
 The non-fungible token is created and minted using the Statemine `pallet_uniques`.
 
-The fungible token is created and minted using the Statemine `pallet_assets`. 
+The fungible token is created and minted using the Statemine `pallet_assets`.
 
 The NFT/Unique can be unlocked and released if and only if a single origin holds all of the corresponding fungible token.
 
 ![basket-of-fungibles](http://www.plantuml.com/plantuml/proxy?cache=no&src=https://raw.githubusercontent.com/hashed-io/hashed-substrate/main/docs/fungible-basket-frunique.iuml)
 
 ## NFT <--> NFTs
-An NFT as a trie of NFTs. 
+An NFT as a trie of NFTs.
 
 ### Use cases
 #### Tax credit marketplace
 A credit is a single NFT, with an `amount`, state of redemption, expiration year, and other metadata. However, that owner can sell less than the `amount`, in which case the newly created credit NFT has all of the same associated data.  The sum of the children `amount` values must be equal to the parent.
 
-To support this, we'll create a `AggregatedFrunique` type that enforces the aggregation rules. 
+To support this, we'll create a `AggregatedFrunique` type that enforces the aggregation rules.
 
 #### Cannabis compliance
 For the NY state cannabis compliance program, all yield from all plants must be tracked. This aligns to a very similar data structure as above. Each mother plant is an NFT, each clone as an NFT, each package of flower an NFT, etc. Auditing a specific item is fairly easy via traversing all of its ancestors and descendants through to the harvest and dispensary.
@@ -58,7 +58,7 @@ yarn install
 ```bash
 yarn run:api tx.fruniques.create 1 1 1 5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY 1 100 --seed "//Alice"
 ```
-## Check NFT 
+## Check NFT
 ```bash
 yarn run:api query.uniques.class 1
 yarn run:api query.uniques.asset 1 1
@@ -71,9 +71,16 @@ yarn run:api query.assets.asset 1
 yarn run:api query.assets.account 1 5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY
 ```
 
+## Tests of the pallet
+
+To run the test you simply run the following command
+```bash
+cargo test --package pallet-fruniques
+```
+
 # Similar Projects
 ### [Fractional Art](https://fractional.art/)
- 
+
 ### [Detailed Process Explainer with screenshots](https://medium.com/fractional-art/how-to-use-fractional-to-fractionalize-nfts-84da1a465b6d)
 
 License: MIT

pallets/fruniques/src/functions.rs

@@ -30,6 +30,10 @@ impl<T: Config> Pallet<T> {
 		u32::from_ne_bytes(input.try_into().unwrap())
 	}
 
+	pub fn percent_to_permill(input: u8) -> Permill {
+		Permill::from_percent(input as u32)
+	}
+
 	pub fn bytes_to_string(input: Vec<u8>) -> String {
 		let mut s = String::default();
 		for x in input {
@@ -196,7 +200,6 @@ impl<T: Config> Pallet<T> {
 		collection: T::CollectionId,
 		item: T::ItemId,
 		owner: <T::Lookup as sp_runtime::traits::StaticLookup>::Source,
-		_numeric_value: Option<Permill>,
 		attributes: Option<Vec<(BoundedVec<u8, T::KeyLimit>, BoundedVec<u8, T::ValueLimit>)>>,
 	) -> DispatchResult {
 

pallets/fruniques/src/lib.rs

@@ -21,7 +21,6 @@ pub mod pallet {
 	use frame_support::{pallet_prelude::*, transactional, BoundedVec};
 	use frame_system::pallet_prelude::*;
 	use scale_info::prelude::vec::Vec;
-	use sp_runtime::Permill;
 	/// Configure the pallet by specifying the parameters and types on which it depends.
 	#[pallet::config]
 	pub trait Config: frame_system::Config + pallet_uniques::Config {
@@ -41,22 +40,25 @@ pub mod pallet {
 	#[pallet::event]
 	#[pallet::generate_deposit(pub(super) fn deposit_event)]
 	pub enum Event<T: Config> {
-		// A frunique and asset class were succesfully created!
+		// A frunique and asset class were successfully created!
 		FruniqueCollectionCreated(T::AccountId, T::CollectionId),
-		// A frunique and asset class were succesfully created!
+		// A frunique and asset class were successfully created!
 		FruniqueCreated(T::AccountId, T::AccountId, T::CollectionId, T::ItemId),
-		// A frunique/unique was succesfully divided!
+		// A frunique/unique was successfully divided!
 		FruniqueDivided(T::AccountId, T::AccountId, T::CollectionId, T::ItemId),
 		// Counter should work?
 		NextFrunique(u32),
 	}
 
 	#[pallet::error]
 	pub enum Error<T> {
-		NoneValue,
+		// The user does not have permission to perform this action
 		NoPermission,
+		// Only the owner of the Frunique can perform this action
 		NotAdmin,
+		// The storage is full
 		StorageOverflow,
+		// A feature not implemented yet
 		NotYetImplemented,
 		// Too many fruniques were minted
 		FruniqueCntOverflow,
@@ -70,6 +72,10 @@ pub mod pallet {
 		AttributesEmpty,
 		// The collection doesn't exist
 		CollectionNotFound,
+		// The parent doesn't exist
+		ParentNotFound,
+		// The frunique doesn't exist
+		FruniqueNotFound,
 	}
 
 	#[pallet::storage]
@@ -118,7 +124,7 @@ pub mod pallet {
 		Blake2_128Concat,
 		CollectionId, // Parent collection id
 		Blake2_128Concat,
-		ItemId, // Parent item id
+		ItemId,            // Parent item id
 		Option<ChildInfo>, // ParentId and flag if it inherit attributes
 		ValueQuery,
 	>;
@@ -152,7 +158,6 @@ pub mod pallet {
 				Self::account_id_to_lookup_source(&admin),
 			)?;
 
-
 			Self::deposit_event(Event::FruniqueCollectionCreated(admin, new_collection_id));
 
 			<NextCollection<T>>::put(Self::next_collection() + 1);
@@ -178,12 +183,12 @@ pub mod pallet {
 
 		/// ## Set multiple attributes to a frunique.
 		/// `origin` must be signed by the owner of the frunique.
+		/// - `class_id` must be a valid class of the asset class.
+		/// - `instance_id` must be a valid instance of the asset class.
 		/// - `attributes` must be a list of pairs of `key` and `value`.
 		/// `key` must be a valid key for the asset class.
 		/// `value` must be a valid value for the asset class.
 		/// `attributes` must not be empty.
-		/// - `instance_id` must be a valid instance of the asset class.
-		/// - `class_id` must be a valid class of the asset class.
 
 		#[pallet::weight(10_000 + T::DbWeight::get().writes(1))]
 		pub fn set_attributes(
@@ -192,6 +197,8 @@ pub mod pallet {
 			instance_id: T::ItemId,
 			attributes: Vec<(BoundedVec<u8, T::KeyLimit>, BoundedVec<u8, T::ValueLimit>)>,
 		) -> DispatchResult {
+			ensure!(Self::item_exists(&class_id, &instance_id), <Error<T>>::FruniqueNotFound);
+
 			// ! Ensure the admin is the one who can add attributes to the frunique.
 			let admin = Self::admin_of(&class_id, &instance_id);
 			let signer = core::prelude::v1::Some(ensure_signed(origin.clone())?);
@@ -224,59 +231,51 @@ pub mod pallet {
 		///
 		/// ### Parameters needed in order to divide a unique:
 		/// - `class_id`: The type of NFT that the function will create, categorized by numbers.
-		/// - `numeric_value`: The value of the NFT that the function will create.
 		/// - `parent_info`: Information of the parent NFT and a flag to indicate the child would inherit their attributes.
 		/// - `attributes`: Generates a list of attributes for the new NFT.
 		///
 		#[pallet::weight(10_000 + T::DbWeight::get().writes(4))]
 		pub fn spawn(
 			origin: OriginFor<T>,
 			class_id: CollectionId,
-			numeric_value: Option<Permill>,
 			parent_info: Option<HierarchicalInfo>,
 			attributes: Option<Vec<(BoundedVec<u8, T::KeyLimit>, BoundedVec<u8, T::ValueLimit>)>>,
 		) -> DispatchResult {
+			ensure!(Self::collection_exists(&class_id), <Error<T>>::CollectionNotFound);
+
+			if let Some(parent_info) = parent_info {
+				ensure!(Self::item_exists(&class_id, &parent_info.0), <Error<T>>::ParentNotFound);
+			}
+
 			let owner: T::AccountId = ensure_signed(origin.clone())?;
 			let account_id = Self::account_id_to_lookup_source(&owner);
 
 			let instance_id: ItemId = <NextFrunique<T>>::try_get(class_id).unwrap_or(0);
 			<NextFrunique<T>>::insert(class_id, instance_id + 1);
 
-			Self::do_spawn(
-				origin.clone(),
-				class_id,
-				instance_id,
-				account_id,
-				numeric_value,
-				attributes,
-			)?;
-
-			Self::deposit_event(Event::FruniqueCreated(
-				owner.clone(),
-				owner,
-				class_id,
-				instance_id,
-			));
-
-
 			if let Some(parent_info) = parent_info {
-				ensure!(
-					Self::item_exists(&class_id, &parent_info.0),
-					<Error<T>>::CollectionNotFound
-				);
+				ensure!(Self::item_exists(&class_id, &parent_info.0), <Error<T>>::ParentNotFound);
 				<FruniqueParent<T>>::insert(class_id, instance_id, Some(parent_info));
 
 				let child_info = ChildInfo {
 					collection_id: class_id,
 					child_id: instance_id,
 					is_hierarchical: parent_info.1,
-					weight: numeric_value.unwrap()
+					weight: Self::percent_to_permill(parent_info.2),
 				};
 
 				<FruniqueChild<T>>::insert(class_id, instance_id, Some(child_info));
-
 			}
 
+			Self::do_spawn(origin.clone(), class_id, instance_id, account_id, attributes)?;
+
+			Self::deposit_event(Event::FruniqueCreated(
+				owner.clone(),
+				owner,
+				class_id,
+				instance_id,
+			));
+
 			Ok(())
 		}