allowlist: WIP Add test initialization

Author: moCello

allowlist/tests/token.rs allowlist/tests/allowlist.rsallowlist/tests/token.rs renamed to allowlist/tests/allowlist.rs

@@ -5,31 +5,71 @@
 // Copyright (c) DUSK NETWORK. All rights reserved.
 
 use dusk_core::abi::ContractError;
-use dusk_core::abi::{ContractId, CONTRACT_ID_BYTES};
-use dusk_core::signatures::bls::{
-    PublicKey as AccountPublicKey, SecretKey as AccountSecretKey,
+// use dusk_core::abi::{ContractId, CONTRACT_ID_BYTES};
+// use dusk_core::signatures::bls::{
+//     PublicKey as AccountPublicKey, SecretKey as AccountSecretKey,
+// };
+// use dusk_core::transfer::data::ContractCall;
+// use dusk_core::transfer::MoonlightTransactionEvent;
+//
+// use rand::rngs::StdRng;
+// use rand::SeedableRng;
+//
+// use emt_core::token::error;
+// use emt_core::token::events;
+use emt_core::allowlist::{Address, Role};
+use emt_core::{
+    Account,
+    // AccountInfo, ZERO_ADDRESS
 };
-use dusk_core::transfer::data::ContractCall;
-use dusk_core::transfer::MoonlightTransactionEvent;
-
-use rand::rngs::StdRng;
-use rand::SeedableRng;
-
-use emt_core::token::error;
-use emt_core::token::events;
-use emt_core::{Account, AccountInfo, ZERO_ADDRESS};
 
 pub mod instantiate;
 use instantiate::{
-    TestSession, HOLDER_ID, INITIAL_BALANCE, INITIAL_HOLDER_BALANCE,
-    INITIAL_SUPPLY,
+    TestSession,
+    // HOLDER_ID, INITIAL_EMT_BALANCE
 };
 
 #[test]
 fn deploy() {
     TestSession::new();
 }
 
+#[test]
+fn is_allowed() -> Result<(), ContractError> {
+    let mut session = TestSession::new();
+
+    let unregistered: [u8; 32] = rand::random();
+    let unregistered = Address::from(&unregistered);
+    assert_eq!(session.allowlist_is_allowed(&unregistered)?.data, false);
+
+    let role: [u8; 32] = rand::random();
+    let role = Role::from(&role);
+    session.allowlist_register(&*TestSession::SK_0, unregistered, role)?;
+
+    session.allowlist_remove(&*TestSession::SK_0, &unregistered)?;
+
+    Ok(())
+}
+
+/// Test transfer of ownership to test account.
+#[test]
+fn transfer_ownership() {
+    let mut session = TestSession::new();
+    let new_ownership = Account::from(*TestSession::PK_1);
+
+    session
+        .allowlist_transfer_ownership(&*TestSession::SK_0, &new_ownership)
+        .expect("Call should pass");
+
+    assert_eq!(
+        session
+            .allowlist_ownership()
+            .expect("call should pass")
+            .data,
+        new_ownership
+    );
+}
+/*
 #[test]
 fn empty_account() {
     let mut session = TestSession::new();
@@ -450,23 +490,9 @@ fn transfer_from() {
         "The account should have the transferred amount subtracted from its allowance"
     );
 }
+*/
 
-/// Test transfer of ownership to test account.
-#[test]
-fn transfer_ownership() {
-    let mut session = TestSession::new();
-    let new_ownership = Account::from(*TestSession::PK_2);
-
-    session
-        .call_token::<_, ()>(
-            &*TestSession::SK_0,
-            "transfer_ownership",
-            &new_ownership,
-        )
-        .expect("Call should pass");
-
-    assert_eq!(session.ownership(), new_ownership);
-}
+/*
 
 /// Test TransferOwnership, RenounceOwnership with wrong ownership
 /// and check for correct error message.
@@ -968,3 +994,4 @@ fn test_sanctions() {
         )
         .expect("Transfer should succeed again");
 }
+*/