difftreelog
Merge branch 'develop' into feature/NFTPAR-142
in: master
# Conflicts: # pallets/nft/src/lib.rs
5 files changed
pallets/nft/src/benchmarking.rsdiffbeforeafterboth--- a/pallets/nft/src/benchmarking.rs
+++ b/pallets/nft/src/benchmarking.rs
@@ -165,7 +165,7 @@
let token_prefix1: Vec<u8> = b"token_prefix1".to_vec();
let mode: CollectionMode = CollectionMode::NFT;
let caller: T::AccountId = T::AccountId::from(whitelisted_caller());
- let nft_data = CreateNftData {
+ let mut nft_data = CreateNftData {
const_data: vec![],
variable_data: vec![]
};
@@ -173,7 +173,7 @@
nft_data.const_data.push(10);
nft_data.variable_data.push(10);
}
- let mut data = CreateItemData::NFT(nft_data);
+ let data = CreateItemData::NFT(nft_data);
Nft::<T>::create_collection(RawOrigin::Signed(caller.clone()).into(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), mode.clone())?;
}: create_item(RawOrigin::Signed(caller.clone()), 2, caller.clone(), data)
pallets/nft/src/default_weights.rsdiffbeforeafterboth--- a/pallets/nft/src/default_weights.rs
+++ b/pallets/nft/src/default_weights.rs
@@ -63,7 +63,7 @@
}
fn create_item(s: usize, ) -> Weight {
(130_000_000 as Weight)
- .saturating_add((2135 as Weight).saturating_mul(s as Weight))
+ .saturating_add((2135 as Weight).saturating_mul(s as Weight).saturating_mul(500 as Weight)) // 500 is temporary multiplier, fee for storage
.saturating_add(DbWeight::get().reads(10 as Weight))
.saturating_add(DbWeight::get().writes(8 as Weight))
}
pallets/nft/src/lib.rsdiffbeforeafterboth--- a/pallets/nft/src/lib.rs
+++ b/pallets/nft/src/lib.rs
@@ -8,9 +8,9 @@
use codec::{Decode, Encode};
pub use frame_support::{
- construct_runtime, decl_event, decl_module, decl_storage,
+ construct_runtime, decl_event, decl_module, decl_storage, decl_error,
dispatch::DispatchResult,
- ensure, parameter_types, fail,
+ ensure, fail, parameter_types,
traits::{
Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,
Randomness, WithdrawReason,
@@ -22,7 +22,6 @@
},
IsSubType, StorageValue,
};
-// use frame_support::weights::{Weight, constants::RocksDbWeight as DbWeight};
use frame_system::{self as system, ensure_signed, ensure_root};
use sp_runtime::sp_std::prelude::Vec;
@@ -116,13 +115,6 @@
#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]
#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]
-pub struct CollectionAdminsType<AccountId> {
- pub admin: AccountId,
- pub collection_id: u64,
-}
-
-#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]
-#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]
pub struct NftItemType<AccountId> {
pub collection: u64,
pub owner: AccountId,
@@ -268,6 +260,59 @@
}
}
+
+decl_error! {
+ /// Error for non-fungible-token module.
+ pub enum Error for Module<T: Trait> {
+ /// Total collections bound exceeded
+ TotalCollectionsLimitExceeded,
+ /// Decimal_points parameter must be lower than 4
+ CollectionDecimalPointLimitExceeded,
+ /// Collection name can not be longer than 63 char
+ CollectionNameLimitExceeded,
+ /// Collection description can not be longer than 255 char
+ CollectionDescriptionLimitExceeded,
+ /// Token prefix can not be longer than 15 char
+ CollectionTokenPrefixLimitExceeded,
+ /// This collection does not exist
+ CollectionNotFound,
+ /// Item not exists
+ TokenNotFound,
+ /// Arithmetic calculation overflow
+ NumOverflow,
+ /// Account already has admin role
+ AlreadyAdmin,
+ /// You do not own this collection
+ NoPermission,
+ /// This address is not set as sponsor, use setCollectionSponsor first
+ ConfirmUnsetSponsorFail,
+ /// Collection is not in mint mode
+ PublicMintingNotAllowed,
+ /// Sender parameter and item owner must be equal
+ MustBeTokenOwner,
+ /// Item balance not enouth
+ TokenValueTooLow,
+ /// Size of item is too large
+ NftSizeLimitExceeded,
+ /// Size of item must be 0 with fungible type
+ FungibleUnexpectedParam,
+ /// No approve found
+ ApproveNotFound,
+ /// Requested value more than approved
+ TokenValueNotEnough,
+ /// Only approved addresses can call this method
+ ApproveRequired,
+ /// Address is not in white list
+ AddresNotInWhiteList,
+ /// Number of collection admins bound exceeded
+ CollectionAdminsLimitExceeded,
+ /// Owned tokens by a single address bound exceeded
+ AddressOwnershipLimitExceeded,
+ /// Length of items properties must be greater than 0
+ EmptyArgument,
+ }
+}
+
pub trait Trait: system::Trait + Sized + transaction_payment::Trait + pallet_contracts::Trait {
type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;
@@ -386,6 +431,7 @@
pub struct Module<T: Trait> for enum Call where origin: T::Origin {
fn deposit_event() = default;
+ type Error = Error<T>;
fn on_initialize(now: T::BlockNumber) -> Weight {
@@ -432,32 +478,32 @@
};
// bound Total number of collections
- ensure!(CollectionCount::get() < ChainLimit::get().collection_numbers_limit, "Total collections bound exceeded");
+ ensure!(CollectionCount::get() < ChainLimit::get().collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);
// check params
- ensure!(decimal_points <= 4, "decimal_points parameter must be lower than 4");
+ ensure!(decimal_points <= 4, Error::<T>::CollectionDecimalPointLimitExceeded);
let mut name = collection_name.to_vec();
name.push(0);
- ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");
+ ensure!(name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);
let mut description = collection_description.to_vec();
description.push(0);
- ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");
+ ensure!(name.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);
let mut prefix = token_prefix.to_vec();
prefix.push(0);
- ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");
+ ensure!(prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);
// Generate next collection ID
let next_id = CreatedCollectionCount::get()
.checked_add(1)
- .expect("collection id error");
+ .ok_or(Error::<T>::NumOverflow)?;
// bound counter
let total = CollectionCount::get()
.checked_add(1)
- .expect("collection counter error");
+ .ok_or(Error::<T>::NumOverflow)?;
CreatedCollectionCount::put(next_id);
CollectionCount::put(total);
@@ -489,7 +535,7 @@
}
/// **DANGEROUS**: Destroys collection and all NFTs within this collection. Users irrecoverably lose their assets and may lose real money.
- ///
+ ///
/// # Permissions
///
/// * Collection Owner.
@@ -524,7 +570,7 @@
// bound couter
let total = CollectionCount::get()
.checked_sub(1)
- .expect("collection counter error");
+ .ok_or(Error::<T>::NumOverflow)?;
CollectionCount::put(total);
}
@@ -693,11 +739,11 @@
if <AdminList<T>>::contains_key(collection_id)
{
admin_arr = <AdminList<T>>::get(collection_id);
- ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");
+ ensure!(!admin_arr.contains(&new_admin_id), Error::<T>::AlreadyAdmin);
}
// Number of collection admins
- ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, "Number of collection admins bound exceeded");
+ ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, Error::<T>::CollectionAdminsLimitExceeded);
admin_arr.push(new_admin_id);
<AdminList<T>>::insert(collection_id, admin_arr);
@@ -746,10 +792,10 @@
pub fn set_collection_sponsor(origin, collection_id: u64, new_sponsor: T::AccountId) -> DispatchResult {
let sender = ensure_signed(origin)?;
- ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");
+ ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);
let mut target_collection = <Collection<T>>::get(collection_id);
- ensure!(sender == target_collection.owner, "You do not own this collection");
+ ensure!(sender == target_collection.owner, Error::<T>::NoPermission);
target_collection.unconfirmed_sponsor = new_sponsor;
<Collection<T>>::insert(collection_id, target_collection);
@@ -768,10 +814,10 @@
pub fn confirm_sponsorship(origin, collection_id: u64) -> DispatchResult {
let sender = ensure_signed(origin)?;
- ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");
+ ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);
let mut target_collection = <Collection<T>>::get(collection_id);
- ensure!(sender == target_collection.unconfirmed_sponsor, "This address is not set as sponsor, use setCollectionSponsor first");
+ ensure!(sender == target_collection.unconfirmed_sponsor, Error::<T>::ConfirmUnsetSponsorFail);
target_collection.sponsor = target_collection.unconfirmed_sponsor;
target_collection.unconfirmed_sponsor = T::AccountId::default();
@@ -793,10 +839,10 @@
pub fn remove_collection_sponsor(origin, collection_id: u64) -> DispatchResult {
let sender = ensure_signed(origin)?;
- ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");
+ ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);
let mut target_collection = <Collection<T>>::get(collection_id);
- ensure!(sender == target_collection.owner, "You do not own this collection");
+ ensure!(sender == target_collection.owner, Error::<T>::NoPermission);
target_collection.sponsor = T::AccountId::default();
<Collection<T>>::insert(collection_id, target_collection);
@@ -832,79 +878,55 @@
pub fn create_item(origin, collection_id: u64, owner: T::AccountId, data: CreateItemData) -> DispatchResult {
let sender = ensure_signed(origin)?;
+
Self::collection_exists(collection_id)?;
+
let target_collection = <Collection<T>>::get(collection_id);
- if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {
- ensure!(target_collection.mint_mode == true, "Public minting is not allowed for this collection.");
- Self::check_white_list(collection_id, &owner)?;
- Self::check_white_list(collection_id, &sender)?;
- }
+ Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;
+ Self::validate_create_item_args(&target_collection, &data)?;
+ Self::create_item_no_validation(collection_id, &target_collection, owner, data)?;
+
+ Ok(())
+ }
- match target_collection.mode
- {
- CollectionMode::NFT => {
- if let CreateItemData::NFT(data) = data {
- // check sizes
- ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, "const_data exceeded data limit.");
- ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, "variable_data exceeded data limit.");
-
- // Create nft item
- let item = NftItemType {
- collection: collection_id,
- owner: owner,
- const_data: data.const_data.clone(),
- variable_data: data.variable_data.clone()
- };
-
- Self::add_nft_item(item)?;
-
- } else {
- fail!("Not NFT item data used to mint in NFT collection.");
- }
- },
- CollectionMode::Fungible(_) => {
- if let CreateItemData::Fungible(_) = data {
-
- let item = FungibleItemType {
- collection: collection_id,
- owner: owner,
- value: (10 as u128).pow(target_collection.decimal_points)
- };
-
- Self::add_fungible_item(item)?;
- } else {
- fail!("Not Fungible item data used to mint in Fungible collection.");
- }
- },
- CollectionMode::ReFungible(_) => {
- if let CreateItemData::ReFungible(data) = data {
-
- // check sizes
- ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, "const_data exceeded data limit.");
- ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, "variable_data exceeded data limit.");
-
- let mut owner_list = Vec::new();
- let value = (10 as u128).pow(target_collection.decimal_points);
- owner_list.push(Ownership {owner: owner.clone(), fraction: value});
-
- let item = ReFungibleItemType {
- collection: collection_id,
- owner: owner_list,
- const_data: data.const_data.clone(),
- variable_data: data.variable_data.clone()
- };
-
- Self::add_refungible_item(item)?;
- } else {
- fail!("Not Re Fungible item data used to mint in Re Fungible collection.");
- }
- },
- _ => { ensure!(1 == 0,"Unexpected collection type."); }
- };
+ /// This method creates multiple instances of NFT Collection created with CreateCollection method.
+ ///
+ /// # Permissions
+ ///
+ /// * Collection Owner.
+ /// * Collection Admin.
+ /// * Anyone if
+ /// * White List is enabled, and
+ /// * Address is added to white list, and
+ /// * MintPermission is enabled (see SetMintPermission method)
+ ///
+ /// # Arguments
+ ///
+ /// * collection_id: ID of the collection.
+ ///
+ /// * itemsData: Array items properties. Each property is an array of bytes itself, see [create_item].
+ ///
+ /// * owner: Address, initial owner of the NFT.
+ #[weight = T::WeightInfo::create_item(items_data.into_iter()
+ .map(|data| { data.len() })
+ .sum())]
+ pub fn create_multiple_items(origin, collection_id: u64, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {
+
+ ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);
+ let sender = ensure_signed(origin)?;
+
+ Self::collection_exists(collection_id)?;
+ let target_collection = <Collection<T>>::get(collection_id);
+
+ Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;
- // call event
- Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));
+ for data in &items_data {
+ Self::validate_create_item_args(&target_collection, data)?;
+ }
+ for data in &items_data {
+ Self::create_item_no_validation(collection_id, &target_collection, owner.clone(), data.clone())?;
+ }
Ok(())
}
@@ -932,7 +954,7 @@
let target_collection = <Collection<T>>::get(collection_id);
ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||
Self::is_owner_or_admin_permissions(collection_id, sender.clone()),
- "Only item owner, collection owner and admins can modify item");
+ Error::<T>::NoPermission);
if target_collection.access == AccessMode::WhiteList {
Self::check_white_list(collection_id, &sender)?;
@@ -984,7 +1006,7 @@
let target_collection = <Collection<T>>::get(collection_id);
ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||
Self::is_owner_or_admin_permissions(collection_id, sender.clone()),
- "Only item owner, collection owner and admins can modify item");
+ Error::<T>::NoPermission);
if target_collection.access == AccessMode::WhiteList {
Self::check_white_list(collection_id, &sender)?;
@@ -1026,7 +1048,7 @@
let target_collection = <Collection<T>>::get(collection_id);
ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||
Self::is_owner_or_admin_permissions(collection_id, sender.clone()),
- "Only item owner, collection owner and admins can approve");
+ Error::<T>::NoPermission);
if target_collection.access == AccessMode::WhiteList {
Self::check_white_list(collection_id, &sender)?;
@@ -1088,14 +1110,14 @@
if opt_item.is_some()
{
appoved_transfer = true;
- ensure!(opt_item.unwrap().amount >= value, "Requested value more than approved");
+ ensure!(opt_item.unwrap().amount >= value, Error::<T>::TokenValueNotEnough);
}
}
// Transfer permissions check
let target_collection = <Collection<T>>::get(collection_id);
- ensure!(appoved_transfer || Self::is_owner_or_admin_permissions(collection_id, sender.clone()),
- "Only item owner, collection owner and admins can modify items");
+ ensure!(appoved_transfer || Self::is_owner_or_admin_permissions(collection_id, sender.clone()),
+ Error::<T>::NoPermission);
if target_collection.access == AccessMode::WhiteList {
Self::check_white_list(collection_id, &sender)?;
@@ -1134,7 +1156,7 @@
Ok(())
}
-
+
/// Set off-chain data schema.
///
/// # Permissions
@@ -1162,7 +1184,7 @@
let target_collection = <Collection<T>>::get(collection_id);
ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||
Self::is_owner_or_admin_permissions(collection_id, sender.clone()),
- "Only item owner, collection owner and admins can modify item.");
+ Error::<T>::NoPermission);
Self::item_exists(collection_id, item_id, &target_collection.mode)?;
@@ -1296,7 +1318,7 @@
let owner = <ContractOwner<T>>::get(&contract_address);
is_owner = sender == owner;
}
- ensure!(is_owner, "Only contract owner may call this method");
+ ensure!(is_owner, Error::<T>::NoPermission);
<ContractSelfSponsoring<T>>::insert(contract_address, enable);
Ok(())
@@ -1306,10 +1328,101 @@
}
impl<T: Trait> Module<T> {
+
+ fn can_create_items_in_collection(collection_id: u64, collection: &CollectionType<T::AccountId>, sender: &T::AccountId, owner: &T::AccountId) -> DispatchResult {
+
+ if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {
+ ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);
+ Self::check_white_list(collection_id, owner)?;
+ Self::check_white_list(collection_id, sender)?;
+ }
+
+ Ok(())
+ }
+
+ fn validate_create_item_args(target_collection: &CollectionType<T::AccountId>, data: &CreateItemData) -> DispatchResult {
+ match target_collection.mode
+ {
+ CollectionMode::NFT => {
+ if let CreateItemData::NFT(data) = data {
+ // check sizes
+ ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, "const_data exceeded data limit.");
+ ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, "variable_data exceeded data limit.");
+ } else {
+ fail!("Not NFT item data used to mint in NFT collection.");
+ }
+ },
+ CollectionMode::Fungible(_) => {
+ if let CreateItemData::Fungible(_) = data {
+ } else {
+ fail!("Not Fungible item data used to mint in Fungible collection.");
+ }
+ },
+ CollectionMode::ReFungible(_) => {
+ if let CreateItemData::ReFungible(data) = data {
+
+ // check sizes
+ ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, "const_data exceeded data limit.");
+ ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, "variable_data exceeded data limit.");
+ } else {
+ fail!("Not Re Fungible item data used to mint in Re Fungible collection.");
+ }
+ },
+ _ => { fail!("Unexpected collection type."); }
+ };
+
+ Ok(())
+ }
+
+ fn create_item_no_validation(collection_id: u64, collection: &CollectionType<T::AccountId>, owner: T::AccountId, data: CreateItemData) -> DispatchResult {
+ match data
+ {
+ CreateItemData::NFT(data) => {
+ let item = NftItemType {
+ collection: collection_id,
+ owner,
+ const_data: data.const_data,
+ variable_data: data.variable_data
+ };
+
+ Self::add_nft_item(item)?;
+ },
+ CreateItemData::Fungible(_) => {
+ let item = FungibleItemType {
+ collection: collection_id,
+ owner,
+ value: (10 as u128).pow(collection.decimal_points)
+ };
+
+ Self::add_fungible_item(item)?;
+ },
+ CreateItemData::ReFungible(data) => {
+ let mut owner_list = Vec::new();
+ let value = (10 as u128).pow(collection.decimal_points);
+ owner_list.push(Ownership {owner: owner.clone(), fraction: value});
+
+ let item = ReFungibleItemType {
+ collection: collection_id,
+ owner: owner_list,
+ const_data: data.const_data,
+ variable_data: data.variable_data
+ };
+
+ Self::add_refungible_item(item)?;
+ }
+ };
+
+
+ // call event
+ Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));
+
+ Ok(())
+ }
+
fn add_fungible_item(item: FungibleItemType<T::AccountId>) -> DispatchResult {
let current_index = <ItemListIndex>::get(item.collection)
.checked_add(1)
- .expect("Item list index id error");
+ .ok_or(Error::<T>::NumOverflow)?;
let itemcopy = item.clone();
let owner = item.owner.clone();
let value = item.value as u64;
@@ -1326,7 +1439,7 @@
// Update balance
let new_balance = <Balance<T>>::get(item.collection, owner.clone())
.checked_add(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(item.collection, owner.clone(), new_balance);
Ok(())
@@ -1335,7 +1448,7 @@
fn add_refungible_item(item: ReFungibleItemType<T::AccountId>) -> DispatchResult {
let current_index = <ItemListIndex>::get(item.collection)
.checked_add(1)
- .expect("Item list index id error");
+ .ok_or(Error::<T>::NumOverflow)?;
let itemcopy = item.clone();
let value = item.owner.first().unwrap().fraction as u64;
@@ -1353,7 +1466,7 @@
// Update balance
let new_balance = <Balance<T>>::get(item.collection, owner.clone())
.checked_add(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(item.collection, owner.clone(), new_balance);
Ok(())
@@ -1362,7 +1475,7 @@
fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {
let current_index = <ItemListIndex>::get(item.collection)
.checked_add(1)
- .expect("Item list index id error");
+ .ok_or(Error::<T>::NumOverflow)?;
let item_owner = item.owner.clone();
let collection_id = item.collection.clone();
@@ -1378,7 +1491,7 @@
// Update balance
let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())
.checked_add(1)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);
Ok(())
@@ -1391,7 +1504,7 @@
) -> DispatchResult {
ensure!(
<ReFungibleItemList<T>>::contains_key(collection_id, item_id),
- "Item does not exists"
+ Error::<T>::TokenNotFound
);
let collection = <ReFungibleItemList<T>>::get(collection_id, item_id);
let item = collection
@@ -1408,7 +1521,7 @@
// update balance
let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())
.checked_sub(item.fraction as u64)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);
<ReFungibleItemList<T>>::remove(collection_id, item_id);
@@ -1419,7 +1532,7 @@
fn burn_nft_item(collection_id: u64, item_id: u64) -> DispatchResult {
ensure!(
<NftItemList<T>>::contains_key(collection_id, item_id),
- "Item does not exists"
+ Error::<T>::TokenNotFound
);
let item = <NftItemList<T>>::get(collection_id, item_id);
Self::remove_token_index(collection_id, item_id, item.owner.clone())?;
@@ -1430,7 +1543,7 @@
// update balance
let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())
.checked_sub(1)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);
<NftItemList<T>>::remove(collection_id, item_id);
@@ -1440,7 +1553,7 @@
fn burn_fungible_item(collection_id: u64, item_id: u64) -> DispatchResult {
ensure!(
<FungibleItemList<T>>::contains_key(collection_id, item_id),
- "Item does not exists"
+ Error::<T>::TokenNotFound
);
let item = <FungibleItemList<T>>::get(collection_id, item_id);
Self::remove_token_index(collection_id, item_id, item.owner.clone())?;
@@ -1451,7 +1564,7 @@
// update balance
let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())
.checked_sub(item.value as u64)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);
<FungibleItemList<T>>::remove(collection_id, item_id);
@@ -1462,7 +1575,7 @@
fn collection_exists(collection_id: u64) -> DispatchResult {
ensure!(
<Collection<T>>::contains_key(collection_id),
- "This collection does not exist"
+ Error::<T>::CollectionNotFound
);
Ok(())
}
@@ -1473,7 +1586,7 @@
let target_collection = <Collection<T>>::get(collection_id);
ensure!(
subject == target_collection.owner,
- "You do not own this collection"
+ Error::<T>::NoPermission
);
Ok(())
@@ -1502,7 +1615,7 @@
ensure!(
result,
- "You do not have permissions to modify this collection"
+ Error::<T>::NoPermission
);
Ok(())
}
@@ -1528,7 +1641,7 @@
}
fn check_white_list(collection_id: u64, address: &T::AccountId) -> DispatchResult {
- let mes = "Address is not in white list";
+ let mes = Error::<T>::AddresNotInWhiteList;
ensure!(<WhiteList<T>>::contains_key(collection_id), mes);
let wl = <WhiteList<T>>::get(collection_id);
ensure!(wl.contains(address), mes);
@@ -1545,18 +1658,18 @@
) -> DispatchResult {
ensure!(
<FungibleItemList<T>>::contains_key(collection_id, item_id),
- "Item not exists"
+ Error::<T>::TokenNotFound
);
let full_item = <FungibleItemList<T>>::get(collection_id, item_id);
let amount = full_item.value;
- ensure!(amount >= value.into(), "Item balance not enouth");
+ ensure!(amount >= value.into(), Error::<T>::TokenValueTooLow);
// update balance
let balance_old_owner = <Balance<T>>::get(collection_id, owner.clone())
.checked_sub(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, owner.clone(), balance_old_owner);
let mut new_owner_account_id = 0;
@@ -1578,7 +1691,7 @@
// update balance
let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())
.checked_add(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);
// update index collection
@@ -1596,7 +1709,7 @@
// update balance
let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())
.checked_add(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);
<FungibleItemList<T>>::insert(collection_id, new_owner_account_id, item);
@@ -1634,7 +1747,7 @@
) -> DispatchResult {
ensure!(
<ReFungibleItemList<T>>::contains_key(collection_id, item_id),
- "Item not exists"
+ Error::<T>::TokenNotFound
);
let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);
@@ -1643,20 +1756,20 @@
.iter()
.filter(|i| i.owner == owner)
.next()
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
let amount = item.fraction;
- ensure!(amount >= value.into(), "Item balance not enouth");
+ ensure!(amount >= value.into(), Error::<T>::TokenValueTooLow);
// update balance
let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())
.checked_sub(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);
let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())
.checked_add(value)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);
let old_owner = item.owner.clone();
@@ -1719,25 +1832,25 @@
) -> DispatchResult {
ensure!(
<NftItemList<T>>::contains_key(collection_id, item_id),
- "Item not exists"
+ Error::<T>::TokenNotFound
);
let mut item = <NftItemList<T>>::get(collection_id, item_id);
ensure!(
sender == item.owner,
- "sender parameter and item owner must be equal"
+ Error::<T>::MustBeTokenOwner
);
// update balance
let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())
.checked_sub(1)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);
let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())
.checked_add(1)
- .unwrap();
+ .ok_or(Error::<T>::NumOverflow)?;
<Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);
// change owner
@@ -1759,9 +1872,9 @@
mode: &CollectionMode
) -> DispatchResult {
match mode {
- CollectionMode::NFT => ensure!(<NftItemList<T>>::contains_key(collection_id, item_id), "Item does not exists"),
- CollectionMode::ReFungible(_) => ensure!(<ReFungibleItemList<T>>::contains_key(collection_id, item_id), "Item does not exists"),
- CollectionMode::Fungible(_) => ensure!(<FungibleItemList<T>>::contains_key(collection_id, item_id), "Item does not exists"),
+ CollectionMode::NFT => ensure!(<NftItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),
+ CollectionMode::ReFungible(_) => ensure!(<ReFungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),
+ CollectionMode::Fungible(_) => ensure!(<FungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),
_ => ()
};
@@ -1818,7 +1931,7 @@
// Generate next collection ID
let next_id = CreatedCollectionCount::get()
.checked_add(1)
- .expect("collection id error");
+ .unwrap();
CreatedCollectionCount::put(next_id);
}
@@ -1826,7 +1939,7 @@
fn init_nft_token(item: &NftItemType<T::AccountId>) {
let current_index = <ItemListIndex>::get(item.collection)
.checked_add(1)
- .expect("Item list index id error");
+ .unwrap();
let item_owner = item.owner.clone();
let collection_id = item.collection.clone();
@@ -1844,7 +1957,7 @@
fn init_fungible_token(item: &FungibleItemType<T::AccountId>) {
let current_index = <ItemListIndex>::get(item.collection)
.checked_add(1)
- .expect("Item list index id error");
+ .unwrap();
let owner = item.owner.clone();
let value = item.value as u64;
@@ -1862,7 +1975,7 @@
fn init_refungible_token(item: &ReFungibleItemType<T::AccountId>) {
let current_index = <ItemListIndex>::get(item.collection)
.checked_add(1)
- .expect("Item list index id error");
+ .unwrap();
let value = item.owner.first().unwrap().fraction as u64;
let owner = item.owner.first().unwrap().owner.clone();
@@ -1885,10 +1998,11 @@
// bound Owned tokens by a single address
let count = <AccountItemCount<T>>::get(owner.clone());
- ensure!(count < ChainLimit::get().account_token_ownership_limit, "Owned tokens by a single address bound exceeded");
+ ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);
- <AccountItemCount<T>>::insert(owner.clone(),
- count.checked_add(1).unwrap());
+ <AccountItemCount<T>>::insert(owner.clone(), count
+ .checked_add(1)
+ .ok_or(Error::<T>::NumOverflow)?);
}
else {
<AccountItemCount<T>>::insert(owner.clone(), 1);
@@ -1922,7 +2036,9 @@
// update counter
<AccountItemCount<T>>::insert(owner.clone(),
- <AccountItemCount<T>>::get(owner.clone()).checked_sub(1).unwrap());
+ <AccountItemCount<T>>::get(owner.clone())
+ .checked_sub(1)
+ .ok_or(Error::<T>::NumOverflow)?);
let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());
@@ -2236,5 +2352,3 @@
}
// #endregion
-
-
pallets/nft/src/tests.rsdiffbeforeafterboth1// Tests to be written here2use crate::mock::*;3use crate::{AccessMode, ApprovePermissions, CollectionMode, Ownership, ChainLimits, CreateItemData, CreateNftData, CreateFungibleData, CreateReFungibleData};4use frame_support::{assert_noop, assert_ok};5use frame_system::{ RawOrigin };67fn default_collection_numbers_limit() -> u64 {8 109}1011fn default_limits() {12 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 13 collection_numbers_limit: default_collection_numbers_limit(),14 account_token_ownership_limit: 10,15 collections_admins_limit: 5,16 custom_data_limit: 2048,17 nft_sponsor_transfer_timeout: 15,18 fungible_sponsor_transfer_timeout: 15,19 refungible_sponsor_transfer_timeout: 15, 20 }));21}2223fn default_nft_data() -> CreateNftData {24 CreateNftData { const_data: vec![1, 2, 3], variable_data: vec![3, 2, 1] }25}2627fn default_fungible_data () -> CreateFungibleData {28 CreateFungibleData { }29}3031fn default_re_fungible_data () -> CreateReFungibleData {32 CreateReFungibleData { const_data: vec![1, 2, 3], variable_data: vec![3, 2, 1] }33}3435fn create_test_collection_for_owner(mode: &CollectionMode, owner: u64, id: u64) -> u64 {36 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();37 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();38 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();3940 let origin1 = Origin::signed(owner);41 assert_ok!(TemplateModule::create_collection(42 origin1.clone(),43 col_name1.clone(),44 col_desc1.clone(),45 token_prefix1.clone(),46 mode.clone()47 ));4849 let saved_col_name: Vec<u16> = "Test1\0\0".encode_utf16().collect::<Vec<u16>>();50 let saved_description: Vec<u16> = "TestDescription1\0\0".encode_utf16().collect::<Vec<u16>>();51 let saved_prefix: Vec<u8> = b"token_prefix1\0\0".to_vec();52 assert_eq!(TemplateModule::collection(id).owner, owner);53 assert_eq!(TemplateModule::collection(id).name, saved_col_name);54 assert_eq!(TemplateModule::collection(id).mode, *mode);55 assert_eq!(TemplateModule::collection(id).description, saved_description);56 assert_eq!(TemplateModule::collection(id).token_prefix, saved_prefix);57 id58}5960fn create_test_collection(mode: &CollectionMode, id: u64) -> u64 {61 create_test_collection_for_owner(&mode, 1, id)62}6364fn create_test_item(collection_id: u64, data: &CreateItemData) {65 let origin1 = Origin::signed(1);66 assert_ok!(TemplateModule::create_item(67 origin1.clone(),68 collection_id,69 1,70 data.clone()71 ));7273}7475// Use cases tests region76// #region77#[test]78fn create_nft_item() {79 new_test_ext().execute_with(|| {80 default_limits();81 let collection_id = create_test_collection(&CollectionMode::NFT, 1);82 83 let data = default_nft_data();84 create_test_item(collection_id, &data.clone().into());85 assert_eq!(TemplateModule::nft_item_id(collection_id, 1).const_data, data.const_data);86 assert_eq!(TemplateModule::nft_item_id(collection_id, 1).variable_data, data.variable_data);87 });88}8990#[test]91fn create_refungible_item() {92 new_test_ext().execute_with(|| {93 default_limits();94 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);9596 let data = default_re_fungible_data();97 create_test_item(collection_id, &data.clone().into());98 assert_eq!(99 TemplateModule::refungible_item_id(collection_id, 1).const_data,100 data.const_data101 );102 assert_eq!(103 TemplateModule::refungible_item_id(collection_id, 1).variable_data,104 data.variable_data105 );106 assert_eq!(107 TemplateModule::refungible_item_id(collection_id, 1).owner[0],108 Ownership {109 owner: 1,110 fraction: 1000111 }112 );113 });114}115116#[test]117fn create_fungible_item() {118 new_test_ext().execute_with(|| {119 default_limits();120 121 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);122123 let data = default_fungible_data();124 create_test_item(collection_id, &data.into());125126 assert_eq!(TemplateModule::fungible_item_id(collection_id, 1).owner, 1);127 assert_eq!(TemplateModule::balance_count(1, 1), 1000);128 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);129 });130}131132#[test]133fn transfer_fungible_item() {134 new_test_ext().execute_with(|| {135 default_limits();136 137 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);138139 let origin1 = Origin::signed(1);140 let origin2 = Origin::signed(2);141142 let data = default_fungible_data();143 create_test_item(collection_id, &data.into());144145 assert_eq!(TemplateModule::fungible_item_id(1, 1).owner, 1);146 assert_eq!(TemplateModule::balance_count(1, 1), 1000);147 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);148149 // change owner scenario150 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1000));151 assert_eq!(TemplateModule::fungible_item_id(1, 1).owner, 2);152 assert_eq!(TemplateModule::fungible_item_id(1, 1).value, 1000);153 assert_eq!(TemplateModule::balance_count(1, 1), 0);154 assert_eq!(TemplateModule::balance_count(1, 2), 1000);155 // assert_eq!(TemplateModule::address_tokens(1, 1), []);156 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);157158 // split item scenario159 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 500));160 assert_eq!(TemplateModule::fungible_item_id(1, 1).owner, 2);161 assert_eq!(TemplateModule::fungible_item_id(1, 2).owner, 3);162 assert_eq!(TemplateModule::balance_count(1, 2), 500);163 assert_eq!(TemplateModule::balance_count(1, 3), 500);164 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);165 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);166167 // split item and new owner has account scenario168 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 200));169 assert_eq!(TemplateModule::fungible_item_id(1, 1).value, 300);170 assert_eq!(TemplateModule::fungible_item_id(1, 2).value, 700);171 assert_eq!(TemplateModule::balance_count(1, 2), 300);172 assert_eq!(TemplateModule::balance_count(1, 3), 700);173 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);174 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);175 });176}177178#[test]179fn transfer_refungible_item() {180 new_test_ext().execute_with(|| {181 default_limits();182 183 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);184185 let data = default_re_fungible_data();186 create_test_item(collection_id, &data.clone().into());187188 let origin1 = Origin::signed(1);189 let origin2 = Origin::signed(2);190 assert_eq!(191 TemplateModule::refungible_item_id(collection_id, 1).const_data,192 data.const_data193 );194 assert_eq!(195 TemplateModule::refungible_item_id(collection_id, 1).variable_data,196 data.variable_data197 );198 assert_eq!(199 TemplateModule::refungible_item_id(collection_id, 1).owner[0],200 Ownership {201 owner: 1,202 fraction: 1000203 }204 );205 assert_eq!(TemplateModule::balance_count(1, 1), 1000);206 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);207208 // change owner scenario209 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1000));210 assert_eq!(211 TemplateModule::refungible_item_id(1, 1).owner[0],212 Ownership {213 owner: 2,214 fraction: 1000215 }216 );217 assert_eq!(TemplateModule::balance_count(1, 1), 0);218 assert_eq!(TemplateModule::balance_count(1, 2), 1000);219 // assert_eq!(TemplateModule::address_tokens(1, 1), []);220 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);221222 // split item scenario223 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 500));224 assert_eq!(225 TemplateModule::refungible_item_id(1, 1).owner[0],226 Ownership {227 owner: 2,228 fraction: 500229 }230 );231 assert_eq!(232 TemplateModule::refungible_item_id(1, 1).owner[1],233 Ownership {234 owner: 3,235 fraction: 500236 }237 );238 assert_eq!(TemplateModule::balance_count(1, 2), 500);239 assert_eq!(TemplateModule::balance_count(1, 3), 500);240 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);241 assert_eq!(TemplateModule::address_tokens(1, 3), [1]);242243 // split item and new owner has account scenario244 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 200));245 assert_eq!(246 TemplateModule::refungible_item_id(1, 1).owner[0],247 Ownership {248 owner: 2,249 fraction: 300250 }251 );252 assert_eq!(253 TemplateModule::refungible_item_id(1, 1).owner[1],254 Ownership {255 owner: 3,256 fraction: 700257 }258 );259 assert_eq!(TemplateModule::balance_count(1, 2), 300);260 assert_eq!(TemplateModule::balance_count(1, 3), 700);261 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);262 assert_eq!(TemplateModule::address_tokens(1, 3), [1]);263 });264}265266#[test]267fn transfer_nft_item() {268 new_test_ext().execute_with(|| {269 default_limits();270 271 let collection_id = create_test_collection(&CollectionMode::NFT, 1);272273 let data = default_nft_data();274 create_test_item(collection_id, &data.into());275 assert_eq!(TemplateModule::balance_count(1, 1), 1);276 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);277278 let origin1 = Origin::signed(1);279 // default scenario280 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1000));281 assert_eq!(TemplateModule::nft_item_id(1, 1).owner, 2);282 assert_eq!(TemplateModule::balance_count(1, 1), 0);283 assert_eq!(TemplateModule::balance_count(1, 2), 1);284 // assert_eq!(TemplateModule::address_tokens(1, 1), []);285 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);286 });287}288289#[test]290fn nft_approve_and_transfer_from() {291 new_test_ext().execute_with(|| {292 default_limits();293 294 let collection_id = create_test_collection(&CollectionMode::NFT, 1);295296 let data = default_nft_data();297 create_test_item(collection_id, &data.into());298299 let origin1 = Origin::signed(1);300 let origin2 = Origin::signed(2);301302 assert_eq!(TemplateModule::balance_count(1, 1), 1);303 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);304305 // neg transfer306 assert_noop!(TemplateModule::transfer_from(307 origin2.clone(),308 1,309 2,310 1,311 1,312 1), "Only item owner, collection owner and admins can modify items");313314 // do approve315 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));316 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);317 assert_eq!(318 TemplateModule::approved(1, (1, 1))[0],319 ApprovePermissions {320 approved: 2,321 amount: 100000000322 }323 );324325 assert_ok!(TemplateModule::transfer_from(326 origin2.clone(),327 1,328 2,329 1,330 1,331 1332 ));333 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 0);334 });335}336337#[test]338fn nft_approve_and_transfer_from_white_list() {339 new_test_ext().execute_with(|| {340 default_limits();341 342 let collection_id = create_test_collection(&CollectionMode::NFT, 1);343344 let origin1 = Origin::signed(1);345 let origin2 = Origin::signed(2);346347 let data = default_nft_data();348 create_test_item(collection_id, &data.clone().into());349350 assert_eq!(TemplateModule::nft_item_id(1, 1).const_data, data.const_data);351 assert_eq!(TemplateModule::balance_count(1, 1), 1);352 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);353354 assert_ok!(TemplateModule::set_mint_permission(355 origin1.clone(),356 1,357 true358 ));359 assert_ok!(TemplateModule::set_public_access_mode(360 origin1.clone(),361 1,362 AccessMode::WhiteList363 ));364 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));365 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));366 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));367368 // do approve369 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));370 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);371 assert_ok!(TemplateModule::approve(origin1.clone(), 3, 1, 1));372 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 2);373 assert_eq!(374 TemplateModule::approved(1, (1, 1))[0],375 ApprovePermissions {376 approved: 2,377 amount: 100000000378 }379 );380381 assert_ok!(TemplateModule::transfer_from(382 origin2.clone(),383 1,384 3,385 1,386 1,387 1388 ));389 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 0);390 });391}392393#[test]394fn refungible_approve_and_transfer_from() {395 new_test_ext().execute_with(|| {396 default_limits();397 398 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);399 400 let origin1 = Origin::signed(1);401 let origin2 = Origin::signed(2);402403 let data = default_re_fungible_data();404 create_test_item(collection_id, &data.into());405406 assert_eq!(TemplateModule::balance_count(1, 1), 1000);407 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);408409 assert_ok!(TemplateModule::set_mint_permission(410 origin1.clone(),411 1,412 true413 ));414 assert_ok!(TemplateModule::set_public_access_mode(415 origin1.clone(),416 1,417 AccessMode::WhiteList418 ));419 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));420 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));421 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));422423 // do approve424 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));425 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);426 assert_ok!(TemplateModule::approve(origin1.clone(), 3, 1, 1));427 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 2);428 assert_eq!(429 TemplateModule::approved(1, (1, 1))[0],430 ApprovePermissions {431 approved: 2,432 amount: 100000000433 }434 );435436 assert_ok!(TemplateModule::transfer_from(437 origin2.clone(),438 1,439 3,440 1,441 1,442 100443 ));444 assert_eq!(TemplateModule::balance_count(1, 1), 900);445 assert_eq!(TemplateModule::balance_count(1, 3), 100);446 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);447 assert_eq!(TemplateModule::address_tokens(1, 3), [1]);448449 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);450 assert_eq!(451 TemplateModule::approved(1, (1, 1))[0],452 ApprovePermissions {453 approved: 3,454 amount: 100000000455 }456 );457 });458}459460#[test]461fn fungible_approve_and_transfer_from() {462 new_test_ext().execute_with(|| {463 default_limits();464 465 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);466 467 let data = default_fungible_data();468 create_test_item(collection_id, &data.into());469470 let origin1 = Origin::signed(1);471 let origin2 = Origin::signed(2);472473 assert_eq!(TemplateModule::balance_count(1, 1), 1000);474 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);475476 assert_ok!(TemplateModule::set_mint_permission(477 origin1.clone(),478 1,479 true480 ));481 assert_ok!(TemplateModule::set_public_access_mode(482 origin1.clone(),483 1,484 AccessMode::WhiteList485 ));486 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));487 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));488 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));489490 // do approve491 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));492 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);493 assert_ok!(TemplateModule::approve(origin1.clone(), 3, 1, 1));494 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 2);495 assert_eq!(496 TemplateModule::approved(1, (1, 1))[0],497 ApprovePermissions {498 approved: 2,499 amount: 100000000500 }501 );502503 assert_ok!(TemplateModule::transfer_from(504 origin2.clone(),505 1,506 3,507 1,508 1,509 100510 ));511 assert_eq!(TemplateModule::balance_count(1, 1), 900);512 assert_eq!(TemplateModule::balance_count(1, 3), 100);513 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);514 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);515516 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);517 assert_eq!(518 TemplateModule::approved(1, (1, 1))[0],519 ApprovePermissions {520 approved: 3,521 amount: 100000000522 }523 );524525 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));526 assert_ok!(TemplateModule::transfer_from(527 origin2.clone(),528 1,529 3,530 1,531 1,532 900533 ));534 assert_eq!(TemplateModule::balance_count(1, 1), 0);535 assert_eq!(TemplateModule::balance_count(1, 3), 1000);536 // assert_eq!(TemplateModule::address_tokens(1, 1), []);537 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);538539 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 0);540 });541}542543#[test]544fn change_collection_owner() {545 new_test_ext().execute_with(|| {546 default_limits();547 548 let collection_id = create_test_collection(&CollectionMode::NFT, 1);549 550 let origin1 = Origin::signed(1);551 assert_ok!(TemplateModule::change_collection_owner(552 origin1.clone(),553 collection_id,554 2555 ));556 assert_eq!(TemplateModule::collection(collection_id).owner, 2);557 });558}559560#[test]561fn destroy_collection() {562 new_test_ext().execute_with(|| {563 default_limits();564 565 let collection_id = create_test_collection(&CollectionMode::NFT, 1);566 567 let origin1 = Origin::signed(1);568 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), collection_id));569 });570}571572#[test]573fn burn_nft_item() {574 new_test_ext().execute_with(|| {575 default_limits();576 577 let collection_id = create_test_collection(&CollectionMode::NFT, 1);578579 let origin1 = Origin::signed(1);580 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));581 582 let data = default_nft_data();583 create_test_item(collection_id, &data.into());584585 // check balance (collection with id = 1, user id = 1)586 assert_eq!(TemplateModule::balance_count(1, 1), 1);587588 // burn item589 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));590 assert_noop!(591 TemplateModule::burn_item(origin1.clone(), 1, 1),592 "Item does not exists"593 );594595 assert_eq!(TemplateModule::balance_count(1, 1), 0);596 });597}598599#[test]600fn burn_fungible_item() {601 new_test_ext().execute_with(|| {602 default_limits();603 604 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);605 606 let origin1 = Origin::signed(1);607 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));608 609 let data = default_fungible_data();610 create_test_item(collection_id, &data.into());611612 // check balance (collection with id = 1, user id = 1)613 assert_eq!(TemplateModule::balance_count(1, 1), 1000);614615 // burn item616 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));617 assert_noop!(618 TemplateModule::burn_item(origin1.clone(), 1, 1),619 "Item does not exists"620 );621622 assert_eq!(TemplateModule::balance_count(1, 1), 0);623 });624}625626#[test]627fn burn_refungible_item() {628 new_test_ext().execute_with(|| {629 default_limits();630 631 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);632 let origin1 = Origin::signed(1);633634 assert_ok!(TemplateModule::set_mint_permission(635 origin1.clone(),636 collection_id,637 true638 ));639 assert_ok!(TemplateModule::set_public_access_mode(640 origin1.clone(),641 collection_id,642 AccessMode::WhiteList643 ));644 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));645646 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));647 648 let data = default_re_fungible_data();649 create_test_item(collection_id, &data.into());650651 // check balance (collection with id = 1, user id = 2)652 assert_eq!(TemplateModule::balance_count(1, 1), 1000);653654 // burn item655 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));656 assert_noop!(657 TemplateModule::burn_item(origin1.clone(), 1, 1),658 "Item does not exists"659 );660661 assert_eq!(TemplateModule::balance_count(1, 1), 0);662 });663}664665#[test]666fn add_collection_admin() {667 new_test_ext().execute_with(|| {668 default_limits();669 670 let collection1_id = create_test_collection_for_owner(&CollectionMode::NFT, 1, 1);671 create_test_collection_for_owner(&CollectionMode::NFT, 2, 2);672 create_test_collection_for_owner(&CollectionMode::NFT, 3, 3);673 674 let origin1 = Origin::signed(1);675676 // collection admin677 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 2));678 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 3));679680 assert_eq!(TemplateModule::admin_list_collection(collection1_id).contains(&2), true);681 assert_eq!(TemplateModule::admin_list_collection(collection1_id).contains(&3), true);682 });683}684685#[test]686fn remove_collection_admin() {687 new_test_ext().execute_with(|| {688 default_limits();689 690 let collection1_id = create_test_collection_for_owner(&CollectionMode::NFT, 1, 1);691 create_test_collection_for_owner(&CollectionMode::NFT, 2, 2);692 create_test_collection_for_owner(&CollectionMode::NFT, 3, 3);693694 let origin1 = Origin::signed(1);695 let origin2 = Origin::signed(2);696697 // collection admin698 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 2));699 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 3));700701 assert_eq!(TemplateModule::admin_list_collection(1).contains(&2), true);702 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), true);703704 // remove admin705 assert_ok!(TemplateModule::remove_collection_admin(706 origin2.clone(),707 1,708 3709 ));710 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), false);711 });712}713714#[test]715fn balance_of() {716 new_test_ext().execute_with(|| {717 default_limits();718 719 let nft_collection_id = create_test_collection(&CollectionMode::NFT, 1);720 let fungible_collection_id = create_test_collection(&CollectionMode::Fungible(3), 2);721 let re_fungible_collection_id = create_test_collection(&CollectionMode::ReFungible(3), 3);722 723 // check balance before724 assert_eq!(TemplateModule::balance_count(nft_collection_id, 1), 0);725 assert_eq!(TemplateModule::balance_count(fungible_collection_id, 1), 0);726 assert_eq!(TemplateModule::balance_count(re_fungible_collection_id, 1), 0);727728 let nft_data = default_nft_data();729 create_test_item(nft_collection_id, &nft_data.into());730 731 let fungible_data = default_fungible_data();732 create_test_item(fungible_collection_id, &fungible_data.into());733 734 let re_fungible_data = default_re_fungible_data();735 create_test_item(re_fungible_collection_id, &re_fungible_data.into());736737 // check balance (collection with id = 1, user id = 1)738 assert_eq!(TemplateModule::balance_count(nft_collection_id, 1), 1);739 assert_eq!(TemplateModule::balance_count(fungible_collection_id, 1), 1000);740 assert_eq!(TemplateModule::balance_count(re_fungible_collection_id, 1), 1000);741 assert_eq!(TemplateModule::nft_item_id(nft_collection_id, 1).owner, 1);742 assert_eq!(TemplateModule::fungible_item_id(fungible_collection_id, 1).owner, 1);743 assert_eq!(TemplateModule::refungible_item_id(re_fungible_collection_id, 1).owner[0].owner, 1);744 });745}746747#[test]748fn approve() {749 new_test_ext().execute_with(|| {750 default_limits();751 752 let collection_id = create_test_collection(&CollectionMode::NFT, 1);753 754 let data = default_nft_data();755 create_test_item(collection_id, &data.into());756757 let origin1 = Origin::signed(1);758 759 // approve760 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));761 assert_eq!(TemplateModule::approved(1, (1, 1))[0].approved, 2);762 });763}764765#[test]766fn transfer_from() {767 new_test_ext().execute_with(|| {768 default_limits();769 770 let collection_id = create_test_collection(&CollectionMode::NFT, 1);771 let origin1 = Origin::signed(1);772 let origin2 = Origin::signed(2);773774 let data = default_nft_data();775 create_test_item(collection_id, &data.into());776777 // approve778 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));779 assert_eq!(TemplateModule::approved(1, (1, 1))[0].approved, 2);780781 assert_ok!(TemplateModule::set_mint_permission(782 origin1.clone(),783 1,784 true785 ));786 assert_ok!(TemplateModule::set_public_access_mode(787 origin1.clone(),788 1,789 AccessMode::WhiteList790 ));791 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));792 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));793 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));794795 assert_ok!(TemplateModule::transfer_from(796 origin2.clone(),797 1,798 2,799 1,800 1,801 1802 ));803804 // after transfer805 assert_eq!(TemplateModule::balance_count(1, 1), 0);806 assert_eq!(TemplateModule::balance_count(1, 2), 1);807 });808}809810// #endregion811812// Coverage tests region813// #region814815#[test]816fn owner_can_add_address_to_white_list() {817 new_test_ext().execute_with(|| {818 default_limits();819 820 let collection_id = create_test_collection(&CollectionMode::NFT, 1);821822 let origin1 = Origin::signed(1);823 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));824 assert_eq!(TemplateModule::white_list(collection_id)[0], 2);825 });826}827828#[test]829fn admin_can_add_address_to_white_list() {830 new_test_ext().execute_with(|| {831 default_limits();832 833 let collection_id = create_test_collection(&CollectionMode::NFT, 1);834 let origin1 = Origin::signed(1);835 let origin2 = Origin::signed(2);836837 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));838 assert_ok!(TemplateModule::add_to_white_list(origin2.clone(), collection_id, 3));839 assert_eq!(TemplateModule::white_list(collection_id)[0], 3);840 });841}842843#[test]844fn nonprivileged_user_cannot_add_address_to_white_list() {845 new_test_ext().execute_with(|| {846 default_limits();847 848 let collection_id = create_test_collection(&CollectionMode::NFT, 1);849850 let origin2 = Origin::signed(2);851 assert_noop!(852 TemplateModule::add_to_white_list(origin2.clone(), collection_id, 3),853 "You do not have permissions to modify this collection"854 );855 });856}857858#[test]859fn nobody_can_add_address_to_white_list_of_nonexisting_collection() {860 new_test_ext().execute_with(|| {861 default_limits();862863 let origin1 = Origin::signed(1);864865 assert_noop!(866 TemplateModule::add_to_white_list(origin1.clone(), 1, 2),867 "This collection does not exist"868 );869 });870}871872#[test]873fn nobody_can_add_address_to_white_list_of_deleted_collection() {874 new_test_ext().execute_with(|| {875 default_limits();876 877 let collection_id = create_test_collection(&CollectionMode::NFT, 1);878879 let origin1 = Origin::signed(1);880 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), collection_id));881 assert_noop!(882 TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2),883 "This collection does not exist"884 );885 });886}887888// If address is already added to white list, nothing happens889#[test]890fn address_is_already_added_to_white_list() {891 new_test_ext().execute_with(|| {892 default_limits();893 894 let collection_id = create_test_collection(&CollectionMode::NFT, 1);895 let origin1 = Origin::signed(1);896 897 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));898 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));899 assert_eq!(TemplateModule::white_list(collection_id)[0], 2);900 assert_eq!(TemplateModule::white_list(collection_id).len(), 1);901 });902}903904#[test]905fn owner_can_remove_address_from_white_list() {906 new_test_ext().execute_with(|| {907 default_limits();908 909 let collection_id = create_test_collection(&CollectionMode::NFT, 1);910911 let origin1 = Origin::signed(1);912 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));913 assert_ok!(TemplateModule::remove_from_white_list(914 origin1.clone(),915 collection_id,916 2917 ));918 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);919 });920}921922#[test]923fn admin_can_remove_address_from_white_list() {924 new_test_ext().execute_with(|| {925 default_limits();926 927 let collection_id = create_test_collection(&CollectionMode::NFT, 1);928 let origin1 = Origin::signed(1);929 let origin2 = Origin::signed(2);930931 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));932933 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 3));934 assert_ok!(TemplateModule::remove_from_white_list(935 origin2.clone(),936 collection_id,937 3938 ));939 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);940 });941}942943#[test]944fn nonprivileged_user_cannot_remove_address_from_white_list() {945 new_test_ext().execute_with(|| {946 default_limits();947 948 let collection_id = create_test_collection(&CollectionMode::NFT, 1);949 let origin1 = Origin::signed(1);950 let origin2 = Origin::signed(2);951952 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));953 assert_noop!(954 TemplateModule::remove_from_white_list(origin2.clone(), collection_id, 2),955 "You do not have permissions to modify this collection"956 );957 assert_eq!(TemplateModule::white_list(collection_id)[0], 2);958 });959}960961#[test]962fn nobody_can_remove_address_from_white_list_of_nonexisting_collection() {963 new_test_ext().execute_with(|| {964 default_limits();965 let origin1 = Origin::signed(1);966967 assert_noop!(968 TemplateModule::remove_from_white_list(origin1.clone(), 1, 2),969 "This collection does not exist"970 );971 });972}973974#[test]975fn nobody_can_remove_address_from_white_list_of_deleted_collection() {976 new_test_ext().execute_with(|| {977 default_limits();978 979 let collection_id = create_test_collection(&CollectionMode::NFT, 1);980 let origin1 = Origin::signed(1);981 let origin2 = Origin::signed(2);982983 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));984 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), collection_id));985 assert_noop!(986 TemplateModule::remove_from_white_list(origin2.clone(), collection_id, 2),987 "This collection does not exist"988 );989 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);990 });991}992993// If address is already removed from white list, nothing happens994#[test]995fn address_is_already_removed_from_white_list() {996 new_test_ext().execute_with(|| {997 default_limits();998 999 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1000 let origin1 = Origin::signed(1);10011002 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));1003 assert_ok!(TemplateModule::remove_from_white_list(1004 origin1.clone(),1005 collection_id,1006 21007 ));1008 assert_ok!(TemplateModule::remove_from_white_list(1009 origin1.clone(),1010 collection_id,1011 21012 ));1013 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);1014 });1015}10161017// If Public Access mode is set to WhiteList, tokens can’t be transferred from a non-whitelisted address with transfer or transferFrom (2 tests)1018#[test]1019fn white_list_test_1() {1020 new_test_ext().execute_with(|| {1021 default_limits();1022 1023 let collection_id = create_test_collection(&CollectionMode::NFT, 1);10241025 let origin1 = Origin::signed(1);1026 1027 let data = default_nft_data();1028 create_test_item(collection_id, &data.into());10291030 assert_ok!(TemplateModule::set_public_access_mode(1031 origin1.clone(),1032 collection_id,1033 AccessMode::WhiteList1034 ));1035 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));10361037 assert_noop!(1038 TemplateModule::transfer(origin1.clone(), 3, 1, 1, 1),1039 "Address is not in white list"1040 );1041 });1042}10431044#[test]1045fn white_list_test_2() {1046 new_test_ext().execute_with(|| {1047 default_limits();1048 1049 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1050 let origin1 = Origin::signed(1);1051 1052 let data = default_nft_data();1053 create_test_item(collection_id, &data.into());10541055 assert_ok!(TemplateModule::set_public_access_mode(1056 origin1.clone(),1057 collection_id,1058 AccessMode::WhiteList1059 ));1060 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));1061 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));10621063 // do approve1064 assert_ok!(TemplateModule::approve(origin1.clone(), 1, 1, 1));1065 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);10661067 assert_ok!(TemplateModule::remove_from_white_list(1068 origin1.clone(),1069 1,1070 11071 ));10721073 assert_noop!(1074 TemplateModule::transfer_from(origin1.clone(), 1, 3, 1, 1, 1),1075 "Address is not in white list"1076 );1077 });1078}10791080// If Public Access mode is set to WhiteList, tokens can’t be transferred to a non-whitelisted address with transfer or transferFrom (2 tests)1081#[test]1082fn white_list_test_3() {1083 new_test_ext().execute_with(|| {1084 default_limits();1085 1086 let collection_id = create_test_collection(&CollectionMode::NFT, 1);10871088 let origin1 = Origin::signed(1);1089 1090 let data = default_nft_data();1091 create_test_item(collection_id, &data.into());10921093 assert_ok!(TemplateModule::set_public_access_mode(1094 origin1.clone(),1095 collection_id,1096 AccessMode::WhiteList1097 ));1098 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));10991100 assert_noop!(1101 TemplateModule::transfer(origin1.clone(), 3, 1, 1, 1),1102 "Address is not in white list"1103 );1104 });1105}11061107#[test]1108fn white_list_test_4() {1109 new_test_ext().execute_with(|| {1110 default_limits();1111 1112 let collection_id = create_test_collection(&CollectionMode::NFT, 1);11131114 let origin1 = Origin::signed(1);11151116 let data = default_nft_data();1117 create_test_item(collection_id, &data.into());11181119 assert_ok!(TemplateModule::set_public_access_mode(1120 origin1.clone(),1121 collection_id,1122 AccessMode::WhiteList1123 ));1124 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 1));1125 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));11261127 // do approve1128 assert_ok!(TemplateModule::approve(origin1.clone(), 1, 1, 1));1129 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);11301131 assert_ok!(TemplateModule::remove_from_white_list(1132 origin1.clone(),1133 collection_id,1134 21135 ));11361137 assert_noop!(1138 TemplateModule::transfer_from(origin1.clone(), 1, 3, 1, 1, 1),1139 "Address is not in white list"1140 );1141 });1142}11431144// If Public Access mode is set to WhiteList, tokens can’t be destroyed by a non-whitelisted address (even if it owned them before enabling WhiteList mode)1145#[test]1146fn white_list_test_5() {1147 new_test_ext().execute_with(|| {1148 default_limits();1149 1150 let collection_id = create_test_collection(&CollectionMode::NFT, 1);11511152 let origin1 = Origin::signed(1);11531154 let data = default_nft_data();1155 create_test_item(collection_id, &data.into());11561157 assert_ok!(TemplateModule::set_public_access_mode(1158 origin1.clone(),1159 collection_id,1160 AccessMode::WhiteList1161 ));1162 assert_noop!(1163 TemplateModule::burn_item(origin1.clone(), 1, 1),1164 "Address is not in white list"1165 );1166 });1167}11681169// If Public Access mode is set to WhiteList, oken transfers can’t be Approved by a non-whitelisted address (see Approve method).1170#[test]1171fn white_list_test_6() {1172 new_test_ext().execute_with(|| {1173 default_limits();1174 1175 let collection_id = create_test_collection(&CollectionMode::NFT, 1);11761177 let origin1 = Origin::signed(1);1178 assert_ok!(TemplateModule::set_public_access_mode(1179 origin1.clone(),1180 collection_id,1181 AccessMode::WhiteList1182 ));11831184 // do approve1185 assert_noop!(1186 TemplateModule::approve(origin1.clone(), 1, 1, 1),1187 "Address is not in white list"1188 );1189 });1190}11911192// If Public Access mode is set to WhiteList, tokens can be transferred from a whitelisted address with transfer or transferFrom (2 tests) and1193// tokens can be transferred from a whitelisted address with transfer or transferFrom (2 tests)1194#[test]1195fn white_list_test_7() {1196 new_test_ext().execute_with(|| {1197 default_limits();1198 1199 let collection_id = create_test_collection(&CollectionMode::NFT, 1);12001201 let data = default_nft_data();1202 create_test_item(collection_id, &data.into());1203 1204 let origin1 = Origin::signed(1);12051206 assert_ok!(TemplateModule::set_public_access_mode(1207 origin1.clone(),1208 collection_id,1209 AccessMode::WhiteList1210 ));1211 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 1));1212 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));12131214 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1));1215 });1216}12171218#[test]1219fn white_list_test_8() {1220 new_test_ext().execute_with(|| {1221 default_limits();1222 1223 let collection_id = create_test_collection(&CollectionMode::NFT, 1);12241225 let data = default_nft_data();1226 create_test_item(collection_id, &data.into());1227 1228 let origin1 = Origin::signed(1);12291230 assert_ok!(TemplateModule::set_public_access_mode(1231 origin1.clone(),1232 collection_id,1233 AccessMode::WhiteList1234 ));1235 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 1));1236 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));12371238 // do approve1239 assert_ok!(TemplateModule::approve(origin1.clone(), 1, 1, 1));1240 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);12411242 assert_ok!(TemplateModule::transfer_from(1243 origin1.clone(),1244 1,1245 2,1246 1,1247 1,1248 11249 ));1250 });1251}12521253// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens can be created by owner.1254#[test]1255fn white_list_test_9() {1256 new_test_ext().execute_with(|| {1257 default_limits();1258 1259 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1260 let origin1 = Origin::signed(1);12611262 assert_ok!(TemplateModule::set_public_access_mode(1263 origin1.clone(),1264 collection_id,1265 AccessMode::WhiteList1266 ));1267 assert_ok!(TemplateModule::set_mint_permission(1268 origin1.clone(),1269 collection_id,1270 false1271 ));12721273 let data = default_nft_data();1274 create_test_item(collection_id, &data.into());1275 });1276}12771278// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens can be created by admin.1279#[test]1280fn white_list_test_10() {1281 new_test_ext().execute_with(|| {1282 default_limits();1283 1284 let collection_id = create_test_collection(&CollectionMode::NFT, 1);12851286 let origin1 = Origin::signed(1);1287 let origin2 = Origin::signed(2);12881289 assert_ok!(TemplateModule::set_public_access_mode(1290 origin1.clone(),1291 collection_id,1292 AccessMode::WhiteList1293 ));1294 assert_ok!(TemplateModule::set_mint_permission(1295 origin1.clone(),1296 collection_id,1297 false1298 ));12991300 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));13011302 assert_ok!(TemplateModule::create_item(1303 origin2.clone(),1304 collection_id,1305 2,1306 default_nft_data().into()1307 ));1308 });1309}13101311// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens cannot be created by non-privileged and white listed address.1312#[test]1313fn white_list_test_11() {1314 new_test_ext().execute_with(|| {1315 default_limits();1316 1317 let collection_id = create_test_collection(&CollectionMode::NFT, 1);13181319 let origin1 = Origin::signed(1);1320 let origin2 = Origin::signed(2);13211322 assert_ok!(TemplateModule::set_public_access_mode(1323 origin1.clone(),1324 collection_id,1325 AccessMode::WhiteList1326 ));1327 assert_ok!(TemplateModule::set_mint_permission(1328 origin1.clone(),1329 collection_id,1330 false1331 ));1332 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));13331334 assert_noop!(1335 TemplateModule::create_item(origin2.clone(), 1, 2, default_nft_data().into()),1336 "Public minting is not allowed for this collection."1337 );1338 });1339}13401341// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens cannot be created by non-privileged and non-white listed address.1342#[test]1343fn white_list_test_12() {1344 new_test_ext().execute_with(|| {1345 default_limits();1346 1347 let collection_id = create_test_collection(&CollectionMode::NFT, 1);13481349 let origin1 = Origin::signed(1);1350 let origin2 = Origin::signed(2);13511352 assert_ok!(TemplateModule::set_public_access_mode(1353 origin1.clone(),1354 collection_id,1355 AccessMode::WhiteList1356 ));1357 assert_ok!(TemplateModule::set_mint_permission(1358 origin1.clone(),1359 collection_id,1360 false1361 ));13621363 assert_noop!(1364 TemplateModule::create_item(origin2.clone(), 1, 2, default_nft_data().into()),1365 "Public minting is not allowed for this collection."1366 );1367 });1368}13691370// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens can be created by owner.1371#[test]1372fn white_list_test_13() {1373 new_test_ext().execute_with(|| {1374 default_limits();1375 1376 let collection_id = create_test_collection(&CollectionMode::NFT, 1);13771378 let origin1 = Origin::signed(1);13791380 assert_ok!(TemplateModule::set_public_access_mode(1381 origin1.clone(),1382 collection_id,1383 AccessMode::WhiteList1384 ));1385 assert_ok!(TemplateModule::set_mint_permission(1386 origin1.clone(),1387 collection_id,1388 true1389 ));13901391 let data = default_nft_data();1392 create_test_item(collection_id, &data.into());1393 });1394}13951396// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens can be created by admin.1397#[test]1398fn white_list_test_14() {1399 new_test_ext().execute_with(|| {1400 default_limits();1401 1402 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14031404 let origin1 = Origin::signed(1);1405 let origin2 = Origin::signed(2);14061407 assert_ok!(TemplateModule::set_public_access_mode(1408 origin1.clone(),1409 collection_id,1410 AccessMode::WhiteList1411 ));1412 assert_ok!(TemplateModule::set_mint_permission(1413 origin1.clone(),1414 collection_id,1415 true1416 ));14171418 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));14191420 assert_ok!(TemplateModule::create_item(1421 origin2.clone(),1422 1,1423 2,1424 default_nft_data().into()1425 ));1426 });1427}14281429// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens cannot be created by non-privileged and non-white listed address.1430#[test]1431fn white_list_test_15() {1432 new_test_ext().execute_with(|| {1433 default_limits();1434 1435 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14361437 let origin1 = Origin::signed(1);1438 let origin2 = Origin::signed(2);14391440 assert_ok!(TemplateModule::set_public_access_mode(1441 origin1.clone(),1442 collection_id,1443 AccessMode::WhiteList1444 ));1445 assert_ok!(TemplateModule::set_mint_permission(1446 origin1.clone(),1447 collection_id,1448 true1449 ));14501451 assert_noop!(1452 TemplateModule::create_item(origin2.clone(), 1, 2, default_nft_data().into()),1453 "Address is not in white list"1454 );1455 });1456}14571458// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens can be created by non-privileged and white listed address.1459#[test]1460fn white_list_test_16() {1461 new_test_ext().execute_with(|| {1462 default_limits();1463 1464 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14651466 let origin1 = Origin::signed(1);1467 let origin2 = Origin::signed(2);14681469 assert_ok!(TemplateModule::set_public_access_mode(1470 origin1.clone(),1471 collection_id,1472 AccessMode::WhiteList1473 ));1474 assert_ok!(TemplateModule::set_mint_permission(1475 origin1.clone(),1476 collection_id,1477 true1478 ));1479 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));14801481 assert_ok!(TemplateModule::create_item(1482 origin2.clone(),1483 1,1484 2,1485 default_nft_data().into()1486 ));1487 });1488}14891490// Total number of collections. Positive test1491#[test]1492fn total_number_collections_bound() {1493 new_test_ext().execute_with(|| {1494 default_limits();1495 1496 create_test_collection(&CollectionMode::NFT, 1);1497 });1498}14991500// Total number of collections. Negotive test1501#[test]1502fn total_number_collections_bound_neg() {1503 new_test_ext().execute_with(|| {1504 default_limits();15051506 let origin1 = Origin::signed(1);15071508 for i in 0..default_collection_numbers_limit() {1509 create_test_collection(&CollectionMode::NFT, i + 1);1510 }15111512 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();1513 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();1514 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();15151516 // 11-th collection in chain. Expects error1517 assert_noop!(TemplateModule::create_collection(1518 origin1.clone(),1519 col_name1.clone(),1520 col_desc1.clone(),1521 token_prefix1.clone(),1522 CollectionMode::NFT1523 ), "Total collections bound exceeded");1524 });1525}15261527// Owned tokens by a single address. Positive test1528#[test]1529fn owned_tokens_bound() {1530 new_test_ext().execute_with(|| {1531 default_limits();1532 1533 let collection_id = create_test_collection(&CollectionMode::NFT, 1);15341535 let data = default_nft_data();1536 create_test_item(collection_id, &data.clone().into());1537 create_test_item(collection_id, &data.into());1538 });1539}15401541// Owned tokens by a single address. Negotive test1542#[test]1543fn owned_tokens_bound_neg() {1544 new_test_ext().execute_with(|| {1545 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1546 collection_numbers_limit: 10,1547 account_token_ownership_limit: 1,1548 collections_admins_limit: 5,1549 custom_data_limit: 2048,1550 nft_sponsor_transfer_timeout: 15,1551 fungible_sponsor_transfer_timeout: 15,1552 refungible_sponsor_transfer_timeout: 15, 1553 }));1554 1555 let collection_id = create_test_collection(&CollectionMode::NFT, 1);15561557 let origin1 = Origin::signed(1);1558 let data = default_nft_data();1559 create_test_item(collection_id, &data.clone().into());15601561 assert_noop!(TemplateModule::create_item(1562 origin1.clone(),1563 1,1564 1,1565 data.into()1566 ), "Owned tokens by a single address bound exceeded");1567 });1568}15691570// Number of collection admins. Positive test1571#[test]1572fn collection_admins_bound() {1573 new_test_ext().execute_with(|| {1574 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1575 collection_numbers_limit: 10,1576 account_token_ownership_limit: 10,1577 collections_admins_limit: 2,1578 custom_data_limit: 2048,1579 nft_sponsor_transfer_timeout: 15,1580 fungible_sponsor_transfer_timeout: 15,1581 refungible_sponsor_transfer_timeout: 15, 1582 }));1583 1584 let collection_id = create_test_collection(&CollectionMode::NFT, 1);15851586 let origin1 = Origin::signed(1);1587 1588 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));1589 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 3));1590 });1591}15921593// Number of collection admins. Negotive test1594#[test]1595fn collection_admins_bound_neg() {1596 new_test_ext().execute_with(|| {1597 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1598 collection_numbers_limit: 10,1599 account_token_ownership_limit: 1,1600 collections_admins_limit: 1,1601 custom_data_limit: 2048,1602 nft_sponsor_transfer_timeout: 15,1603 fungible_sponsor_transfer_timeout: 15,1604 refungible_sponsor_transfer_timeout: 15, 1605 }));1606 1607 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16081609 let origin1 = Origin::signed(1);16101611 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));1612 assert_noop!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 3), "Number of collection admins bound exceeded");1613 });1614}16151616// NFT custom data size. Negative test const_data.1617#[test]1618fn custom_data_size_nft_const_data_bound_neg() {1619 new_test_ext().execute_with(|| {1620 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1621 collection_numbers_limit: 10,1622 account_token_ownership_limit: 10,1623 collections_admins_limit: 5,1624 custom_data_limit: 2,1625 nft_sponsor_transfer_timeout: 15,1626 fungible_sponsor_transfer_timeout: 15,1627 refungible_sponsor_transfer_timeout: 15, 1628 }));1629 1630 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16311632 let origin1 = Origin::signed(1);1633 let too_big_const_data = CreateItemData::NFT(CreateNftData{1634 const_data: vec![1, 2, 3, 4],1635 variable_data: vec![]1636 });16371638 assert_noop!(TemplateModule::create_item(1639 origin1.clone(),1640 collection_id,1641 1,1642 too_big_const_data1643 ), "const_data exceeded data limit.");1644 });1645}16461647// NFT custom data size. Negative test variable_data.1648#[test]1649fn custom_data_size_nft_variable_data_bound_neg() {1650 new_test_ext().execute_with(|| {1651 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1652 collection_numbers_limit: 10,1653 account_token_ownership_limit: 10,1654 collections_admins_limit: 5,1655 custom_data_limit: 2,1656 nft_sponsor_transfer_timeout: 15,1657 fungible_sponsor_transfer_timeout: 15,1658 refungible_sponsor_transfer_timeout: 15, 1659 }));16601661 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16621663 let origin1 = Origin::signed(1);1664 let too_big_const_data = CreateItemData::NFT(CreateNftData{1665 const_data: vec![],1666 variable_data: vec![1, 2, 3, 4]1667 });16681669 assert_noop!(TemplateModule::create_item(1670 origin1.clone(),1671 collection_id,1672 1,1673 too_big_const_data1674 ), "variable_data exceeded data limit.");1675 });1676}16771678// Re fungible custom data size. Negative test const_data.1679#[test]1680fn custom_data_size_re_fungible_const_data_bound_neg() {1681 new_test_ext().execute_with(|| {1682 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1683 collection_numbers_limit: 10,1684 account_token_ownership_limit: 10,1685 collections_admins_limit: 5,1686 custom_data_limit: 2,1687 nft_sponsor_transfer_timeout: 15,1688 fungible_sponsor_transfer_timeout: 15,1689 refungible_sponsor_transfer_timeout: 15, 1690 }));16911692 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16931694 let origin1 = Origin::signed(1);1695 let too_big_const_data = CreateItemData::NFT(CreateNftData{1696 const_data: vec![1, 2, 3, 4],1697 variable_data: vec![]1698 });16991700 assert_noop!(TemplateModule::create_item(1701 origin1.clone(),1702 collection_id,1703 1,1704 too_big_const_data1705 ), "const_data exceeded data limit.");1706 });1707}17081709// Re fungible custom data size. Negative test variable_data.1710#[test]1711fn custom_data_size_re_fungible_variable_data_bound_neg() {1712 new_test_ext().execute_with(|| {1713 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1714 collection_numbers_limit: 10,1715 account_token_ownership_limit: 10,1716 collections_admins_limit: 5,1717 custom_data_limit: 2,1718 nft_sponsor_transfer_timeout: 15,1719 fungible_sponsor_transfer_timeout: 15,1720 refungible_sponsor_transfer_timeout: 15, 1721 }));17221723 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17241725 let origin1 = Origin::signed(1);1726 let too_big_const_data = CreateItemData::NFT(CreateNftData{1727 const_data: vec![],1728 variable_data: vec![1, 2, 3, 4]1729 });17301731 assert_noop!(TemplateModule::create_item(1732 origin1.clone(),1733 collection_id,1734 1,1735 too_big_const_data1736 ), "variable_data exceeded data limit.");1737 });1738}1739// #endregion17401741#[test]1742fn set_const_on_chain_schema() {1743 new_test_ext().execute_with(|| {1744 default_limits();17451746 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17471748 let origin1 = Origin::signed(1);1749 assert_ok!(TemplateModule::set_const_on_chain_schema(origin1, collection_id, b"test const on chain schema".to_vec()));17501751 assert_eq!(TemplateModule::collection(collection_id).const_on_chain_schema, b"test const on chain schema".to_vec());1752 assert_eq!(TemplateModule::collection(collection_id).variable_on_chain_schema, b"".to_vec());1753 });1754}17551756#[test]1757fn set_variable_on_chain_schema() {1758 new_test_ext().execute_with(|| {1759 default_limits();1760 1761 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17621763 let origin1 = Origin::signed(1);1764 assert_ok!(TemplateModule::set_variable_on_chain_schema(origin1, collection_id, b"test variable on chain schema".to_vec()));17651766 assert_eq!(TemplateModule::collection(collection_id).const_on_chain_schema, b"".to_vec());1767 assert_eq!(TemplateModule::collection(collection_id).variable_on_chain_schema, b"test variable on chain schema".to_vec());1768 });1769}17701771#[test]1772fn set_variable_meta_data_on_nft_token_stores_variable_meta_data() {1773 new_test_ext().execute_with(|| {1774 default_limits();17751776 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17771778 let origin1 = Origin::signed(1);1779 1780 let data = default_nft_data();1781 create_test_item(1, &data.into());1782 1783 let variable_data = b"test set_variable_meta_data method.".to_vec();1784 assert_ok!(TemplateModule::set_variable_meta_data(origin1, collection_id, 1, variable_data.clone()));17851786 assert_eq!(TemplateModule::nft_item_id(collection_id, 1).variable_data, variable_data);1787 });1788}17891790#[test]1791fn set_variable_meta_data_on_re_fungible_token_stores_variable_meta_data() {1792 new_test_ext().execute_with(|| {1793 default_limits();17941795 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);17961797 let origin1 = Origin::signed(1);17981799 let data = default_re_fungible_data();1800 create_test_item(1, &data.into());18011802 let variable_data = b"test set_variable_meta_data method.".to_vec();1803 assert_ok!(TemplateModule::set_variable_meta_data(origin1, collection_id, 1, variable_data.clone()));18041805 assert_eq!(TemplateModule::refungible_item_id(collection_id, 1).variable_data, variable_data);1806 });1807}180818091810#[test]1811fn set_variable_meta_data_on_fungible_token_fails() {1812 new_test_ext().execute_with(|| {1813 default_limits();18141815 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);18161817 let origin1 = Origin::signed(1);18181819 let data = default_fungible_data();1820 create_test_item(1, &data.into());18211822 let variable_data = b"test set_variable_meta_data method.".to_vec();1823 assert_noop!(TemplateModule::set_variable_meta_data(origin1, collection_id, 1, variable_data.clone()), "Can't store metadata in fungible tokens.");1824 });1825}1// Tests to be written here2use crate::mock::*;3use crate::{AccessMode, ApprovePermissions, CollectionMode, Ownership, ChainLimits, CreateItemData, CreateNftData, CreateFungibleData, CreateReFungibleData};4use frame_support::{assert_noop, assert_ok};5use frame_system::{ RawOrigin };67fn default_collection_numbers_limit() -> u64 {8 109}1011fn default_limits() {12 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 13 collection_numbers_limit: default_collection_numbers_limit(),14 account_token_ownership_limit: 10,15 collections_admins_limit: 5,16 custom_data_limit: 2048,17 nft_sponsor_transfer_timeout: 15,18 fungible_sponsor_transfer_timeout: 15,19 refungible_sponsor_transfer_timeout: 15, 20 }));21}2223fn default_nft_data() -> CreateNftData {24 CreateNftData { const_data: vec![1, 2, 3], variable_data: vec![3, 2, 1] }25}2627fn default_fungible_data () -> CreateFungibleData {28 CreateFungibleData { }29}3031fn default_re_fungible_data () -> CreateReFungibleData {32 CreateReFungibleData { const_data: vec![1, 2, 3], variable_data: vec![3, 2, 1] }33}3435fn create_test_collection_for_owner(mode: &CollectionMode, owner: u64, id: u64) -> u64 {36 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();37 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();38 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();3940 let origin1 = Origin::signed(owner);41 assert_ok!(TemplateModule::create_collection(42 origin1.clone(),43 col_name1.clone(),44 col_desc1.clone(),45 token_prefix1.clone(),46 mode.clone()47 ));4849 let saved_col_name: Vec<u16> = "Test1\0\0".encode_utf16().collect::<Vec<u16>>();50 let saved_description: Vec<u16> = "TestDescription1\0\0".encode_utf16().collect::<Vec<u16>>();51 let saved_prefix: Vec<u8> = b"token_prefix1\0\0".to_vec();52 assert_eq!(TemplateModule::collection(id).owner, owner);53 assert_eq!(TemplateModule::collection(id).name, saved_col_name);54 assert_eq!(TemplateModule::collection(id).mode, *mode);55 assert_eq!(TemplateModule::collection(id).description, saved_description);56 assert_eq!(TemplateModule::collection(id).token_prefix, saved_prefix);57 id58}5960fn create_test_collection(mode: &CollectionMode, id: u64) -> u64 {61 create_test_collection_for_owner(&mode, 1, id)62}6364fn create_test_item(collection_id: u64, data: &CreateItemData) {65 let origin1 = Origin::signed(1);66 assert_ok!(TemplateModule::create_item(67 origin1.clone(),68 collection_id,69 1,70 data.clone()71 ));7273}7475// Use cases tests region76// #region77#[test]78fn create_nft_item() {79 new_test_ext().execute_with(|| {80 default_limits();81 let collection_id = create_test_collection(&CollectionMode::NFT, 1);82 83 let data = default_nft_data();84 create_test_item(collection_id, &data.clone().into());85 assert_eq!(TemplateModule::nft_item_id(collection_id, 1).const_data, data.const_data);86 assert_eq!(TemplateModule::nft_item_id(collection_id, 1).variable_data, data.variable_data);87 });88}8990// Use cases tests region91// #region92#[test]93fn create_nft_multiple_items() {94 new_test_ext().execute_with(|| {95 default_limits();96 97 create_test_collection(&CollectionMode::NFT, 1);9899 let origin1 = Origin::signed(1);100101 let items_data = vec![default_nft_data(), default_nft_data(), default_nft_data()];102103 assert_ok!(TemplateModule::create_multiple_items(104 origin1.clone(),105 1,106 1,107 items_data.clone().into_iter().map(|d| { d.into() }).collect()108 ));109 for (index, data) in items_data.iter().enumerate() {110 assert_eq!(TemplateModule::nft_item_id(1, (index + 1) as u64).const_data.to_vec(), data.const_data);111 assert_eq!(TemplateModule::nft_item_id(1, (index + 1) as u64).variable_data.to_vec(), data.variable_data);112 }113 });114}115116#[test]117fn create_refungible_item() {118 new_test_ext().execute_with(|| {119 default_limits();120 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);121122 let data = default_re_fungible_data();123 create_test_item(collection_id, &data.clone().into());124 assert_eq!(125 TemplateModule::refungible_item_id(collection_id, 1).const_data,126 data.const_data127 );128 assert_eq!(129 TemplateModule::refungible_item_id(collection_id, 1).variable_data,130 data.variable_data131 );132 assert_eq!(133 TemplateModule::refungible_item_id(collection_id, 1).owner[0],134 Ownership {135 owner: 1,136 fraction: 1000137 }138 );139 });140}141142#[test]143fn create_multiple_refungible_items() {144 new_test_ext().execute_with(|| {145 default_limits();146 147 create_test_collection(&CollectionMode::ReFungible(3), 1);148149 let origin1 = Origin::signed(1);150151 let items_data = vec![default_re_fungible_data(), default_re_fungible_data(), default_re_fungible_data()];152153 assert_ok!(TemplateModule::create_multiple_items(154 origin1.clone(),155 1,156 1,157 items_data.clone().into_iter().map(|d| { d.into() }).collect()158 ));159 for (index, data) in items_data.iter().enumerate() {160161 let item = TemplateModule::refungible_item_id(1, (index + 1) as u64);162 assert_eq!(item.const_data.to_vec(), data.const_data);163 assert_eq!(item.variable_data.to_vec(), data.variable_data);164 assert_eq!(165 item.owner[0],166 Ownership {167 owner: 1,168 fraction: 1000169 }170 );171 }172 });173}174175#[test]176fn create_fungible_item() {177 new_test_ext().execute_with(|| {178 default_limits();179 180 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);181182 let data = default_fungible_data();183 create_test_item(collection_id, &data.into());184185 assert_eq!(TemplateModule::fungible_item_id(collection_id, 1).owner, 1);186 });187}188189#[test]190fn create_multiple_fungible_items() {191 new_test_ext().execute_with(|| {192 default_limits();193194 create_test_collection(&CollectionMode::Fungible(3), 1);195196 let origin1 = Origin::signed(1);197198 let items_data = vec![default_fungible_data(), default_fungible_data(), default_fungible_data()];199200 assert_ok!(TemplateModule::create_multiple_items(201 origin1.clone(),202 1,203 1,204 items_data.clone().into_iter().map(|d| { d.into() }).collect()205 ));206 207 for (index, _) in items_data.iter().enumerate() {208 assert_eq!(TemplateModule::fungible_item_id(1, (index + 1) as u64).owner, 1);209 }210 assert_eq!(TemplateModule::balance_count(1, 1), 3000);211 assert_eq!(TemplateModule::address_tokens(1, 1), [1, 2, 3]);212 });213}214215#[test]216fn transfer_fungible_item() {217 new_test_ext().execute_with(|| {218 default_limits();219 220 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);221222 let origin1 = Origin::signed(1);223 let origin2 = Origin::signed(2);224225 let data = default_fungible_data();226 create_test_item(collection_id, &data.into());227228 assert_eq!(TemplateModule::fungible_item_id(1, 1).owner, 1);229 assert_eq!(TemplateModule::balance_count(1, 1), 1000);230 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);231232 // change owner scenario233 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1000));234 assert_eq!(TemplateModule::fungible_item_id(1, 1).owner, 2);235 assert_eq!(TemplateModule::fungible_item_id(1, 1).value, 1000);236 assert_eq!(TemplateModule::balance_count(1, 1), 0);237 assert_eq!(TemplateModule::balance_count(1, 2), 1000);238 // assert_eq!(TemplateModule::address_tokens(1, 1), []);239 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);240241 // split item scenario242 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 500));243 assert_eq!(TemplateModule::fungible_item_id(1, 1).owner, 2);244 assert_eq!(TemplateModule::fungible_item_id(1, 2).owner, 3);245 assert_eq!(TemplateModule::balance_count(1, 2), 500);246 assert_eq!(TemplateModule::balance_count(1, 3), 500);247 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);248 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);249250 // split item and new owner has account scenario251 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 200));252 assert_eq!(TemplateModule::fungible_item_id(1, 1).value, 300);253 assert_eq!(TemplateModule::fungible_item_id(1, 2).value, 700);254 assert_eq!(TemplateModule::balance_count(1, 2), 300);255 assert_eq!(TemplateModule::balance_count(1, 3), 700);256 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);257 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);258 });259}260261#[test]262fn transfer_refungible_item() {263 new_test_ext().execute_with(|| {264 default_limits();265 266 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);267268 let data = default_re_fungible_data();269 create_test_item(collection_id, &data.clone().into());270271 let origin1 = Origin::signed(1);272 let origin2 = Origin::signed(2);273 assert_eq!(274 TemplateModule::refungible_item_id(collection_id, 1).const_data,275 data.const_data276 );277 assert_eq!(278 TemplateModule::refungible_item_id(collection_id, 1).variable_data,279 data.variable_data280 );281 assert_eq!(282 TemplateModule::refungible_item_id(collection_id, 1).owner[0],283 Ownership {284 owner: 1,285 fraction: 1000286 }287 );288 assert_eq!(TemplateModule::balance_count(1, 1), 1000);289 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);290291 // change owner scenario292 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1000));293 assert_eq!(294 TemplateModule::refungible_item_id(1, 1).owner[0],295 Ownership {296 owner: 2,297 fraction: 1000298 }299 );300 assert_eq!(TemplateModule::balance_count(1, 1), 0);301 assert_eq!(TemplateModule::balance_count(1, 2), 1000);302 // assert_eq!(TemplateModule::address_tokens(1, 1), []);303 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);304305 // split item scenario306 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 500));307 assert_eq!(308 TemplateModule::refungible_item_id(1, 1).owner[0],309 Ownership {310 owner: 2,311 fraction: 500312 }313 );314 assert_eq!(315 TemplateModule::refungible_item_id(1, 1).owner[1],316 Ownership {317 owner: 3,318 fraction: 500319 }320 );321 assert_eq!(TemplateModule::balance_count(1, 2), 500);322 assert_eq!(TemplateModule::balance_count(1, 3), 500);323 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);324 assert_eq!(TemplateModule::address_tokens(1, 3), [1]);325326 // split item and new owner has account scenario327 assert_ok!(TemplateModule::transfer(origin2.clone(), 3, 1, 1, 200));328 assert_eq!(329 TemplateModule::refungible_item_id(1, 1).owner[0],330 Ownership {331 owner: 2,332 fraction: 300333 }334 );335 assert_eq!(336 TemplateModule::refungible_item_id(1, 1).owner[1],337 Ownership {338 owner: 3,339 fraction: 700340 }341 );342 assert_eq!(TemplateModule::balance_count(1, 2), 300);343 assert_eq!(TemplateModule::balance_count(1, 3), 700);344 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);345 assert_eq!(TemplateModule::address_tokens(1, 3), [1]);346 });347}348349#[test]350fn transfer_nft_item() {351 new_test_ext().execute_with(|| {352 default_limits();353 354 let collection_id = create_test_collection(&CollectionMode::NFT, 1);355356 let data = default_nft_data();357 create_test_item(collection_id, &data.into());358 assert_eq!(TemplateModule::balance_count(1, 1), 1);359 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);360361 let origin1 = Origin::signed(1);362 // default scenario363 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1000));364 assert_eq!(TemplateModule::nft_item_id(1, 1).owner, 2);365 assert_eq!(TemplateModule::balance_count(1, 1), 0);366 assert_eq!(TemplateModule::balance_count(1, 2), 1);367 // assert_eq!(TemplateModule::address_tokens(1, 1), []);368 assert_eq!(TemplateModule::address_tokens(1, 2), [1]);369 });370}371372#[test]373fn nft_approve_and_transfer_from() {374 new_test_ext().execute_with(|| {375 default_limits();376 377 let collection_id = create_test_collection(&CollectionMode::NFT, 1);378379 let data = default_nft_data();380 create_test_item(collection_id, &data.into());381382 let origin1 = Origin::signed(1);383 let origin2 = Origin::signed(2);384385 assert_eq!(TemplateModule::balance_count(1, 1), 1);386 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);387388 // neg transfer389 assert_noop!(TemplateModule::transfer_from(390 origin2.clone(),391 1,392 2,393 1,394 1,395 1), "Only item owner, collection owner and admins can modify items");396397 // do approve398 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));399 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);400 assert_eq!(401 TemplateModule::approved(1, (1, 1))[0],402 ApprovePermissions {403 approved: 2,404 amount: 100000000405 }406 );407408 assert_ok!(TemplateModule::transfer_from(409 origin2.clone(),410 1,411 2,412 1,413 1,414 1415 ));416 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 0);417 });418}419420#[test]421fn nft_approve_and_transfer_from_white_list() {422 new_test_ext().execute_with(|| {423 default_limits();424 425 let collection_id = create_test_collection(&CollectionMode::NFT, 1);426427 let origin1 = Origin::signed(1);428 let origin2 = Origin::signed(2);429430 let data = default_nft_data();431 create_test_item(collection_id, &data.clone().into());432433 assert_eq!(TemplateModule::nft_item_id(1, 1).const_data, data.const_data);434 assert_eq!(TemplateModule::balance_count(1, 1), 1);435 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);436437 assert_ok!(TemplateModule::set_mint_permission(438 origin1.clone(),439 1,440 true441 ));442 assert_ok!(TemplateModule::set_public_access_mode(443 origin1.clone(),444 1,445 AccessMode::WhiteList446 ));447 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));448 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));449 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));450451 // do approve452 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));453 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);454 assert_ok!(TemplateModule::approve(origin1.clone(), 3, 1, 1));455 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 2);456 assert_eq!(457 TemplateModule::approved(1, (1, 1))[0],458 ApprovePermissions {459 approved: 2,460 amount: 100000000461 }462 );463464 assert_ok!(TemplateModule::transfer_from(465 origin2.clone(),466 1,467 3,468 1,469 1,470 1471 ));472 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 0);473 });474}475476#[test]477fn refungible_approve_and_transfer_from() {478 new_test_ext().execute_with(|| {479 default_limits();480 481 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);482 483 let origin1 = Origin::signed(1);484 let origin2 = Origin::signed(2);485486 let data = default_re_fungible_data();487 create_test_item(collection_id, &data.into());488489 assert_eq!(TemplateModule::balance_count(1, 1), 1000);490 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);491492 assert_ok!(TemplateModule::set_mint_permission(493 origin1.clone(),494 1,495 true496 ));497 assert_ok!(TemplateModule::set_public_access_mode(498 origin1.clone(),499 1,500 AccessMode::WhiteList501 ));502 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));503 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));504 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));505506 // do approve507 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));508 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);509 assert_ok!(TemplateModule::approve(origin1.clone(), 3, 1, 1));510 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 2);511 assert_eq!(512 TemplateModule::approved(1, (1, 1))[0],513 ApprovePermissions {514 approved: 2,515 amount: 100000000516 }517 );518519 assert_ok!(TemplateModule::transfer_from(520 origin2.clone(),521 1,522 3,523 1,524 1,525 100526 ));527 assert_eq!(TemplateModule::balance_count(1, 1), 900);528 assert_eq!(TemplateModule::balance_count(1, 3), 100);529 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);530 assert_eq!(TemplateModule::address_tokens(1, 3), [1]);531532 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);533 assert_eq!(534 TemplateModule::approved(1, (1, 1))[0],535 ApprovePermissions {536 approved: 3,537 amount: 100000000538 }539 );540 });541}542543#[test]544fn fungible_approve_and_transfer_from() {545 new_test_ext().execute_with(|| {546 default_limits();547 548 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);549 550 let data = default_fungible_data();551 create_test_item(collection_id, &data.into());552553 let origin1 = Origin::signed(1);554 let origin2 = Origin::signed(2);555556 assert_eq!(TemplateModule::balance_count(1, 1), 1000);557 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);558559 assert_ok!(TemplateModule::set_mint_permission(560 origin1.clone(),561 1,562 true563 ));564 assert_ok!(TemplateModule::set_public_access_mode(565 origin1.clone(),566 1,567 AccessMode::WhiteList568 ));569 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));570 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));571 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));572573 // do approve574 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));575 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);576 assert_ok!(TemplateModule::approve(origin1.clone(), 3, 1, 1));577 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 2);578 assert_eq!(579 TemplateModule::approved(1, (1, 1))[0],580 ApprovePermissions {581 approved: 2,582 amount: 100000000583 }584 );585586 assert_ok!(TemplateModule::transfer_from(587 origin2.clone(),588 1,589 3,590 1,591 1,592 100593 ));594 assert_eq!(TemplateModule::balance_count(1, 1), 900);595 assert_eq!(TemplateModule::balance_count(1, 3), 100);596 assert_eq!(TemplateModule::address_tokens(1, 1), [1]);597 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);598599 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);600 assert_eq!(601 TemplateModule::approved(1, (1, 1))[0],602 ApprovePermissions {603 approved: 3,604 amount: 100000000605 }606 );607608 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));609 assert_ok!(TemplateModule::transfer_from(610 origin2.clone(),611 1,612 3,613 1,614 1,615 900616 ));617 assert_eq!(TemplateModule::balance_count(1, 1), 0);618 assert_eq!(TemplateModule::balance_count(1, 3), 1000);619 // assert_eq!(TemplateModule::address_tokens(1, 1), []);620 assert_eq!(TemplateModule::address_tokens(1, 3), [2]);621622 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 0);623 });624}625626#[test]627fn change_collection_owner() {628 new_test_ext().execute_with(|| {629 default_limits();630 631 let collection_id = create_test_collection(&CollectionMode::NFT, 1);632 633 let origin1 = Origin::signed(1);634 assert_ok!(TemplateModule::change_collection_owner(635 origin1.clone(),636 collection_id,637 2638 ));639 assert_eq!(TemplateModule::collection(collection_id).owner, 2);640 });641}642643#[test]644fn destroy_collection() {645 new_test_ext().execute_with(|| {646 default_limits();647 648 let collection_id = create_test_collection(&CollectionMode::NFT, 1);649 650 let origin1 = Origin::signed(1);651 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), collection_id));652 });653}654655#[test]656fn burn_nft_item() {657 new_test_ext().execute_with(|| {658 default_limits();659 660 let collection_id = create_test_collection(&CollectionMode::NFT, 1);661662 let origin1 = Origin::signed(1);663 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));664 665 let data = default_nft_data();666 create_test_item(collection_id, &data.into());667668 // check balance (collection with id = 1, user id = 1)669 assert_eq!(TemplateModule::balance_count(1, 1), 1);670671 // burn item672 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));673 assert_noop!(674 TemplateModule::burn_item(origin1.clone(), 1, 1),675 "Item does not exists"676 );677678 assert_eq!(TemplateModule::balance_count(1, 1), 0);679 });680}681682#[test]683fn burn_fungible_item() {684 new_test_ext().execute_with(|| {685 default_limits();686 687 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);688 689 let origin1 = Origin::signed(1);690 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));691 692 let data = default_fungible_data();693 create_test_item(collection_id, &data.into());694695 // check balance (collection with id = 1, user id = 1)696 assert_eq!(TemplateModule::balance_count(1, 1), 1000);697698 // burn item699 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));700 assert_noop!(701 TemplateModule::burn_item(origin1.clone(), 1, 1),702 "Item does not exists"703 );704705 assert_eq!(TemplateModule::balance_count(1, 1), 0);706 });707}708709#[test]710fn burn_refungible_item() {711 new_test_ext().execute_with(|| {712 default_limits();713 714 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);715 let origin1 = Origin::signed(1);716717 assert_ok!(TemplateModule::set_mint_permission(718 origin1.clone(),719 collection_id,720 true721 ));722 assert_ok!(TemplateModule::set_public_access_mode(723 origin1.clone(),724 collection_id,725 AccessMode::WhiteList726 ));727 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));728729 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));730 731 let data = default_re_fungible_data();732 create_test_item(collection_id, &data.into());733734 // check balance (collection with id = 1, user id = 2)735 assert_eq!(TemplateModule::balance_count(1, 1), 1000);736737 // burn item738 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));739 assert_noop!(740 TemplateModule::burn_item(origin1.clone(), 1, 1),741 "Item does not exists"742 );743744 assert_eq!(TemplateModule::balance_count(1, 1), 0);745 });746}747748#[test]749fn add_collection_admin() {750 new_test_ext().execute_with(|| {751 default_limits();752 753 let collection1_id = create_test_collection_for_owner(&CollectionMode::NFT, 1, 1);754 create_test_collection_for_owner(&CollectionMode::NFT, 2, 2);755 create_test_collection_for_owner(&CollectionMode::NFT, 3, 3);756 757 let origin1 = Origin::signed(1);758759 // collection admin760 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 2));761 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 3));762763 assert_eq!(TemplateModule::admin_list_collection(collection1_id).contains(&2), true);764 assert_eq!(TemplateModule::admin_list_collection(collection1_id).contains(&3), true);765 });766}767768#[test]769fn remove_collection_admin() {770 new_test_ext().execute_with(|| {771 default_limits();772 773 let collection1_id = create_test_collection_for_owner(&CollectionMode::NFT, 1, 1);774 create_test_collection_for_owner(&CollectionMode::NFT, 2, 2);775 create_test_collection_for_owner(&CollectionMode::NFT, 3, 3);776777 let origin1 = Origin::signed(1);778 let origin2 = Origin::signed(2);779780 // collection admin781 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 2));782 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection1_id, 3));783784 assert_eq!(TemplateModule::admin_list_collection(1).contains(&2), true);785 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), true);786787 // remove admin788 assert_ok!(TemplateModule::remove_collection_admin(789 origin2.clone(),790 1,791 3792 ));793 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), false);794 });795}796797#[test]798fn balance_of() {799 new_test_ext().execute_with(|| {800 default_limits();801 802 let nft_collection_id = create_test_collection(&CollectionMode::NFT, 1);803 let fungible_collection_id = create_test_collection(&CollectionMode::Fungible(3), 2);804 let re_fungible_collection_id = create_test_collection(&CollectionMode::ReFungible(3), 3);805 806 // check balance before807 assert_eq!(TemplateModule::balance_count(nft_collection_id, 1), 0);808 assert_eq!(TemplateModule::balance_count(fungible_collection_id, 1), 0);809 assert_eq!(TemplateModule::balance_count(re_fungible_collection_id, 1), 0);810811 let nft_data = default_nft_data();812 create_test_item(nft_collection_id, &nft_data.into());813 814 let fungible_data = default_fungible_data();815 create_test_item(fungible_collection_id, &fungible_data.into());816 817 let re_fungible_data = default_re_fungible_data();818 create_test_item(re_fungible_collection_id, &re_fungible_data.into());819820 // check balance (collection with id = 1, user id = 1)821 assert_eq!(TemplateModule::balance_count(nft_collection_id, 1), 1);822 assert_eq!(TemplateModule::balance_count(fungible_collection_id, 1), 1000);823 assert_eq!(TemplateModule::balance_count(re_fungible_collection_id, 1), 1000);824 assert_eq!(TemplateModule::nft_item_id(nft_collection_id, 1).owner, 1);825 assert_eq!(TemplateModule::fungible_item_id(fungible_collection_id, 1).owner, 1);826 assert_eq!(TemplateModule::refungible_item_id(re_fungible_collection_id, 1).owner[0].owner, 1);827 });828}829830#[test]831fn approve() {832 new_test_ext().execute_with(|| {833 default_limits();834 835 let collection_id = create_test_collection(&CollectionMode::NFT, 1);836 837 let data = default_nft_data();838 create_test_item(collection_id, &data.into());839840 let origin1 = Origin::signed(1);841 842 // approve843 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));844 assert_eq!(TemplateModule::approved(1, (1, 1))[0].approved, 2);845 });846}847848#[test]849fn transfer_from() {850 new_test_ext().execute_with(|| {851 default_limits();852 853 let collection_id = create_test_collection(&CollectionMode::NFT, 1);854 let origin1 = Origin::signed(1);855 let origin2 = Origin::signed(2);856857 let data = default_nft_data();858 create_test_item(collection_id, &data.into());859860 // approve861 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));862 assert_eq!(TemplateModule::approved(1, (1, 1))[0].approved, 2);863864 assert_ok!(TemplateModule::set_mint_permission(865 origin1.clone(),866 1,867 true868 ));869 assert_ok!(TemplateModule::set_public_access_mode(870 origin1.clone(),871 1,872 AccessMode::WhiteList873 ));874 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));875 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));876 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 3));877878 assert_ok!(TemplateModule::transfer_from(879 origin2.clone(),880 1,881 2,882 1,883 1,884 1885 ));886887 // after transfer888 assert_eq!(TemplateModule::balance_count(1, 1), 0);889 assert_eq!(TemplateModule::balance_count(1, 2), 1);890 });891}892893// #endregion894895// Coverage tests region896// #region897898#[test]899fn owner_can_add_address_to_white_list() {900 new_test_ext().execute_with(|| {901 default_limits();902 903 let collection_id = create_test_collection(&CollectionMode::NFT, 1);904905 let origin1 = Origin::signed(1);906 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));907 assert_eq!(TemplateModule::white_list(collection_id)[0], 2);908 });909}910911#[test]912fn admin_can_add_address_to_white_list() {913 new_test_ext().execute_with(|| {914 default_limits();915 916 let collection_id = create_test_collection(&CollectionMode::NFT, 1);917 let origin1 = Origin::signed(1);918 let origin2 = Origin::signed(2);919920 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));921 assert_ok!(TemplateModule::add_to_white_list(origin2.clone(), collection_id, 3));922 assert_eq!(TemplateModule::white_list(collection_id)[0], 3);923 });924}925926#[test]927fn nonprivileged_user_cannot_add_address_to_white_list() {928 new_test_ext().execute_with(|| {929 default_limits();930 931 let collection_id = create_test_collection(&CollectionMode::NFT, 1);932933 let origin2 = Origin::signed(2);934 assert_noop!(935 TemplateModule::add_to_white_list(origin2.clone(), collection_id, 3),936 "You do not have permissions to modify this collection"937 );938 });939}940941#[test]942fn nobody_can_add_address_to_white_list_of_nonexisting_collection() {943 new_test_ext().execute_with(|| {944 default_limits();945946 let origin1 = Origin::signed(1);947948 assert_noop!(949 TemplateModule::add_to_white_list(origin1.clone(), 1, 2),950 "This collection does not exist"951 );952 });953}954955#[test]956fn nobody_can_add_address_to_white_list_of_deleted_collection() {957 new_test_ext().execute_with(|| {958 default_limits();959 960 let collection_id = create_test_collection(&CollectionMode::NFT, 1);961962 let origin1 = Origin::signed(1);963 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), collection_id));964 assert_noop!(965 TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2),966 "This collection does not exist"967 );968 });969}970971// If address is already added to white list, nothing happens972#[test]973fn address_is_already_added_to_white_list() {974 new_test_ext().execute_with(|| {975 default_limits();976 977 let collection_id = create_test_collection(&CollectionMode::NFT, 1);978 let origin1 = Origin::signed(1);979 980 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));981 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));982 assert_eq!(TemplateModule::white_list(collection_id)[0], 2);983 assert_eq!(TemplateModule::white_list(collection_id).len(), 1);984 });985}986987#[test]988fn owner_can_remove_address_from_white_list() {989 new_test_ext().execute_with(|| {990 default_limits();991 992 let collection_id = create_test_collection(&CollectionMode::NFT, 1);993994 let origin1 = Origin::signed(1);995 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));996 assert_ok!(TemplateModule::remove_from_white_list(997 origin1.clone(),998 collection_id,999 21000 ));1001 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);1002 });1003}10041005#[test]1006fn admin_can_remove_address_from_white_list() {1007 new_test_ext().execute_with(|| {1008 default_limits();1009 1010 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1011 let origin1 = Origin::signed(1);1012 let origin2 = Origin::signed(2);10131014 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));10151016 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 3));1017 assert_ok!(TemplateModule::remove_from_white_list(1018 origin2.clone(),1019 collection_id,1020 31021 ));1022 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);1023 });1024}10251026#[test]1027fn nonprivileged_user_cannot_remove_address_from_white_list() {1028 new_test_ext().execute_with(|| {1029 default_limits();1030 1031 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1032 let origin1 = Origin::signed(1);1033 let origin2 = Origin::signed(2);10341035 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));1036 assert_noop!(1037 TemplateModule::remove_from_white_list(origin2.clone(), collection_id, 2),1038 "You do not have permissions to modify this collection"1039 );1040 assert_eq!(TemplateModule::white_list(collection_id)[0], 2);1041 });1042}10431044#[test]1045fn nobody_can_remove_address_from_white_list_of_nonexisting_collection() {1046 new_test_ext().execute_with(|| {1047 default_limits();1048 let origin1 = Origin::signed(1);10491050 assert_noop!(1051 TemplateModule::remove_from_white_list(origin1.clone(), 1, 2),1052 "This collection does not exist"1053 );1054 });1055}10561057#[test]1058fn nobody_can_remove_address_from_white_list_of_deleted_collection() {1059 new_test_ext().execute_with(|| {1060 default_limits();1061 1062 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1063 let origin1 = Origin::signed(1);1064 let origin2 = Origin::signed(2);10651066 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));1067 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), collection_id));1068 assert_noop!(1069 TemplateModule::remove_from_white_list(origin2.clone(), collection_id, 2),1070 "This collection does not exist"1071 );1072 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);1073 });1074}10751076// If address is already removed from white list, nothing happens1077#[test]1078fn address_is_already_removed_from_white_list() {1079 new_test_ext().execute_with(|| {1080 default_limits();1081 1082 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1083 let origin1 = Origin::signed(1);10841085 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));1086 assert_ok!(TemplateModule::remove_from_white_list(1087 origin1.clone(),1088 collection_id,1089 21090 ));1091 assert_ok!(TemplateModule::remove_from_white_list(1092 origin1.clone(),1093 collection_id,1094 21095 ));1096 assert_eq!(TemplateModule::white_list(collection_id).len(), 0);1097 });1098}10991100// If Public Access mode is set to WhiteList, tokens can’t be transferred from a non-whitelisted address with transfer or transferFrom (2 tests)1101#[test]1102fn white_list_test_1() {1103 new_test_ext().execute_with(|| {1104 default_limits();1105 1106 let collection_id = create_test_collection(&CollectionMode::NFT, 1);11071108 let origin1 = Origin::signed(1);1109 1110 let data = default_nft_data();1111 create_test_item(collection_id, &data.into());11121113 assert_ok!(TemplateModule::set_public_access_mode(1114 origin1.clone(),1115 collection_id,1116 AccessMode::WhiteList1117 ));1118 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));11191120 assert_noop!(1121 TemplateModule::transfer(origin1.clone(), 3, 1, 1, 1),1122 "Address is not in white list"1123 );1124 });1125}11261127#[test]1128fn white_list_test_2() {1129 new_test_ext().execute_with(|| {1130 default_limits();1131 1132 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1133 let origin1 = Origin::signed(1);1134 1135 let data = default_nft_data();1136 create_test_item(collection_id, &data.into());11371138 assert_ok!(TemplateModule::set_public_access_mode(1139 origin1.clone(),1140 collection_id,1141 AccessMode::WhiteList1142 ));1143 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));1144 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 2));11451146 // do approve1147 assert_ok!(TemplateModule::approve(origin1.clone(), 1, 1, 1));1148 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);11491150 assert_ok!(TemplateModule::remove_from_white_list(1151 origin1.clone(),1152 1,1153 11154 ));11551156 assert_noop!(1157 TemplateModule::transfer_from(origin1.clone(), 1, 3, 1, 1, 1),1158 "Address is not in white list"1159 );1160 });1161}11621163// If Public Access mode is set to WhiteList, tokens can’t be transferred to a non-whitelisted address with transfer or transferFrom (2 tests)1164#[test]1165fn white_list_test_3() {1166 new_test_ext().execute_with(|| {1167 default_limits();1168 1169 let collection_id = create_test_collection(&CollectionMode::NFT, 1);11701171 let origin1 = Origin::signed(1);1172 1173 let data = default_nft_data();1174 create_test_item(collection_id, &data.into());11751176 assert_ok!(TemplateModule::set_public_access_mode(1177 origin1.clone(),1178 collection_id,1179 AccessMode::WhiteList1180 ));1181 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), 1, 1));11821183 assert_noop!(1184 TemplateModule::transfer(origin1.clone(), 3, 1, 1, 1),1185 "Address is not in white list"1186 );1187 });1188}11891190#[test]1191fn white_list_test_4() {1192 new_test_ext().execute_with(|| {1193 default_limits();1194 1195 let collection_id = create_test_collection(&CollectionMode::NFT, 1);11961197 let origin1 = Origin::signed(1);11981199 let data = default_nft_data();1200 create_test_item(collection_id, &data.into());12011202 assert_ok!(TemplateModule::set_public_access_mode(1203 origin1.clone(),1204 collection_id,1205 AccessMode::WhiteList1206 ));1207 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 1));1208 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));12091210 // do approve1211 assert_ok!(TemplateModule::approve(origin1.clone(), 1, 1, 1));1212 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);12131214 assert_ok!(TemplateModule::remove_from_white_list(1215 origin1.clone(),1216 collection_id,1217 21218 ));12191220 assert_noop!(1221 TemplateModule::transfer_from(origin1.clone(), 1, 3, 1, 1, 1),1222 "Address is not in white list"1223 );1224 });1225}12261227// If Public Access mode is set to WhiteList, tokens can’t be destroyed by a non-whitelisted address (even if it owned them before enabling WhiteList mode)1228#[test]1229fn white_list_test_5() {1230 new_test_ext().execute_with(|| {1231 default_limits();1232 1233 let collection_id = create_test_collection(&CollectionMode::NFT, 1);12341235 let origin1 = Origin::signed(1);12361237 let data = default_nft_data();1238 create_test_item(collection_id, &data.into());12391240 assert_ok!(TemplateModule::set_public_access_mode(1241 origin1.clone(),1242 collection_id,1243 AccessMode::WhiteList1244 ));1245 assert_noop!(1246 TemplateModule::burn_item(origin1.clone(), 1, 1),1247 "Address is not in white list"1248 );1249 });1250}12511252// If Public Access mode is set to WhiteList, oken transfers can’t be Approved by a non-whitelisted address (see Approve method).1253#[test]1254fn white_list_test_6() {1255 new_test_ext().execute_with(|| {1256 default_limits();1257 1258 let collection_id = create_test_collection(&CollectionMode::NFT, 1);12591260 let origin1 = Origin::signed(1);1261 assert_ok!(TemplateModule::set_public_access_mode(1262 origin1.clone(),1263 collection_id,1264 AccessMode::WhiteList1265 ));12661267 // do approve1268 assert_noop!(1269 TemplateModule::approve(origin1.clone(), 1, 1, 1),1270 "Address is not in white list"1271 );1272 });1273}12741275// If Public Access mode is set to WhiteList, tokens can be transferred from a whitelisted address with transfer or transferFrom (2 tests) and1276// tokens can be transferred from a whitelisted address with transfer or transferFrom (2 tests)1277#[test]1278fn white_list_test_7() {1279 new_test_ext().execute_with(|| {1280 default_limits();1281 1282 let collection_id = create_test_collection(&CollectionMode::NFT, 1);12831284 let data = default_nft_data();1285 create_test_item(collection_id, &data.into());1286 1287 let origin1 = Origin::signed(1);12881289 assert_ok!(TemplateModule::set_public_access_mode(1290 origin1.clone(),1291 collection_id,1292 AccessMode::WhiteList1293 ));1294 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 1));1295 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));12961297 assert_ok!(TemplateModule::transfer(origin1.clone(), 2, 1, 1, 1));1298 });1299}13001301#[test]1302fn white_list_test_8() {1303 new_test_ext().execute_with(|| {1304 default_limits();1305 1306 let collection_id = create_test_collection(&CollectionMode::NFT, 1);13071308 let data = default_nft_data();1309 create_test_item(collection_id, &data.into());1310 1311 let origin1 = Origin::signed(1);13121313 assert_ok!(TemplateModule::set_public_access_mode(1314 origin1.clone(),1315 collection_id,1316 AccessMode::WhiteList1317 ));1318 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 1));1319 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));13201321 // do approve1322 assert_ok!(TemplateModule::approve(origin1.clone(), 1, 1, 1));1323 assert_eq!(TemplateModule::approved(1, (1, 1)).len(), 1);13241325 assert_ok!(TemplateModule::transfer_from(1326 origin1.clone(),1327 1,1328 2,1329 1,1330 1,1331 11332 ));1333 });1334}13351336// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens can be created by owner.1337#[test]1338fn white_list_test_9() {1339 new_test_ext().execute_with(|| {1340 default_limits();1341 1342 let collection_id = create_test_collection(&CollectionMode::NFT, 1);1343 let origin1 = Origin::signed(1);13441345 assert_ok!(TemplateModule::set_public_access_mode(1346 origin1.clone(),1347 collection_id,1348 AccessMode::WhiteList1349 ));1350 assert_ok!(TemplateModule::set_mint_permission(1351 origin1.clone(),1352 collection_id,1353 false1354 ));13551356 let data = default_nft_data();1357 create_test_item(collection_id, &data.into());1358 });1359}13601361// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens can be created by admin.1362#[test]1363fn white_list_test_10() {1364 new_test_ext().execute_with(|| {1365 default_limits();1366 1367 let collection_id = create_test_collection(&CollectionMode::NFT, 1);13681369 let origin1 = Origin::signed(1);1370 let origin2 = Origin::signed(2);13711372 assert_ok!(TemplateModule::set_public_access_mode(1373 origin1.clone(),1374 collection_id,1375 AccessMode::WhiteList1376 ));1377 assert_ok!(TemplateModule::set_mint_permission(1378 origin1.clone(),1379 collection_id,1380 false1381 ));13821383 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));13841385 assert_ok!(TemplateModule::create_item(1386 origin2.clone(),1387 collection_id,1388 2,1389 default_nft_data().into()1390 ));1391 });1392}13931394// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens cannot be created by non-privileged and white listed address.1395#[test]1396fn white_list_test_11() {1397 new_test_ext().execute_with(|| {1398 default_limits();1399 1400 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14011402 let origin1 = Origin::signed(1);1403 let origin2 = Origin::signed(2);14041405 assert_ok!(TemplateModule::set_public_access_mode(1406 origin1.clone(),1407 collection_id,1408 AccessMode::WhiteList1409 ));1410 assert_ok!(TemplateModule::set_mint_permission(1411 origin1.clone(),1412 collection_id,1413 false1414 ));1415 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));14161417 assert_noop!(1418 TemplateModule::create_item(origin2.clone(), 1, 2, default_nft_data().into()),1419 "Public minting is not allowed for this collection"1420 );1421 });1422}14231424// If Public Access mode is set to WhiteList, and Mint Permission is set to false, tokens cannot be created by non-privileged and non-white listed address.1425#[test]1426fn white_list_test_12() {1427 new_test_ext().execute_with(|| {1428 default_limits();1429 1430 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14311432 let origin1 = Origin::signed(1);1433 let origin2 = Origin::signed(2);14341435 assert_ok!(TemplateModule::set_public_access_mode(1436 origin1.clone(),1437 collection_id,1438 AccessMode::WhiteList1439 ));1440 assert_ok!(TemplateModule::set_mint_permission(1441 origin1.clone(),1442 collection_id,1443 false1444 ));14451446 assert_noop!(1447 TemplateModule::create_item(origin2.clone(), 1, 2, default_nft_data().into()),1448 "Public minting is not allowed for this collection"1449 );1450 });1451}14521453// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens can be created by owner.1454#[test]1455fn white_list_test_13() {1456 new_test_ext().execute_with(|| {1457 default_limits();1458 1459 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14601461 let origin1 = Origin::signed(1);14621463 assert_ok!(TemplateModule::set_public_access_mode(1464 origin1.clone(),1465 collection_id,1466 AccessMode::WhiteList1467 ));1468 assert_ok!(TemplateModule::set_mint_permission(1469 origin1.clone(),1470 collection_id,1471 true1472 ));14731474 let data = default_nft_data();1475 create_test_item(collection_id, &data.into());1476 });1477}14781479// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens can be created by admin.1480#[test]1481fn white_list_test_14() {1482 new_test_ext().execute_with(|| {1483 default_limits();1484 1485 let collection_id = create_test_collection(&CollectionMode::NFT, 1);14861487 let origin1 = Origin::signed(1);1488 let origin2 = Origin::signed(2);14891490 assert_ok!(TemplateModule::set_public_access_mode(1491 origin1.clone(),1492 collection_id,1493 AccessMode::WhiteList1494 ));1495 assert_ok!(TemplateModule::set_mint_permission(1496 origin1.clone(),1497 collection_id,1498 true1499 ));15001501 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));15021503 assert_ok!(TemplateModule::create_item(1504 origin2.clone(),1505 1,1506 2,1507 default_nft_data().into()1508 ));1509 });1510}15111512// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens cannot be created by non-privileged and non-white listed address.1513#[test]1514fn white_list_test_15() {1515 new_test_ext().execute_with(|| {1516 default_limits();1517 1518 let collection_id = create_test_collection(&CollectionMode::NFT, 1);15191520 let origin1 = Origin::signed(1);1521 let origin2 = Origin::signed(2);15221523 assert_ok!(TemplateModule::set_public_access_mode(1524 origin1.clone(),1525 collection_id,1526 AccessMode::WhiteList1527 ));1528 assert_ok!(TemplateModule::set_mint_permission(1529 origin1.clone(),1530 collection_id,1531 true1532 ));15331534 assert_noop!(1535 TemplateModule::create_item(origin2.clone(), 1, 2, default_nft_data().into()),1536 "Address is not in white list"1537 );1538 });1539}15401541// If Public Access mode is set to WhiteList, and Mint Permission is set to true, tokens can be created by non-privileged and white listed address.1542#[test]1543fn white_list_test_16() {1544 new_test_ext().execute_with(|| {1545 default_limits();1546 1547 let collection_id = create_test_collection(&CollectionMode::NFT, 1);15481549 let origin1 = Origin::signed(1);1550 let origin2 = Origin::signed(2);15511552 assert_ok!(TemplateModule::set_public_access_mode(1553 origin1.clone(),1554 collection_id,1555 AccessMode::WhiteList1556 ));1557 assert_ok!(TemplateModule::set_mint_permission(1558 origin1.clone(),1559 collection_id,1560 true1561 ));1562 assert_ok!(TemplateModule::add_to_white_list(origin1.clone(), collection_id, 2));15631564 assert_ok!(TemplateModule::create_item(1565 origin2.clone(),1566 1,1567 2,1568 default_nft_data().into()1569 ));1570 });1571}15721573// Total number of collections. Positive test1574#[test]1575fn total_number_collections_bound() {1576 new_test_ext().execute_with(|| {1577 default_limits();1578 1579 create_test_collection(&CollectionMode::NFT, 1);1580 });1581}15821583// Total number of collections. Negotive test1584#[test]1585fn total_number_collections_bound_neg() {1586 new_test_ext().execute_with(|| {1587 default_limits();15881589 let origin1 = Origin::signed(1);15901591 for i in 0..default_collection_numbers_limit() {1592 create_test_collection(&CollectionMode::NFT, i + 1);1593 }15941595 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();1596 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();1597 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();15981599 // 11-th collection in chain. Expects error1600 assert_noop!(TemplateModule::create_collection(1601 origin1.clone(),1602 col_name1.clone(),1603 col_desc1.clone(),1604 token_prefix1.clone(),1605 CollectionMode::NFT1606 ), "Total collections bound exceeded");1607 });1608}16091610// Owned tokens by a single address. Positive test1611#[test]1612fn owned_tokens_bound() {1613 new_test_ext().execute_with(|| {1614 default_limits();1615 1616 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16171618 let data = default_nft_data();1619 create_test_item(collection_id, &data.clone().into());1620 create_test_item(collection_id, &data.into());1621 });1622}16231624// Owned tokens by a single address. Negotive test1625#[test]1626fn owned_tokens_bound_neg() {1627 new_test_ext().execute_with(|| {1628 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1629 collection_numbers_limit: 10,1630 account_token_ownership_limit: 1,1631 collections_admins_limit: 5,1632 custom_data_limit: 2048,1633 nft_sponsor_transfer_timeout: 15,1634 fungible_sponsor_transfer_timeout: 15,1635 refungible_sponsor_transfer_timeout: 15, 1636 }));1637 1638 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16391640 let origin1 = Origin::signed(1);1641 let data = default_nft_data();1642 create_test_item(collection_id, &data.clone().into());16431644 assert_noop!(TemplateModule::create_item(1645 origin1.clone(),1646 1,1647 1,1648 data.into()1649 ), "Owned tokens by a single address bound exceeded");1650 });1651}16521653// Number of collection admins. Positive test1654#[test]1655fn collection_admins_bound() {1656 new_test_ext().execute_with(|| {1657 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1658 collection_numbers_limit: 10,1659 account_token_ownership_limit: 10,1660 collections_admins_limit: 2,1661 custom_data_limit: 2048,1662 nft_sponsor_transfer_timeout: 15,1663 fungible_sponsor_transfer_timeout: 15,1664 refungible_sponsor_transfer_timeout: 15, 1665 }));1666 1667 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16681669 let origin1 = Origin::signed(1);1670 1671 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));1672 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 3));1673 });1674}16751676// Number of collection admins. Negotive test1677#[test]1678fn collection_admins_bound_neg() {1679 new_test_ext().execute_with(|| {1680 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1681 collection_numbers_limit: 10,1682 account_token_ownership_limit: 1,1683 collections_admins_limit: 1,1684 custom_data_limit: 2048,1685 nft_sponsor_transfer_timeout: 15,1686 fungible_sponsor_transfer_timeout: 15,1687 refungible_sponsor_transfer_timeout: 15, 1688 }));1689 1690 let collection_id = create_test_collection(&CollectionMode::NFT, 1);16911692 let origin1 = Origin::signed(1);16931694 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 2));1695 assert_noop!(TemplateModule::add_collection_admin(origin1.clone(), collection_id, 3), "Number of collection admins bound exceeded");1696 });1697}16981699// NFT custom data size. Negative test const_data.1700#[test]1701fn custom_data_size_nft_const_data_bound_neg() {1702 new_test_ext().execute_with(|| {1703 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1704 collection_numbers_limit: 10,1705 account_token_ownership_limit: 10,1706 collections_admins_limit: 5,1707 custom_data_limit: 2,1708 nft_sponsor_transfer_timeout: 15,1709 fungible_sponsor_transfer_timeout: 15,1710 refungible_sponsor_transfer_timeout: 15, 1711 }));1712 1713 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17141715 let origin1 = Origin::signed(1);1716 let too_big_const_data = CreateItemData::NFT(CreateNftData{1717 const_data: vec![1, 2, 3, 4],1718 variable_data: vec![]1719 });17201721 assert_noop!(TemplateModule::create_item(1722 origin1.clone(),1723 collection_id,1724 1,1725 too_big_const_data1726 ), "const_data exceeded data limit.");1727 });1728}17291730// NFT custom data size. Negative test variable_data.1731#[test]1732fn custom_data_size_nft_variable_data_bound_neg() {1733 new_test_ext().execute_with(|| {1734 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1735 collection_numbers_limit: 10,1736 account_token_ownership_limit: 10,1737 collections_admins_limit: 5,1738 custom_data_limit: 2,1739 nft_sponsor_transfer_timeout: 15,1740 fungible_sponsor_transfer_timeout: 15,1741 refungible_sponsor_transfer_timeout: 15, 1742 }));17431744 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17451746 let origin1 = Origin::signed(1);1747 let too_big_const_data = CreateItemData::NFT(CreateNftData{1748 const_data: vec![],1749 variable_data: vec![1, 2, 3, 4]1750 });17511752 assert_noop!(TemplateModule::create_item(1753 origin1.clone(),1754 collection_id,1755 1,1756 too_big_const_data1757 ), "variable_data exceeded data limit.");1758 });1759}17601761// Re fungible custom data size. Negative test const_data.1762#[test]1763fn custom_data_size_re_fungible_const_data_bound_neg() {1764 new_test_ext().execute_with(|| {1765 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1766 collection_numbers_limit: 10,1767 account_token_ownership_limit: 10,1768 collections_admins_limit: 5,1769 custom_data_limit: 2,1770 nft_sponsor_transfer_timeout: 15,1771 fungible_sponsor_transfer_timeout: 15,1772 refungible_sponsor_transfer_timeout: 15, 1773 }));17741775 let collection_id = create_test_collection(&CollectionMode::NFT, 1);17761777 let origin1 = Origin::signed(1);1778 let too_big_const_data = CreateItemData::NFT(CreateNftData{1779 const_data: vec![1, 2, 3, 4],1780 variable_data: vec![]1781 });17821783 assert_noop!(TemplateModule::create_item(1784 origin1.clone(),1785 collection_id,1786 1,1787 too_big_const_data1788 ), "const_data exceeded data limit.");1789 });1790}17911792// Re fungible custom data size. Negative test variable_data.1793#[test]1794fn custom_data_size_re_fungible_variable_data_bound_neg() {1795 new_test_ext().execute_with(|| {1796 assert_ok!(TemplateModule::set_chain_limits(RawOrigin::Root.into(), ChainLimits { 1797 collection_numbers_limit: 10,1798 account_token_ownership_limit: 10,1799 collections_admins_limit: 5,1800 custom_data_limit: 2,1801 nft_sponsor_transfer_timeout: 15,1802 fungible_sponsor_transfer_timeout: 15,1803 refungible_sponsor_transfer_timeout: 15, 1804 }));18051806 let collection_id = create_test_collection(&CollectionMode::NFT, 1);18071808 let origin1 = Origin::signed(1);1809 let too_big_const_data = CreateItemData::NFT(CreateNftData{1810 const_data: vec![],1811 variable_data: vec![1, 2, 3, 4]1812 });18131814 assert_noop!(TemplateModule::create_item(1815 origin1.clone(),1816 collection_id,1817 1,1818 too_big_const_data1819 ), "variable_data exceeded data limit.");1820 });1821}1822// #endregion18231824#[test]1825fn set_const_on_chain_schema() {1826 new_test_ext().execute_with(|| {1827 default_limits();18281829 let collection_id = create_test_collection(&CollectionMode::NFT, 1);18301831 let origin1 = Origin::signed(1);1832 assert_ok!(TemplateModule::set_const_on_chain_schema(origin1, collection_id, b"test const on chain schema".to_vec()));18331834 assert_eq!(TemplateModule::collection(collection_id).const_on_chain_schema, b"test const on chain schema".to_vec());1835 assert_eq!(TemplateModule::collection(collection_id).variable_on_chain_schema, b"".to_vec());1836 });1837}18381839#[test]1840fn set_variable_on_chain_schema() {1841 new_test_ext().execute_with(|| {1842 default_limits();1843 1844 let collection_id = create_test_collection(&CollectionMode::NFT, 1);18451846 let origin1 = Origin::signed(1);1847 assert_ok!(TemplateModule::set_variable_on_chain_schema(origin1, collection_id, b"test variable on chain schema".to_vec()));18481849 assert_eq!(TemplateModule::collection(collection_id).const_on_chain_schema, b"".to_vec());1850 assert_eq!(TemplateModule::collection(collection_id).variable_on_chain_schema, b"test variable on chain schema".to_vec());1851 });1852}18531854#[test]1855fn set_variable_meta_data_on_nft_token_stores_variable_meta_data() {1856 new_test_ext().execute_with(|| {1857 default_limits();18581859 let collection_id = create_test_collection(&CollectionMode::NFT, 1);18601861 let origin1 = Origin::signed(1);1862 1863 let data = default_nft_data();1864 create_test_item(1, &data.into());1865 1866 let variable_data = b"test set_variable_meta_data method.".to_vec();1867 assert_ok!(TemplateModule::set_variable_meta_data(origin1, collection_id, 1, variable_data.clone()));18681869 assert_eq!(TemplateModule::nft_item_id(collection_id, 1).variable_data, variable_data);1870 });1871}18721873#[test]1874fn set_variable_meta_data_on_re_fungible_token_stores_variable_meta_data() {1875 new_test_ext().execute_with(|| {1876 default_limits();18771878 let collection_id = create_test_collection(&CollectionMode::ReFungible(3), 1);18791880 let origin1 = Origin::signed(1);18811882 let data = default_re_fungible_data();1883 create_test_item(1, &data.into());18841885 let variable_data = b"test set_variable_meta_data method.".to_vec();1886 assert_ok!(TemplateModule::set_variable_meta_data(origin1, collection_id, 1, variable_data.clone()));18871888 assert_eq!(TemplateModule::refungible_item_id(collection_id, 1).variable_data, variable_data);1889 });1890}189118921893#[test]1894fn set_variable_meta_data_on_fungible_token_fails() {1895 new_test_ext().execute_with(|| {1896 default_limits();18971898 let collection_id = create_test_collection(&CollectionMode::Fungible(3), 1);18991900 let origin1 = Origin::signed(1);19011902 let data = default_fungible_data();1903 create_test_item(1, &data.into());19041905 let variable_data = b"test set_variable_meta_data method.".to_vec();1906 assert_noop!(TemplateModule::set_variable_meta_data(origin1, collection_id, 1, variable_data.clone()), "Can't store metadata in fungible tokens.");1907 });1908}runtime/src/nft_weights.rsdiffbeforeafterboth--- a/runtime/src/nft_weights.rs
+++ b/runtime/src/nft_weights.rs
@@ -64,7 +64,7 @@
}
fn create_item(s: usize, ) -> Weight {
(130_000_000 as Weight)
- .saturating_add((2135 as Weight).saturating_mul(s as Weight))
+ .saturating_add((2135 as Weight).saturating_mul(s as Weight).saturating_mul(500 as Weight)) // 500 is temparary multiplier, fee for storage
.saturating_add(DbWeight::get().reads(10 as Weight))
.saturating_add(DbWeight::get().writes(8 as Weight))
}