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

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}