difftreelog
NFTPAR-56
in: master
3 files changed
pallets/nft/src/lib.rsdiffbeforeafterboth--- a/pallets/nft/src/lib.rs
+++ b/pallets/nft/src/lib.rs
@@ -24,6 +24,9 @@
pub struct CollectionType<AccountId> {
pub owner: AccountId,
pub next_item_id: u64,
+ pub name: Vec<u16>, // 64 include null escape char
+ pub description: Vec<u16>, // 256 include null escape char
+ pub token_trefix: Vec<u8>, // 16 include null escape char
pub custom_data_size: u32,
}
@@ -79,7 +82,9 @@
where
AccountId = <T as system::Trait>::AccountId,
{
- Created(u32, AccountId),
+ Created(u64, AccountId),
+ ItemCreated(u64),
+ ItemDestroyed(u64, u64),
}
);
@@ -91,10 +96,6 @@
// Initializing events
// this is needed only if you are using events in your pallet
fn deposit_event() = default;
-
- // Initializing events
- // this is needed only if you are using events in your module
- // fn deposit_event<T>() = default;
// Create collection of NFT with given parameters
//
@@ -106,10 +107,28 @@
// @param customDataSz size of custom data in each collection item
// returns collection ID
#[weight = 0]
- pub fn create_collection(origin, custom_data_sz: u32) -> DispatchResult {
+ pub fn create_collection( origin,
+ collection_name: Vec<u16>,
+ collection_description: Vec<u16>,
+ token_prefix: Vec<u8>,
+ custom_data_sz: u32) -> DispatchResult {
+
// Anyone can create a collection
let who = ensure_signed(origin)?;
+ // check params
+ let mut name = collection_name.to_vec();
+ name.push(0);
+ ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");
+
+ let mut description = collection_description.to_vec();
+ description.push(0);
+ ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");
+
+ let mut token_trefix = token_prefix.to_vec();
+ token_trefix.push(0);
+ ensure!(token_trefix.len() <= 16, "Token prefix can not be longer than 15 char");
+
// Generate next collection ID
let next_id = NextCollectionID::get();
@@ -117,7 +136,10 @@
// Create new collection
let new_collection = CollectionType {
- owner: who,
+ owner: who.clone(),
+ name: name,
+ description: description,
+ token_trefix: token_trefix,
next_item_id: next_id,
custom_data_size: custom_data_sz,
};
@@ -125,6 +147,9 @@
// Add new collection to map
<Collection<T>>::insert(next_id, new_collection);
+ // call event
+ Self::deposit_event(RawEvent::Created(next_id, who.clone()));
+
Ok(())
}
@@ -252,6 +277,9 @@
<ItemListIndex>::insert(collection_id, current_index);
<ItemList<T>>::insert((collection_id, current_index), new_item);
+ // call event
+ Self::deposit_event(RawEvent::ItemCreated(collection_id));
+
Ok(())
}
@@ -286,6 +314,9 @@
let new_balance = <Balance<T>>::get((collection_id, item.owner.clone())) - 1;
<Balance<T>>::insert((collection_id, item.owner.clone()), new_balance);
+ // call event
+ Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));
+
Ok(())
}
pallets/nft/src/mock.rsdiffbeforeafterboth--- a/pallets/nft/src/mock.rs
+++ b/pallets/nft/src/mock.rs
@@ -30,7 +30,7 @@
pub const MaximumBlockWeight: Weight = 1024;
pub const MaximumBlockLength: u32 = 2 * 1024;
pub const AvailableBlockRatio: Perbill = Perbill::from_percent(75);
- pub MaximumExtrinsicWeight: Weight = 10 * WEIGHT_PER_SECOND;
+ pub const MaximumExtrinsicWeight: Weight = 10 * WEIGHT_PER_SECOND;
}
impl system::Trait for Test {
pallets/nft/src/tests.rsdiffbeforeafterboth1// Tests to be written here2use crate::mock::*;3use frame_support::{assert_noop, assert_ok};45#[test]6fn create_collection_test() {7 new_test_ext().execute_with(|| {8 let size = 1024;9 let origin1 = Origin::signed(1);10 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));11 assert_eq!(TemplateModule::collection(1).owner, 1);12 });13}1415#[test]16fn change_collection_owner() {17 new_test_ext().execute_with(|| {18 let size = 1024;19 let origin1 = Origin::signed(1);2021 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));22 assert_ok!(TemplateModule::change_collection_owner(23 origin1.clone(),24 1,25 226 ));27 assert_eq!(TemplateModule::collection(1).owner, 2);28 });29}3031#[test]32fn destroy_collection() {33 new_test_ext().execute_with(|| {34 let size = 1024;35 let origin1 = Origin::signed(1);3637 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));38 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), 1));39 });40}4142#[test]43fn create_item() {44 new_test_ext().execute_with(|| {45 let size = 1024;46 let origin1 = Origin::signed(1);47 let origin2 = Origin::signed(2);4849 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));50 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));51 assert_ok!(TemplateModule::create_item(52 origin2.clone(),53 1,54 [1, 1, 1].to_vec()55 ));5657 // check balance (collection with id = 1, user id = 2)58 assert_eq!(TemplateModule::balance_count((1, 2)), 1);59 });60}6162#[test]63fn burn_item() {64 new_test_ext().execute_with(|| {65 let size = 1024;66 let origin1 = Origin::signed(1);67 let origin2 = Origin::signed(2);6869 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));70 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));71 assert_ok!(TemplateModule::create_item(72 origin2.clone(),73 1,74 [1, 1, 1].to_vec()75 ));7677 // check balance (collection with id = 1, user id = 2)78 assert_eq!(TemplateModule::balance_count((1, 2)), 1);7980 // burn item81 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));82 assert_noop!(83 TemplateModule::burn_item(origin1.clone(), 1, 1),84 "Item does not exists"85 );8687 assert_eq!(TemplateModule::balance_count((1, 1)), 0);88 });89}9091#[test]92fn add_collection_admin() {93 new_test_ext().execute_with(|| {94 let size = 1024;95 let origin1 = Origin::signed(1);96 let origin2 = Origin::signed(2);97 let origin3 = Origin::signed(3);9899 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));100 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));101 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));102103 assert_eq!(TemplateModule::collection(1).owner, 1);104 assert_eq!(TemplateModule::collection(2).owner, 2);105 assert_eq!(TemplateModule::collection(3).owner, 3);106107 // collection admin108 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));109 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 3));110111 assert_eq!(TemplateModule::admin_list_collection(1).contains(&2), true);112 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), true);113 });114}115116#[test]117fn remove_collection_admin() {118 new_test_ext().execute_with(|| {119 let size = 1024;120 let origin1 = Origin::signed(1);121 let origin2 = Origin::signed(2);122 let origin3 = Origin::signed(3);123124 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));125 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));126 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));127128 assert_eq!(TemplateModule::collection(1).owner, 1);129 assert_eq!(TemplateModule::collection(2).owner, 2);130 assert_eq!(TemplateModule::collection(3).owner, 3);131132 // collection admin133 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));134 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 3));135136 assert_eq!(TemplateModule::admin_list_collection(1).contains(&2), true);137 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), true);138139 // remove admin140 assert_ok!(TemplateModule::remove_collection_admin(141 origin2.clone(),142 1,143 3144 ));145 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), false);146 });147}148149#[test]150fn balance_of() {151 new_test_ext().execute_with(|| {152 let size = 1024;153 let origin1 = Origin::signed(1);154 let origin2 = Origin::signed(2);155 let origin3 = Origin::signed(3);156157 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));158 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));159 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));160161 assert_eq!(TemplateModule::collection(1).owner, 1);162 assert_eq!(TemplateModule::collection(2).owner, 2);163 assert_eq!(TemplateModule::collection(3).owner, 3);164165 // check balance before166 assert_eq!(TemplateModule::balance_count((1, 1)), 0);167168 // create item169 assert_ok!(TemplateModule::create_item(170 origin1.clone(),171 1,172 [1, 1, 1].to_vec()173 ));174175 // check balance (collection with id = 1, user id = 2)176 assert_eq!(TemplateModule::balance_count((1, 1)), 1);177 assert_eq!(TemplateModule::item_id((1, 1)).owner, 1);178 });179}180181#[test]182fn transfer() {183 new_test_ext().execute_with(|| {184 let size = 1024;185 let origin1 = Origin::signed(1);186 let origin2 = Origin::signed(2);187 let origin3 = Origin::signed(3);188189 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));190 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));191 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));192193 assert_eq!(TemplateModule::collection(1).owner, 1);194 assert_eq!(TemplateModule::collection(2).owner, 2);195 assert_eq!(TemplateModule::collection(3).owner, 3);196197 // create item198 assert_ok!(TemplateModule::create_item(199 origin1.clone(),200 1,201 [1, 1, 1].to_vec()202 ));203204 // transfer205 assert_ok!(TemplateModule::transfer(origin1.clone(), 1, 1, 2));206 assert_eq!(TemplateModule::item_id((1, 1)).owner, 2);207208 // balance_of check209 assert_eq!(TemplateModule::balance_count((1, 1)), 0);210 assert_eq!(TemplateModule::balance_count((1, 2)), 1);211 });212}213214#[test]215fn approve() {216 new_test_ext().execute_with(|| {217 let size = 1024;218 let origin1 = Origin::signed(1);219 let origin2 = Origin::signed(2);220 let origin3 = Origin::signed(3);221222 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));223 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));224 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));225226 assert_eq!(TemplateModule::collection(1).owner, 1);227 assert_eq!(TemplateModule::collection(2).owner, 2);228 assert_eq!(TemplateModule::collection(3).owner, 3);229230 // create item231 assert_ok!(TemplateModule::create_item(232 origin1.clone(),233 1,234 [1, 1, 1].to_vec()235 ));236237 // approve238 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));239 assert_eq!(TemplateModule::approved((1, 1)).contains(&2), true);240 });241}242243#[test]244fn get_approved() {245 new_test_ext().execute_with(|| {246 let size = 1024;247 let origin1 = Origin::signed(1);248 let origin2 = Origin::signed(2);249 let origin3 = Origin::signed(3);250251 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));252 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));253 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));254255 assert_eq!(TemplateModule::collection(1).owner, 1);256 assert_eq!(TemplateModule::collection(2).owner, 2);257 assert_eq!(TemplateModule::collection(3).owner, 3);258259 // create item260 assert_ok!(TemplateModule::create_item(261 origin1.clone(),262 1,263 [1, 1, 1].to_vec()264 ));265266 // approve267 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));268 assert_eq!(TemplateModule::approved((1, 1)).contains(&2), true);269 });270}271272#[test]273fn transfer_from() {274 new_test_ext().execute_with(|| {275 let size = 1024;276 let origin1 = Origin::signed(1);277 let origin2 = Origin::signed(2);278 let origin3 = Origin::signed(3);279280 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));281 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));282 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));283284 assert_eq!(TemplateModule::collection(1).owner, 1);285 assert_eq!(TemplateModule::collection(2).owner, 2);286 assert_eq!(TemplateModule::collection(3).owner, 3);287288 // create item289 assert_ok!(TemplateModule::create_item(290 origin1.clone(),291 1,292 [1, 1, 1].to_vec()293 ));294295 // approve296 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));297 assert_ok!(TemplateModule::transfer_from(origin1.clone(), 1, 1, 2));298 });299}300301#[test]302fn index_list() {303 new_test_ext().execute_with(|| {304 let size = 1024;305 let origin1 = Origin::signed(1);306 let origin2 = Origin::signed(2);307 let origin3 = Origin::signed(3);308309 assert_ok!(TemplateModule::create_collection(origin1.clone(), size));310 assert_ok!(TemplateModule::create_collection(origin2.clone(), size));311 assert_ok!(TemplateModule::create_collection(origin3.clone(), size));312313 assert_eq!(TemplateModule::collection(1).owner, 1);314 assert_eq!(TemplateModule::collection(2).owner, 2);315 assert_eq!(TemplateModule::collection(3).owner, 3);316317 // create items318 assert_ok!(TemplateModule::create_item(319 origin1.clone(),320 1,321 [1, 1, 1].to_vec()322 ));323324 assert_ok!(TemplateModule::create_item(325 origin1.clone(),326 1,327 [1, 1, 2].to_vec()328 ));329330 assert_ok!(TemplateModule::create_item(331 origin1.clone(),332 1,333 [1, 2, 3].to_vec()334 ));335336 assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 3);337338 // burn one339 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 2));340 assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 2);341342 // burn another one343 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 3));344 assert_eq!(TemplateModule::address_tokens((1, 1))[0], 1);345 });346}1// Tests to be written here2use crate::mock::*;3use frame_support::{assert_noop, assert_ok};45#[test]6fn create_collection_test() {7 new_test_ext().execute_with(|| {89 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();10 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();11 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();1213 let size = 1024;14 let origin1 = Origin::signed(1);15 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));16 assert_eq!(TemplateModule::collection(1).owner, 1);17 });18}1920#[test]21fn change_collection_owner() {22 new_test_ext().execute_with(|| {2324 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();25 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();26 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();2728 let size = 1024;29 let origin1 = Origin::signed(1);3031 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));32 assert_ok!(TemplateModule::change_collection_owner(33 origin1.clone(),34 1,35 236 ));37 assert_eq!(TemplateModule::collection(1).owner, 2);38 });39}4041#[test]42fn destroy_collection() {43 new_test_ext().execute_with(|| {4445 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();46 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();47 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();48 let size = 1024;49 let origin1 = Origin::signed(1);5051 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));52 assert_ok!(TemplateModule::destroy_collection(origin1.clone(), 1));53 });54}5556#[test]57fn create_item() {58 new_test_ext().execute_with(|| {5960 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();61 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();62 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();6364 let size = 1024;65 let origin1 = Origin::signed(1);66 let origin2 = Origin::signed(2);6768 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));69 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));70 assert_ok!(TemplateModule::create_item(71 origin2.clone(),72 1,73 [1, 1, 1].to_vec()74 ));7576 // check balance (collection with id = 1, user id = 2)77 assert_eq!(TemplateModule::balance_count((1, 2)), 1);78 });79}8081#[test]82fn burn_item() {83 new_test_ext().execute_with(|| {8485 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();86 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();87 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();8889 let size = 1024;90 let origin1 = Origin::signed(1);91 let origin2 = Origin::signed(2);9293 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));94 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));95 assert_ok!(TemplateModule::create_item(96 origin2.clone(),97 1,98 [1, 1, 1].to_vec()99 ));100101 // check balance (collection with id = 1, user id = 2)102 assert_eq!(TemplateModule::balance_count((1, 2)), 1);103104 // burn item105 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 1));106 assert_noop!(107 TemplateModule::burn_item(origin1.clone(), 1, 1),108 "Item does not exists"109 );110111 assert_eq!(TemplateModule::balance_count((1, 1)), 0);112 });113}114115#[test]116fn add_collection_admin() {117 new_test_ext().execute_with(|| {118119 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();120 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();121 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();122123 let size = 1024;124 let origin1 = Origin::signed(1);125 let origin2 = Origin::signed(2);126 let origin3 = Origin::signed(3);127128 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));129 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));130 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));131132 assert_eq!(TemplateModule::collection(1).owner, 1);133 assert_eq!(TemplateModule::collection(2).owner, 2);134 assert_eq!(TemplateModule::collection(3).owner, 3);135136 // collection admin137 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));138 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 3));139140 assert_eq!(TemplateModule::admin_list_collection(1).contains(&2), true);141 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), true);142 });143}144145#[test]146fn remove_collection_admin() {147 new_test_ext().execute_with(|| {148149 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();150 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();151 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();152153 let size = 1024;154 let origin1 = Origin::signed(1);155 let origin2 = Origin::signed(2);156 let origin3 = Origin::signed(3);157158 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));159 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));160 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));161162 assert_eq!(TemplateModule::collection(1).owner, 1);163 assert_eq!(TemplateModule::collection(2).owner, 2);164 assert_eq!(TemplateModule::collection(3).owner, 3);165166 // collection admin167 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));168 assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 3));169170 assert_eq!(TemplateModule::admin_list_collection(1).contains(&2), true);171 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), true);172173 // remove admin174 assert_ok!(TemplateModule::remove_collection_admin(175 origin2.clone(),176 1,177 3178 ));179 assert_eq!(TemplateModule::admin_list_collection(1).contains(&3), false);180 });181}182183#[test]184fn balance_of() {185 new_test_ext().execute_with(|| {186187 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();188 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();189 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();190191 let size = 1024;192 let origin1 = Origin::signed(1);193 let origin2 = Origin::signed(2);194 let origin3 = Origin::signed(3);195196 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));197 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));198 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));199200 assert_eq!(TemplateModule::collection(1).owner, 1);201 assert_eq!(TemplateModule::collection(2).owner, 2);202 assert_eq!(TemplateModule::collection(3).owner, 3);203204 // check balance before205 assert_eq!(TemplateModule::balance_count((1, 1)), 0);206207 // create item208 assert_ok!(TemplateModule::create_item(209 origin1.clone(),210 1,211 [1, 1, 1].to_vec()212 ));213214 // check balance (collection with id = 1, user id = 2)215 assert_eq!(TemplateModule::balance_count((1, 1)), 1);216 assert_eq!(TemplateModule::item_id((1, 1)).owner, 1);217 });218}219220#[test]221fn transfer() {222 new_test_ext().execute_with(|| {223224 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();225 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();226 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();227228 let size = 1024;229 let origin1 = Origin::signed(1);230 let origin2 = Origin::signed(2);231 let origin3 = Origin::signed(3);232233 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));234 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));235 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));236237 assert_eq!(TemplateModule::collection(1).owner, 1);238 assert_eq!(TemplateModule::collection(2).owner, 2);239 assert_eq!(TemplateModule::collection(3).owner, 3);240241 // create item242 assert_ok!(TemplateModule::create_item(243 origin1.clone(),244 1,245 [1, 1, 1].to_vec()246 ));247248 // transfer249 assert_ok!(TemplateModule::transfer(origin1.clone(), 1, 1, 2));250 assert_eq!(TemplateModule::item_id((1, 1)).owner, 2);251252 // balance_of check253 assert_eq!(TemplateModule::balance_count((1, 1)), 0);254 assert_eq!(TemplateModule::balance_count((1, 2)), 1);255 });256}257258#[test]259fn approve() {260 new_test_ext().execute_with(|| {261262 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();263 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();264 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();265266 let size = 1024;267 let origin1 = Origin::signed(1);268 let origin2 = Origin::signed(2);269 let origin3 = Origin::signed(3);270271 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));272 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));273 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));274275 assert_eq!(TemplateModule::collection(1).owner, 1);276 assert_eq!(TemplateModule::collection(2).owner, 2);277 assert_eq!(TemplateModule::collection(3).owner, 3);278279 // create item280 assert_ok!(TemplateModule::create_item(281 origin1.clone(),282 1,283 [1, 1, 1].to_vec()284 ));285286 // approve287 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));288 assert_eq!(TemplateModule::approved((1, 1)).contains(&2), true);289 });290}291292#[test]293fn get_approved() {294 new_test_ext().execute_with(|| {295296 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();297 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();298 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();299300 let size = 1024;301 let origin1 = Origin::signed(1);302 let origin2 = Origin::signed(2);303 let origin3 = Origin::signed(3);304305 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));306 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));307 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));308309 assert_eq!(TemplateModule::collection(1).owner, 1);310 assert_eq!(TemplateModule::collection(2).owner, 2);311 assert_eq!(TemplateModule::collection(3).owner, 3);312313 // create item314 assert_ok!(TemplateModule::create_item(315 origin1.clone(),316 1,317 [1, 1, 1].to_vec()318 ));319320 // approve321 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));322 assert_eq!(TemplateModule::approved((1, 1)).contains(&2), true);323 });324}325326#[test]327fn transfer_from() {328 new_test_ext().execute_with(|| {329330 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();331 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();332 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();333334 let size = 1024;335 let origin1 = Origin::signed(1);336 let origin2 = Origin::signed(2);337 let origin3 = Origin::signed(3);338339 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));340 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));341 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));342343 assert_eq!(TemplateModule::collection(1).owner, 1);344 assert_eq!(TemplateModule::collection(2).owner, 2);345 assert_eq!(TemplateModule::collection(3).owner, 3);346347 // create item348 assert_ok!(TemplateModule::create_item(349 origin1.clone(),350 1,351 [1, 1, 1].to_vec()352 ));353354 // approve355 assert_ok!(TemplateModule::approve(origin1.clone(), 2, 1, 1));356 assert_ok!(TemplateModule::transfer_from(origin1.clone(), 1, 1, 2));357 });358}359360#[test]361fn index_list() {362 new_test_ext().execute_with(|| {363364 let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();365 let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();366 let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();367368 let size = 1024;369 let origin1 = Origin::signed(1);370 let origin2 = Origin::signed(2);371 let origin3 = Origin::signed(3);372373 assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));374 assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));375 assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));376377 assert_eq!(TemplateModule::collection(1).owner, 1);378 assert_eq!(TemplateModule::collection(2).owner, 2);379 assert_eq!(TemplateModule::collection(3).owner, 3);380381 // create items382 assert_ok!(TemplateModule::create_item(383 origin1.clone(),384 1,385 [1, 1, 1].to_vec()386 ));387388 assert_ok!(TemplateModule::create_item(389 origin1.clone(),390 1,391 [1, 1, 2].to_vec()392 ));393394 assert_ok!(TemplateModule::create_item(395 origin1.clone(),396 1,397 [1, 2, 3].to_vec()398 ));399400 assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 3);401402 // burn one403 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 2));404 assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 2);405406 // burn another one407 assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 3));408 assert_eq!(TemplateModule::address_tokens((1, 1))[0], 1);409 });410}