unit test for lend and unlend, expose abi

Author: theboiledcorn

.gitignore

@@ -18,4 +18,5 @@ docs/
 .idea
 cache
 lib
-lib/chainlink-local
+lib/chainlink-local
+!out/CrossCredit.sol

src/CrossCredit.sol

@@ -192,7 +192,10 @@ contract CrossCredit is Ownable, ReentrancyGuard, CCIPReceiver, ICrossCredit {
         uint256 userTotalBorrowUSDValue = getTotalUSDValueOfUserByType(msg.sender, 2);
 
         (int256 price, uint8 priceFeedDecimals) = _getAssetPriceData(_asset);
-        uint256 currentUnlendAmountUSD = (_amount * uint256(price)) / (10 ** priceFeedDecimals);
+
+        uint8 localAssetDecimals = s_assetDecimals[_asset];
+
+        uint256 currentUnlendAmountUSD = (_amount * uint256(price)) / (10 ** localAssetDecimals);
 
         if (userTotalLendUSDValue < currentUnlendAmountUSD) revert Error.InsufficientCollateralRemaining();
 

test/CrossCreditBorrow.t.sol

@@ -8,7 +8,7 @@ contract CrossCreditBorrow is Base {
     function test_lendOnSourceAndBorrowOnDest() public {
         _setOraclePrices(1);
 
-        uint256 amountToLend = 1e6; // 1 USDC
+        uint256 amountToLend = 1e6; // ~1 USDC
         address assetToLend = address(sourceUSDC);
 
         // --- SOURCE CHAIN ACTIONS (Lend) ---
@@ -52,7 +52,7 @@ contract CrossCreditBorrow is Base {
 
         // Calculate max borrowable amount and the specific amount to borrow
         (,int usdPriceETH_dest, , ,) = priceFeedOnDestETH.latestRoundData();
-        uint256 LTV = crossCreditOnDest.LTV(); // Assuming LTV is 80 (0.8)
+        uint256 LTV = crossCreditOnDest.LTV();
         uint256 maxBorrowableUSD = (LTV * userTotalLendValue) / 100;
         uint256 targetBorrowUSD = ((LTV - 10) * userTotalLendValue) / 100; // Borrow 10% less than max for safety
 
@@ -99,4 +99,62 @@ contract CrossCreditBorrow is Base {
         assertEq(userTotalBorrowValueOnDest, userTotalBorrowValueOnSource, "Final Borrow USD value mismatch between chains");
     }
 
+    function test_lendOnSourceBorrowAndRepayOnDest() public {
+        _setOraclePrices(1);
+
+        uint256 amountToLend = 1e6; // ~1 USDC
+        address assetToLend = address(sourceUSDC);
+
+        vm.selectFork(sourceFork);
+
+        vm.startPrank(firstUser);
+        sourceUSDC.approve(address(crossCreditOnSource), amountToLend);
+        crossCreditOnSource.lend(amountToLend, assetToLend);
+        vm.stopPrank();
+
+        ccipLocalSimulatorFork.switchChainAndRouteMessage(destinationFork);
+
+        vm.selectFork(destinationFork);
+        uint userTotalLendValue = crossCreditOnDest.getTotalUSDValueOfUserByType(firstUser, 1);
+        (,int usdPriceETH_dest, , ,) = priceFeedOnDestETH.latestRoundData();
+        uint256 LTV = crossCreditOnDest.LTV();
+        uint256 maxBorrowableUSD = (LTV * userTotalLendValue) / 100;
+        uint256 targetBorrowUSD = ((LTV - 10) * userTotalLendValue) / 100; // Borrow 10% less than max for safety
+
+        uint256 nativeAssetDecimals = crossCreditOnDest.getAssetDecimalsOnSource(NATIVE_ASSET);
+        uint256 amountToBorrowRaw = (targetBorrowUSD * (10 ** nativeAssetDecimals)) / uint256(usdPriceETH_dest);
+
+        vm.startPrank(firstUser);
+        // BORROW
+        uint userBalOnDestBefore = firstUser.balance;
+        crossCreditOnDest.borrow(amountToBorrowRaw, NATIVE_ASSET);
+        uint userBalOnDestAfter = firstUser.balance;
+        vm.stopPrank();
+
+        ccipLocalSimulatorFork.switchChainAndRouteMessage(sourceFork);
+
+        vm.selectFork(destinationFork);
+        // REPAY
+        vm.startPrank(firstUser);
+        uint userTotalBorrowValue = crossCreditOnDest.getTotalUSDValueOfUserByType(firstUser, 2);
+        bool usdValCalc = userTotalBorrowValue < targetBorrowUSD ? targetBorrowUSD - userTotalBorrowValue < 10 : userTotalBorrowValue - targetBorrowUSD < 10;
+        assertEq(usdValCalc, true, "Borrowed USD value not within tolerance of target");
+
+        _setOraclePrices(1); // Price change
+        (,usdPriceETH_dest, , ,) = priceFeedOnDestETH.latestRoundData();
+        uint amountToRepay = (userTotalBorrowValue * (10 ** nativeAssetDecimals)) / uint256(usdPriceETH_dest);
+        vm.deal(firstUser, amountToRepay);
+        crossCreditOnDest.repay{value: amountToRepay}(amountToRepay, NATIVE_ASSET);
+        userTotalBorrowValue = crossCreditOnDest.getTotalUSDValueOfUserByType(firstUser, 2);
+
+        bool userBorrowValueRepaid = userTotalBorrowValue == 0 || userTotalBorrowValue < 10;
+        assertTrue(userBorrowValueRepaid, "Loan repayment not within range");
+        vm.stopPrank();
+        ccipLocalSimulatorFork.switchChainAndRouteMessage(sourceFork);
+
+        vm.selectFork(sourceFork);
+        userTotalBorrowValue = crossCreditOnSource.getTotalUSDValueOfUserByType(firstUser, 2);
+        userBorrowValueRepaid = userTotalBorrowValue == 0 || userTotalBorrowValue < 10;
+        assertTrue(userBorrowValueRepaid, "Loan repayment not within range");
+    }
 }

test/CrossCreditLend.t.sol

@@ -0,0 +1,43 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.13;
+
+import {Base} from "./Base.t.sol";
+import {console} from "forge-std/console.sol";
+
+
+contract CrossCreditLend is Base {
+    function test_lendAndUnlend() public {
+        _setOraclePrices(1);
+
+        uint256 amountToLend = 1e6; // ~1 USDC
+        address assetToLend = address(sourceUSDC);
+
+        vm.selectFork(sourceFork);
+
+        vm.startPrank(firstUser);
+        sourceUSDC.approve(address(crossCreditOnSource), amountToLend);
+        crossCreditOnSource.lend(amountToLend, assetToLend);
+        vm.stopPrank();
+
+        ccipLocalSimulatorFork.switchChainAndRouteMessage(destinationFork);
+
+        vm.selectFork(sourceFork);
+        vm.startPrank(firstUser);
+        uint userLendUSDValue = crossCreditOnSource.getTotalUSDValueOfUserByType(firstUser, 1);
+        console.log("User lend usd val before unlend: ", userLendUSDValue);
+        uint userBalBeforeUnlend = sourceUSDC.balanceOf(firstUser);
+        crossCreditOnSource.unlend(amountToLend, assetToLend);
+        uint userBalAfterUnlend = sourceUSDC.balanceOf(firstUser);
+        userLendUSDValue = crossCreditOnSource.getTotalUSDValueOfUserByType(firstUser, 1);
+
+        assertEq(userLendUSDValue, 0, "User Lend Value mismatch");
+        assertEq(userBalAfterUnlend, userBalBeforeUnlend + amountToLend, "User balance mismatch");
+        vm.stopPrank();
+        ccipLocalSimulatorFork.switchChainAndRouteMessage(destinationFork);
+
+        vm.selectFork(destinationFork);
+        userLendUSDValue = crossCreditOnDest.getTotalUSDValueOfUserByType(firstUser, 1);
+        assertEq(userLendUSDValue, 0, "User Lend Value mismatch");
+
+    }
+}