Merge pull request #75 from aave/mike/test/directional-rounding

Mike/test/directional rounding

Author: matchv

aave-core/sources/aave-logic/generic_logic.move

@@ -229,4 +229,26 @@ module aave_pool::generic_logic {
 
         available_borrows_in_base_currency - total_debt_in_base_currency
     }
+
+    #[test_only]
+    /// Test helper function that exposes get_user_debt_in_base_currency for testing
+    public fun get_user_debt_in_base_currency_for_testing(
+        user: address,
+        reserve_data: Object<ReserveData>,
+        asset_price: u256,
+        asset_unit: u256
+    ): u256 {
+        get_user_debt_in_base_currency(user, reserve_data, asset_price, asset_unit)
+    }
+
+    #[test_only]
+    /// Test helper function that exposes get_user_balance_in_base_currency for testing
+    public fun get_user_balance_in_base_currency_for_testing(
+        user: address,
+        reserve_data: Object<ReserveData>,
+        asset_price: u256,
+        asset_unit: u256
+    ): u256 {
+        get_user_balance_in_base_currency(user, reserve_data, asset_price, asset_unit)
+    }
 }

aave-core/sources/aave-pool/pool_logic.move

@@ -473,4 +473,22 @@ module aave_pool::pool_logic {
             reserve_last_update_timestamp: 0
         }
     }
+
+    #[test_only]
+    /// Test helper function that exposes update_interest_rates_and_virtual_balance for testing
+    public fun update_interest_rates_and_virtual_balance_for_testing(
+        reserve_data: Object<ReserveData>,
+        reserve_cache: &ReserveCache,
+        reserve_address: address,
+        liquidity_added: u256,
+        liquidity_taken: u256
+    ) {
+        update_interest_rates_and_virtual_balance(
+            reserve_data,
+            reserve_cache,
+            reserve_address,
+            liquidity_added,
+            liquidity_taken
+        )
+    }
 }

aave-core/sources/aave-tokens/a_token_factory.move

@@ -746,4 +746,16 @@ module aave_pool::a_token_factory {
     public fun assert_token_exists_for_testing(metadata_address: address) acquires TokenMap {
         assert_token_exists(metadata_address);
     }
+
+    #[test_only]
+    /// Test helper function that exposes transfer_on_liquidation for testing
+    public fun transfer_on_liquidation_for_testing(
+        from: address,
+        to: address,
+        amount: u256,
+        index: u256,
+        metadata_address: address
+    ) acquires TokenMap {
+        transfer_on_liquidation(from, to, amount, index, metadata_address)
+    }
 }