git.delta.rocks / unique-network / refs/commits / dc1cc2f6399a

difftreelog

NFT-56 + 47 added

str-mv2020-07-09parent: #7c556ae.patch.diff
in: master

2 files changed

modifiedpallets/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_prefix: Vec<u8>, // 16 include null escape char
     pub custom_data_size: u32,
 }
 
@@ -58,21 +61,17 @@
 
         /// Next available collection ID
         pub NextCollectionID get(fn next_collection_id): u64;
-
-        pub Collection get(collection): map hasher(identity) u64 => CollectionType<T::AccountId>;
-        //pub Collection get(collection): map hasher(identity) u64 => CollectionType<T::AccountId>;
-
-        pub AdminList get(admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;
+        pub Collection get(fn collection): map hasher(identity) u64 => CollectionType<T::AccountId>;
+        pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;
 
         /// Balance owner per collection map
-        pub Balance get(balance_count): map hasher(blake2_128_concat) (u64, T::AccountId) => u64;
-        pub ApprovedList get(approved): map hasher(blake2_128_concat) (u64, u64) => Vec<T::AccountId>;
+        pub Balance get(fn balance_count): map hasher(blake2_128_concat) (u64, T::AccountId) => u64;
+        pub ApprovedList get(fn approved): map hasher(blake2_128_concat) (u64, u64) => Vec<T::AccountId>;
 
-        pub ItemList get(item_id): map hasher(blake2_128_concat) (u64, u64) => NftItemType<T::AccountId>;
-        // pub ItemList get(item_id): map hasher(blake2_128_concat) (u64, u64) => NftItemType<T::AccountId>;
+        pub ItemList get(fn item_id): map hasher(blake2_128_concat) (u64, u64) => NftItemType<T::AccountId>;
+        pub ItemListIndex get(fn item_index): map hasher(blake2_128_concat) u64 => u64;
 
-        pub ItemListIndex get(item_index): map hasher(blake2_128_concat) u64 => u64;
-        // pub ItemListIndex get(item_index): map hasher(blake2_128_concat) u64 => u64;
+        pub AddressTokens get(fn address_tokens): map hasher(blake2_128_concat) (u64, T::AccountId) => Vec<u64>;
     }
 }
 
@@ -82,7 +81,9 @@
     where
         AccountId = <T as system::Trait>::AccountId,
     {
-        Created(u32, AccountId),
+        Created(u64, AccountId),
+        ItemCreated(u64),
+        ItemDestroyed(u64, u64),
     }
 );
 
@@ -109,18 +110,41 @@
         // @param customDataSz size of custom data in each collection item
         // returns collection ID
         #[weight = frame_support::weights::SimpleDispatchInfo::default()]
-        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 prefix = token_prefix.to_vec();
+            prefix.push(0);
+            ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");
+
             // Generate next collection ID
-            let next_id = NextCollectionID::get();
+            let next_id = NextCollectionID::get()
+                .checked_add(1)
+                .expect("collection id error");
 
             NextCollectionID::put(next_id);
 
             // Create new collection
             let new_collection = CollectionType {
-                owner: who,
+                owner: who.clone(),
+                name: name,
+                description: description,
+                token_prefix: prefix,
                 next_item_id: next_id,
                 custom_data_size: custom_data_sz,
             };
@@ -128,6 +152,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(())
         }
 
@@ -247,10 +274,19 @@
                 data: properties,
             };
 
-            let current_index = <ItemListIndex>::get(collection_id);
+
+            let current_index = <ItemListIndex>::get(collection_id)
+                .checked_add(1)
+                .expect("Item list index id error");
+
+            Self::add_token_index(collection_id, current_index, new_item.owner.clone())?;
+
             <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(())
         }
 
@@ -279,10 +315,15 @@
             }
             <ItemList<T>>::remove((collection_id, item_id));
 
+            Self::remove_token_index(collection_id, item_id, item.owner.clone())?;
+
             // update balance
             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(())
         }
 
@@ -319,9 +360,13 @@
             <Balance<T>>::insert((collection_id, new_owner.clone()), balance_new_owner);
 
             // change owner
-            item.owner = new_owner;
+            let old_owner = item.owner.clone();
+            item.owner = new_owner.clone();
             <ItemList<T>>::insert((collection_id, item_id), item);
 
+            // update index collection
+            Self::move_token_index(collection_id, item_id, old_owner, new_owner.clone())?;
+
             // reset approved list
             let itm: Vec<T::AccountId> = Vec::new();
             <ApprovedList<T>>::insert((collection_id, item_id), itm);
@@ -401,3 +446,55 @@
         }
     }
 }
+
+
+impl<T: Trait> Module<T> {
+    fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {
+        
+        let list_exists = <AddressTokens<T>>::contains_key((collection_id, owner.clone()));
+        if list_exists {
+
+            let mut list = <AddressTokens<T>>::get((collection_id, owner.clone()));
+            let item_contains = list.contains(&item_index.clone());
+
+            if !item_contains {
+                list.push(item_index.clone());
+            }
+
+            <AddressTokens<T>>::insert((collection_id, owner.clone()), list);
+
+        } else {
+
+            let mut itm = Vec::new();
+            itm.push(item_index.clone());
+            <AddressTokens<T>>::insert((collection_id, owner), itm);
+        }
+
+        Ok(())
+    }
+
+    fn remove_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {
+        
+        let list_exists = <AddressTokens<T>>::contains_key((collection_id, owner.clone()));
+        if list_exists {
+
+            let mut list = <AddressTokens<T>>::get((collection_id, owner.clone()));
+            let item_contains = list.contains(&item_index.clone());
+
+            if item_contains {
+                list.retain(|&item| item != item_index);
+                <AddressTokens<T>>::insert((collection_id, owner), list);
+            }
+        }
+
+        Ok(())
+    }
+
+    fn move_token_index(collection_id: u64, item_index: u64, old_owner: T::AccountId, new_owner: T::AccountId) -> DispatchResult {
+        
+        Self::remove_token_index(collection_id, item_index, old_owner)?;
+        Self::add_token_index(collection_id, item_index, new_owner)?;
+        
+        Ok(())
+    }
+}
\ No newline at end of file
modifiedpallets/nft/src/tests.rsdiffbeforeafterboth
5#[test]5#[test]
6fn create_collection_test() {6fn create_collection_test() {
7    new_test_ext().execute_with(|| {7    new_test_ext().execute_with(|| {
8
9        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();
12
8        let size = 1024;13        let size = 1024;
9        let origin1 = Origin::signed(1);14        let origin1 = Origin::signed(1);
10        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));15        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
11        assert_eq!(TemplateModule::collection(1).owner, 1);16        assert_eq!(TemplateModule::collection(1).owner, 1);
12    });17    });
13}18}
1419
15#[test]20#[test]
16fn change_collection_owner() {21fn change_collection_owner() {
17    new_test_ext().execute_with(|| {22    new_test_ext().execute_with(|| {
23
24        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();
27
18        let size = 1024;28        let size = 1024;
19        let origin1 = Origin::signed(1);29        let origin1 = Origin::signed(1);
2030
21        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));31        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
22        assert_ok!(TemplateModule::change_collection_owner(32        assert_ok!(TemplateModule::change_collection_owner(
23            origin1.clone(),33            origin1.clone(),
24            1,34            1,
31#[test]41#[test]
32fn destroy_collection() {42fn destroy_collection() {
33    new_test_ext().execute_with(|| {43    new_test_ext().execute_with(|| {
44
45        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
34        let size = 1024;49        let size = 1024;
35        let origin1 = Origin::signed(1);50        let origin1 = Origin::signed(1);
3651
37        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));52        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
38        assert_ok!(TemplateModule::destroy_collection(origin1.clone(), 1));53        assert_ok!(TemplateModule::destroy_collection(origin1.clone(), 1));
39    });54    });
40}55}
4156
42#[test]57#[test]
43fn create_item() {58fn create_item() {
44    new_test_ext().execute_with(|| {59    new_test_ext().execute_with(|| {
60
61        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
62        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
63        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
64
45        let size = 1024;65        let size = 1024;
46        let origin1 = Origin::signed(1);66        let origin1 = Origin::signed(1);
47        let origin2 = Origin::signed(2);67        let origin2 = Origin::signed(2);
4868
49        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));69        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
50        assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));70        assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));
51        assert_ok!(TemplateModule::create_item(71        assert_ok!(TemplateModule::create_item(
52            origin2.clone(),72            origin2.clone(),
62#[test]82#[test]
63fn burn_item() {83fn burn_item() {
64    new_test_ext().execute_with(|| {84    new_test_ext().execute_with(|| {
85
86        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
87        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
88        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
89
65        let size = 1024;90        let size = 1024;
66        let origin1 = Origin::signed(1);91        let origin1 = Origin::signed(1);
67        let origin2 = Origin::signed(2);92        let origin2 = Origin::signed(2);
6893
69        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));94        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
70        assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));95        assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));
71        assert_ok!(TemplateModule::create_item(96        assert_ok!(TemplateModule::create_item(
72            origin2.clone(),97            origin2.clone(),
91#[test]116#[test]
92fn add_collection_admin() {117fn add_collection_admin() {
93    new_test_ext().execute_with(|| {118    new_test_ext().execute_with(|| {
119
120        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
121        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
122        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
123
94        let size = 1024;124        let size = 1024;
95        let origin1 = Origin::signed(1);125        let origin1 = Origin::signed(1);
96        let origin2 = Origin::signed(2);126        let origin2 = Origin::signed(2);
97        let origin3 = Origin::signed(3);127        let origin3 = Origin::signed(3);
98128
99        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));129        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
100        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));130        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
101        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));131        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
102132
103        assert_eq!(TemplateModule::collection(1).owner, 1);133        assert_eq!(TemplateModule::collection(1).owner, 1);
104        assert_eq!(TemplateModule::collection(2).owner, 2);134        assert_eq!(TemplateModule::collection(2).owner, 2);
116#[test]146#[test]
117fn remove_collection_admin() {147fn remove_collection_admin() {
118    new_test_ext().execute_with(|| {148    new_test_ext().execute_with(|| {
149
150        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
151        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
152        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
153
119        let size = 1024;154        let size = 1024;
120        let origin1 = Origin::signed(1);155        let origin1 = Origin::signed(1);
121        let origin2 = Origin::signed(2);156        let origin2 = Origin::signed(2);
122        let origin3 = Origin::signed(3);157        let origin3 = Origin::signed(3);
123158
124        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));159        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
125        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));160        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
126        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));161        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
127162
128        assert_eq!(TemplateModule::collection(1).owner, 1);163        assert_eq!(TemplateModule::collection(1).owner, 1);
129        assert_eq!(TemplateModule::collection(2).owner, 2);164        assert_eq!(TemplateModule::collection(2).owner, 2);
149#[test]184#[test]
150fn balance_of() {185fn balance_of() {
151    new_test_ext().execute_with(|| {186    new_test_ext().execute_with(|| {
187
188        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
189        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
190        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
191
152        let size = 1024;192        let size = 1024;
153        let origin1 = Origin::signed(1);193        let origin1 = Origin::signed(1);
154        let origin2 = Origin::signed(2);194        let origin2 = Origin::signed(2);
155        let origin3 = Origin::signed(3);195        let origin3 = Origin::signed(3);
156196
157        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));197        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
158        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));198        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
159        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));199        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
160200
161        assert_eq!(TemplateModule::collection(1).owner, 1);201        assert_eq!(TemplateModule::collection(1).owner, 1);
162        assert_eq!(TemplateModule::collection(2).owner, 2);202        assert_eq!(TemplateModule::collection(2).owner, 2);
181#[test]221#[test]
182fn transfer() {222fn transfer() {
183    new_test_ext().execute_with(|| {223    new_test_ext().execute_with(|| {
224
225        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
226        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
227        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
228
184        let size = 1024;229        let size = 1024;
185        let origin1 = Origin::signed(1);230        let origin1 = Origin::signed(1);
186        let origin2 = Origin::signed(2);231        let origin2 = Origin::signed(2);
187        let origin3 = Origin::signed(3);232        let origin3 = Origin::signed(3);
188233
189        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));234        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
190        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));235        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
191        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));236        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
192237
193        assert_eq!(TemplateModule::collection(1).owner, 1);238        assert_eq!(TemplateModule::collection(1).owner, 1);
194        assert_eq!(TemplateModule::collection(2).owner, 2);239        assert_eq!(TemplateModule::collection(2).owner, 2);
214#[test]259#[test]
215fn approve() {260fn approve() {
216    new_test_ext().execute_with(|| {261    new_test_ext().execute_with(|| {
262
263        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
264        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
265        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
266
217        let size = 1024;267        let size = 1024;
218        let origin1 = Origin::signed(1);268        let origin1 = Origin::signed(1);
219        let origin2 = Origin::signed(2);269        let origin2 = Origin::signed(2);
220        let origin3 = Origin::signed(3);270        let origin3 = Origin::signed(3);
221271
222        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));272        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
223        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));273        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
224        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));274        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
225275
226        assert_eq!(TemplateModule::collection(1).owner, 1);276        assert_eq!(TemplateModule::collection(1).owner, 1);
227        assert_eq!(TemplateModule::collection(2).owner, 2);277        assert_eq!(TemplateModule::collection(2).owner, 2);
243#[test]293#[test]
244fn get_approved() {294fn get_approved() {
245    new_test_ext().execute_with(|| {295    new_test_ext().execute_with(|| {
296        
297        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
298        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
299        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
300
246        let size = 1024;301        let size = 1024;
247        let origin1 = Origin::signed(1);302        let origin1 = Origin::signed(1);
248        let origin2 = Origin::signed(2);303        let origin2 = Origin::signed(2);
249        let origin3 = Origin::signed(3);304        let origin3 = Origin::signed(3);
250305
251        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));306        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
252        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));307        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
253        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));308        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
254309
255        assert_eq!(TemplateModule::collection(1).owner, 1);310        assert_eq!(TemplateModule::collection(1).owner, 1);
256        assert_eq!(TemplateModule::collection(2).owner, 2);311        assert_eq!(TemplateModule::collection(2).owner, 2);
272#[test]327#[test]
273fn transfer_from() {328fn transfer_from() {
274    new_test_ext().execute_with(|| {329    new_test_ext().execute_with(|| {
330
331        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
332        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
333        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
334
275        let size = 1024;335        let size = 1024;
276        let origin1 = Origin::signed(1);336        let origin1 = Origin::signed(1);
277        let origin2 = Origin::signed(2);337        let origin2 = Origin::signed(2);
278        let origin3 = Origin::signed(3);338        let origin3 = Origin::signed(3);
279339
280        assert_ok!(TemplateModule::create_collection(origin1.clone(), size));340        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
281        assert_ok!(TemplateModule::create_collection(origin2.clone(), size));341        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
282        assert_ok!(TemplateModule::create_collection(origin3.clone(), size));342        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
283343
284        assert_eq!(TemplateModule::collection(1).owner, 1);344        assert_eq!(TemplateModule::collection(1).owner, 1);
285        assert_eq!(TemplateModule::collection(2).owner, 2);345        assert_eq!(TemplateModule::collection(2).owner, 2);
297        assert_ok!(TemplateModule::transfer_from(origin1.clone(), 1, 1, 2));357        assert_ok!(TemplateModule::transfer_from(origin1.clone(), 1, 1, 2));
298    });358    });
299}359}
360
361#[test]
362fn index_list() {
363    new_test_ext().execute_with(|| {
364
365        let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
366        let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
367        let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
368
369        let size = 1024;
370        let origin1 = Origin::signed(1);
371        let origin2 = Origin::signed(2);
372        let origin3 = Origin::signed(3);
373
374        assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
375        assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
376        assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
377
378        assert_eq!(TemplateModule::collection(1).owner, 1);
379        assert_eq!(TemplateModule::collection(2).owner, 2);
380        assert_eq!(TemplateModule::collection(3).owner, 3);
381        // create items
382        assert_ok!(TemplateModule::create_item(
383            origin1.clone(),
384            1,
385            [1, 1, 1].to_vec()
386        ));
387        assert_ok!(TemplateModule::create_item(
388            origin1.clone(),
389            1,
390            [1, 1, 2].to_vec()
391        ));
392        assert_ok!(TemplateModule::create_item(
393            origin1.clone(),
394            1,
395            [1, 2, 3].to_vec()
396        ));
397        assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 3);
398        // burn one
399        assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 2));
400        assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 2);
401        // burn another one
402        assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 3));
403        assert_eq!(TemplateModule::address_tokens((1, 1))[0], 1);
404    });
405}
406
300407