fix tests

lifinance · May 23, 2024 · e627280 · e627280
1 parent 7392ffb
commit e627280
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/test/solidity/Facets/StargateFacet.t.sol b/test/solidity/Facets/StargateFacet.t.sol
@@ -136,7 +136,7 @@ contract StargateFacetTest is TestBaseFacet {
     function initiateBridgeTxWithFacet(bool isNative) internal override {
         if (isNative) {
             stargateFacet.startBridgeTokensViaStargate{
-                value: bridgeData.minAmount + addToMessageValue
+                value: bridgeData.minAmount + nativeAddToMessageValue
             }(bridgeData, stargateData);
         } else {
             stargateFacet.startBridgeTokensViaStargate{
@@ -147,24 +147,22 @@ contract StargateFacetTest is TestBaseFacet {
 
     function initiateSwapAndBridgeTxWithFacet(bool) internal override {
         stargateFacet.swapAndStartBridgeTokensViaStargate{
-            value: addToMessageValue
+            value: nativeAddToMessageValue
         }(bridgeData, swapData, stargateData);
     }
 
     /// Additional Tests ///
 
     function test_CanBridgeNativeTokensWithDestinationCall()
         public
-        assertBalanceChange(
-            address(0),
-            USER_SENDER,
-            -int256((defaultNativeAmount))
-        )
         assertBalanceChange(address(0), USER_RECEIVER, 0)
         assertBalanceChange(ADDRESS_USDC, USER_SENDER, 0)
         assertBalanceChange(ADDRESS_DAI, USER_SENDER, 0)
     {
         vm.startPrank(USER_SENDER);
+        // Get initial balance
+        uint256 initialBalance = USER_SENDER.balance;
+
         // customize bridgeData
         bridgeData.sendingAssetId = address(0);
         bridgeData.minAmount = defaultNativeAmount;
@@ -184,7 +182,9 @@ contract StargateFacetTest is TestBaseFacet {
             10, // Optimism chainId
             stargateData
         );
-        stargateData.lzFee = nativeFees;
+        stargateData.lzFee = nativeAddToMessageValue =
+            nativeAddToMessageValue +
+            nativeFees;
 
         //prepare check for events
         vm.expectEmit(true, true, true, true, _facetTestContractAddress);
@@ -195,6 +195,13 @@ contract StargateFacetTest is TestBaseFacet {
         emit LiFiTransferStarted(bridgeData);
 
         initiateBridgeTxWithFacet(true);
+
+        // check balance after call and assert
+        uint256 currentBalance = USER_SENDER.balance;
+        uint256 expectedBalance = initialBalance -
+            defaultNativeAmount -
+            stargateData.lzFee;
+        assertEq(currentBalance, expectedBalance);
         vm.stopPrank();
     }
 
@@ -245,7 +252,7 @@ contract StargateFacetTest is TestBaseFacet {
         assertBalanceChange(
             address(0),
             USER_SENDER,
-            -int256((defaultNativeAmount))
+            -int256((defaultNativeAmount + nativeAddToMessageValue))
         )
         assertBalanceChange(address(0), USER_RECEIVER, 0)
         assertBalanceChange(ADDRESS_USDC, USER_SENDER, 0)
@@ -303,6 +310,7 @@ contract StargateFacetTest is TestBaseFacet {
 
         bridgeData.minAmount = amountOut;
         stargateData.minAmountLD = (amountOut * 90) / 100;
+        stargateData.lzFee = nativeAddToMessageValue;
         addToMessageValue = 0;
 
         delete swapData;