astrox_macros 0.2.9

add user management and canister relation management for canister
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use std::collections::BTreeMap;
use candid::Principal;
use astrox_macros::user::User;

#[test]
fn owner_test(){
  let mut user = User::default();

  let owner_1 = "owner_1";
  let owner_1_principal = Principal::from_text("o2ivq-5dsz3-nba5d-pwbk2-hdd3i-vybeq-qfz35-rqg27-lyesf-xghzc-3ae").unwrap();


  assert!(!user.is_owner(owner_1_principal));

  user.owner_add(owner_1.to_string(), owner_1_principal);
  assert!(user.is_owner(owner_1_principal));

  user.owner_remove(owner_1_principal);
  assert!(!user.is_owner(owner_1_principal));

  user.owners_set(BTreeMap::from([(owner_1_principal, owner_1.to_string())]));
  assert!(user.is_owner(owner_1_principal));
}

#[test]
fn user_test(){
  let mut user = User::default();

  let owner_1 = "owner_1";
  let owner_1_principal = Principal::from_text("o2ivq-5dsz3-nba5d-pwbk2-hdd3i-vybeq-qfz35-rqg27-lyesf-xghzc-3ae").unwrap();

  let user_1 = "user_1";
  let user_1_principal = Principal::from_text("3zjeh-xtbtx-mwebn-37a43-7nbck-qgquk-xtrny-42ujn-gzaxw-ncbzw-kqe").unwrap();

  assert!(!user.is_user(owner_1_principal));
  assert!(!user.is_user(user_1_principal));

  user.owner_add(owner_1.to_string(), owner_1_principal);
  assert!(user.is_user(owner_1_principal));

  user.user_add(user_1.to_string(), user_1_principal);
  assert!(user.is_user(user_1_principal));

  user.user_remove(user_1_principal);
  assert!(!user.is_user(user_1_principal));

  user.users_set(BTreeMap::from([(user_1_principal, user_1.to_string())]));

  // owner_1 still in the owners map
  assert!(user.is_user(owner_1_principal));
  assert!(user.is_user(user_1_principal));
}

#[test]
fn op_test(){
  let mut user = User::default();

  let owner_1 = "owner_1";
  let owner_1_principal = Principal::from_text("o2ivq-5dsz3-nba5d-pwbk2-hdd3i-vybeq-qfz35-rqg27-lyesf-xghzc-3ae").unwrap();

  let op_1 = "op_1";
  let op_1_principal = Principal::from_text("3zjeh-xtbtx-mwebn-37a43-7nbck-qgquk-xtrny-42ujn-gzaxw-ncbzw-kqe").unwrap();


  assert!(!user.is_op(owner_1_principal));
  assert!(!user.is_op(op_1_principal));

  user.owner_add(owner_1.to_string(), owner_1_principal);
  assert!(user.is_op(owner_1_principal));

  user.op_add(op_1.to_string(), op_1_principal);
  assert!(user.is_op(op_1_principal));

  user.op_remove(op_1_principal);
  assert!(!user.is_op(op_1_principal));

  user.ops_set(BTreeMap::from([(op_1_principal, op_1.to_string())]));
  assert!(user.is_op(op_1_principal));
}