feat: its tests actual

move/governance/sources/governance/governance.move

@@ -123,37 +123,49 @@ use sui::test_scenario;
 #[test_only]
 use sui::test_utils;
 
+#[test_only]
+public fun new_for_testing(
+    trusted_source_chain: String,
+    trusted_source_address: String,
+    message_type: u256,
+    ctx: &mut TxContext,
+): Governance {
+    Governance {
+        id: object::new(ctx),
+        trusted_source_chain,
+        trusted_source_address,
+        message_type,
+        channel: channel::new(ctx),
+        caps: table::new<ID, UpgradeCap>(ctx),
+    }
+}
 #[test]
 fun test_new() {
     let trusted_source_chain = ascii::string(b"Axelar");
     let trusted_source_address = ascii::string(b"0x0");
     let message_type = 2;
     let mut ctx = tx_context::dummy();
     let package_id = object::id_from_bytes(
-        hex::decode(type_name::get<Governance>().get_address().into_bytes()),
+        hex::decode(
+            type_name::get<Governance>().get_address().into_bytes()
+        )
     );
     let upgrade_cap = package::test_publish(package_id, &mut ctx);
     let initial_owner = @0x1;
     let mut scenario = test_scenario::begin(initial_owner);
-    {
+    {   
         test_scenario::sender(&scenario);
-        new(
-            trusted_source_chain,
-            trusted_source_address,
-            message_type,
-            upgrade_cap,
-            &mut ctx,
-        );
+        new(trusted_source_chain, trusted_source_address, message_type, upgrade_cap, &mut ctx);
     };
 
     test_scenario::next_tx(&mut scenario, initial_owner);
-    {
+    {   
         let governance = test_scenario::take_shared<Governance>(&scenario);
         test_scenario::return_shared(governance);
     };
 
     test_scenario::end(scenario);
-}
+} 
 
 #[test]
 #[expected_failure(abort_code = ENotSelfUpgradeCap)]

move/its/sources/address_tracker.move

@@ -5,6 +5,9 @@ module its::address_tracker {
 
     use sui::table::{Self, Table};
 
+    // ------
+    // Errors
+    // ------
     /// Attempt to borrow a trusted address but it's not registered.
     const ENoAddress: u64 = 0;
 
@@ -13,6 +16,9 @@ module its::address_tracker {
         trusted_addresses: Table<String, String>
     }
 
+    // ------
+    // Getters
+    // ------
     /// Get the trusted address for a chain.
     public fun get_trusted_address(
         self: &InterchainAddressTracker, chain_name: String
@@ -55,6 +61,7 @@ module its::address_tracker {
         }
     }
 
+    // === Tests ===
     #[test]
     fun test_address_tracker() {
         let ctx = &mut sui::tx_context::dummy();
@@ -87,6 +94,22 @@ module its::address_tracker {
         assert!(self.is_trusted_address(chain2, address1) == true, 10);
         assert!(self.is_trusted_address(chain2, address2) == false, 11);
 
+        assert!(self.trusted_addresses.contains(chain1), 12);
+        assert!(self.trusted_addresses.contains(chain2), 13);
+
+        self.set_trusted_address(chain1, std::ascii::string(b""));
+        self.set_trusted_address(chain2, std::ascii::string(b""));
+
+        assert!(!self.trusted_addresses.contains(chain1), 14);
+        assert!(!self.trusted_addresses.contains(chain2), 15);
+
+
+        self.set_trusted_address(chain1, std::ascii::string(b""));
+        self.set_trusted_address(chain2, std::ascii::string(b""));
+
+        assert!(!self.trusted_addresses.contains(chain1), 16);
+        assert!(!self.trusted_addresses.contains(chain2), 17);
+
         sui::test_utils::destroy(self);
     }
 

move/its/sources/coin_info.move

@@ -68,6 +68,7 @@ module its::coin_info {
         &self.metadata
     }
 
+    // === Tests ===
     #[test_only]
     public fun drop<T>(coin_info: CoinInfo<T>) {
         let CoinInfo {name: _, symbol: _, decimals: _, remote_decimals: _, metadata } = coin_info;
@@ -77,4 +78,27 @@ module its::coin_info {
             metadata.destroy_none()
         }
     }
+
+    #[test]
+    fun test_from_metadata() {
+        let ctx = &mut tx_context::dummy();
+        let metadata = its::coin::create_metadata(b"Symbol", 8, ctx);
+        let metadata_bytes = sui::bcs::to_bytes(&metadata);
+
+        let name = metadata.get_name();
+        let symbol = metadata.get_symbol();
+        let decimals = metadata.get_decimals();
+        let remote_decimals = 31;
+
+        let coin_info = from_metadata(metadata, remote_decimals);
+
+        assert!(coin_info.name() == name, 0);
+        assert!(coin_info.symbol() == symbol, 1);
+        assert!(coin_info.decimals() == decimals, 2);
+        assert!(coin_info.remote_decimals() == remote_decimals, 3);
+        assert!(sui::bcs::to_bytes(coin_info.metadata().borrow()) == metadata_bytes, 4);
+        assert!(coin_info.scaling() == utils::pow(10, remote_decimals - decimals), 5);
+
+        sui::test_utils::destroy(coin_info);
+    }
 }

move/its/sources/coin_management.move

@@ -10,6 +10,9 @@ module its::coin_management {
 
     use its::flow_limit::{Self, FlowLimit};
 
+    // ------
+    // Errors
+    // ------
     /// Trying to add a distributor to a `CoinManagement` that does not
     /// have a `TreasuryCap`.
     const EDistributorNeedsTreasuryCap: u64 = 0;
@@ -26,6 +29,9 @@ module its::coin_management {
         dust: u256,
     }
 
+    // ------
+    // Public Functions to create CoinManagement
+    // ------
     /// Create a new `CoinManagement` with a `TreasuryCap`.
     /// This type of `CoinManagement` allows minting and burning of coins.
     public fun new_with_cap<T>(treasury_cap: TreasuryCap<T>): CoinManagement<T> {
@@ -54,6 +60,10 @@ module its::coin_management {
         }
     }
 
+
+    // ------
+    // Functions that modify CoinManagement
+    // ------
     /// Adds the distributor address to the `CoinManagement`.
     /// Only works for a `CoinManagement` with a `TreasuryCap`.
     public fun add_distributor<T>(self: &mut CoinManagement<T>, distributor: address) {
@@ -136,7 +146,7 @@ module its::coin_management {
         self.treasury_cap.is_some()
     }
 
-
+    // === Tests ===
     #[test_only]
     public struct COIN_MANAGEMENT has drop {}
 

move/its/sources/discovery.move

@@ -178,6 +178,7 @@ module its::discovery {
         )
     }
 
+    // === Tests ===
     #[test_only]
     fun get_initial_tx(self: &ITS): Transaction {
         let mut arg = vector[0];
@@ -209,7 +210,7 @@ module its::discovery {
     #[test]
     fun test_discovery_initial() {
         let ctx = &mut sui::tx_context::dummy();
-        let mut its = its::its::new();
+        let mut its = its::its::new_for_testing();
         let mut discovery = axelar_gateway::discovery::new(ctx);
 
         register_transaction(&mut its, &mut discovery);
@@ -224,7 +225,7 @@ module its::discovery {
     #[test]
     fun test_discovery_interchain_transfer() {
         let ctx = &mut sui::tx_context::dummy();
-        let mut its = its::its::new();
+        let mut its = its::its::new_for_testing();
         let mut discovery = axelar_gateway::discovery::new(ctx);
 
         register_transaction(&mut its, &mut discovery);
@@ -245,7 +246,7 @@ module its::discovery {
         let payload = writer.into_bytes();
 
         let type_arg = std::type_name::get<RelayerDiscovery>();
-        its.test_add_registered_coin_type(its::token_id::from_address(token_id), type_arg);
+        its.add_registered_coin_type_for_testing(its::token_id::from_address(token_id), type_arg);
         let tx_block = get_call_info(&its, payload);
 
         let mut reader = abi::new_reader(payload);
@@ -277,7 +278,7 @@ module its::discovery {
     #[test]
     fun test_discovery_interchain_transfer_with_data() {
         let ctx = &mut sui::tx_context::dummy();
-        let mut its = its::its::new();
+        let mut its = its::its::new_for_testing();
         let mut discovery = axelar_gateway::discovery::new(ctx);
 
         register_transaction(&mut its, &mut discovery);
@@ -305,7 +306,7 @@ module its::discovery {
             .write_bytes(data);
         let payload = writer.into_bytes();
 
-        its.test_add_registered_coin_type(its::token_id::from_address(token_id), std::type_name::get<RelayerDiscovery>());
+        its.add_registered_coin_type_for_testing(its::token_id::from_address(token_id), std::type_name::get<RelayerDiscovery>());
 
         let mut reader = abi::new_reader(payload);
         reader.skip_slot(); // skip message_type
@@ -319,7 +320,7 @@ module its::discovery {
     #[test]
     fun test_discovery_deploy_token() {
         let ctx = &mut sui::tx_context::dummy();
-        let mut its = its::its::new();
+        let mut its = its::its::new_for_testing();
         let mut discovery = axelar_gateway::discovery::new(ctx);
 
         register_transaction(&mut its, &mut discovery);
@@ -340,7 +341,7 @@ module its::discovery {
         let payload = writer.into_bytes();
 
         let type_arg = std::type_name::get<RelayerDiscovery>();
-        its.test_add_unregistered_coin_type(its::token_id::unregistered_token_id(&ascii::string(symbol), (decimals as u8)), type_arg);
+        its.add_unregistered_coin_type_for_testing(its::token_id::unregistered_token_id(&ascii::string(symbol), (decimals as u8)), type_arg);
         let tx_block = get_call_info(&its, payload);
 
         let mut reader = abi::new_reader(payload);