fix tests

Signed-off-by: Ihor Farion <ihor@umaproject.org>

Author: grasphoper

script/mintburn/oft/CreateSponsoredDeposit.s.sol

@@ -38,6 +38,7 @@ library DebugQuoteSignLib {
                 p.destinationDex,
                 p.lzReceiveGasLimit,
                 p.lzComposeGasLimit,
+                p.maxOftFeeBps,
                 p.accountCreationMode,
                 p.executionMode,
                 keccak256(p.actionData) // Hash the actionData to keep signature size reasonable
@@ -208,6 +209,7 @@ contract CreateSponsoredDeposit is Script, Config {
             destinationDex: HyperCoreLib.CORE_SPOT_DEX_ID,
             lzReceiveGasLimit: lzReceiveGasLimit,
             lzComposeGasLimit: lzComposeGasLimit,
+            maxOftFeeBps: 0,
             accountCreationMode: 0,
             executionMode: 0,
             actionData: ""
@@ -271,6 +273,7 @@ contract CreateSponsoredDeposit is Script, Config {
 
         bytes memory composeMsg = ComposeMsgCodec._encode(
             quote.signedParams.nonce,
+            quote.signedParams.amountLD,
             quote.signedParams.deadline,
             quote.signedParams.maxBpsToSponsor,
             quote.unsignedParams.maxUserSlippageBps,

test/evm/foundry/local/ComposeMsgCodec.t.sol

@@ -7,6 +7,7 @@ import { ComposeMsgCodec } from "contracts/periphery/mintburn/sponsored-oft/Comp
 contract ComposeMsgCodecTest is Test {
     function test_EncodeDecode() public {
         bytes32 nonce = keccak256("nonce");
+        uint256 amountLD = 100 ether;
         uint256 deadline = 1234567890;
         uint256 maxBpsToSponsor = 500;
         uint256 maxUserSlippageBps = 100;
@@ -19,6 +20,7 @@ contract ComposeMsgCodecTest is Test {
 
         bytes memory encoded = ComposeMsgCodec._encode(
             nonce,
+            amountLD,
             deadline,
             maxBpsToSponsor,
             maxUserSlippageBps,
@@ -31,6 +33,7 @@ contract ComposeMsgCodecTest is Test {
         );
 
         assertEq(ComposeMsgCodec._getNonce(encoded), nonce, "Nonce mismatch");
+        assertEq(ComposeMsgCodec._getAmountLD(encoded), amountLD, "AmountLD mismatch");
         assertEq(ComposeMsgCodec._getDeadline(encoded), deadline, "Deadline mismatch");
         assertEq(ComposeMsgCodec._getMaxBpsToSponsor(encoded), maxBpsToSponsor, "MaxBpsToSponsor mismatch");
         assertEq(ComposeMsgCodec._getMaxUserSlippageBps(encoded), maxUserSlippageBps, "MaxUserSlippageBps mismatch");
@@ -45,6 +48,7 @@ contract ComposeMsgCodecTest is Test {
 
     function testFuzz_EncodeDecode(
         bytes32 nonce,
+        uint256 amountLD,
         uint256 deadline,
         uint256 maxBpsToSponsor,
         uint256 maxUserSlippageBps,
@@ -57,6 +61,7 @@ contract ComposeMsgCodecTest is Test {
     ) public {
         bytes memory encoded = ComposeMsgCodec._encode(
             nonce,
+            amountLD,
             deadline,
             maxBpsToSponsor,
             maxUserSlippageBps,
@@ -69,6 +74,7 @@ contract ComposeMsgCodecTest is Test {
         );
 
         assertEq(ComposeMsgCodec._getNonce(encoded), nonce, "Nonce mismatch");
+        assertEq(ComposeMsgCodec._getAmountLD(encoded), amountLD, "AmountLD mismatch");
         assertEq(ComposeMsgCodec._getDeadline(encoded), deadline, "Deadline mismatch");
         assertEq(ComposeMsgCodec._getMaxBpsToSponsor(encoded), maxBpsToSponsor, "MaxBpsToSponsor mismatch");
         assertEq(ComposeMsgCodec._getMaxUserSlippageBps(encoded), maxUserSlippageBps, "MaxUserSlippageBps mismatch");
@@ -82,13 +88,13 @@ contract ComposeMsgCodecTest is Test {
     }
 
     function test_IsValidComposeMsgBytelength_Boundary() public pure {
-        // Minimum length is 352 bytes (9 static params + actionData offset + actionData length + 0 bytes actionData)
-        // 9 * 32 + 32 + 32 = 352 bytes
+        // Minimum length is 384 bytes (10 static params + actionData offset + actionData length + 0 bytes actionData)
+        // 10 * 32 + 32 + 32 = 384 bytes
 
-        bytes memory data = new bytes(352);
+        bytes memory data = new bytes(384);
         assertTrue(ComposeMsgCodec._isValidComposeMsgBytelength(data));
 
-        bytes memory tooShort = new bytes(351);
+        bytes memory tooShort = new bytes(383);
         assertFalse(ComposeMsgCodec._isValidComposeMsgBytelength(tooShort));
     }
 }

test/evm/foundry/local/SponsoredOFTSrcPeriphery.t.sol

@@ -80,6 +80,7 @@ contract SponsoredOFTSrcPeripheryTest is Test {
             destinationDex: HyperCoreLib.CORE_SPOT_DEX_ID,
             lzReceiveGasLimit: 500_000,
             lzComposeGasLimit: 500_000,
+            maxOftFeeBps: 0,
             accountCreationMode: uint8(0), // Standard
             executionMode: uint8(0), // DirectToCore
             actionData: ""
@@ -153,6 +154,7 @@ contract SponsoredOFTSrcPeripheryTest is Test {
         // Validate composeMsg encoding (layout from ComposeMsgCodec._encode)
         (
             bytes32 gotNonce,
+            uint256 gotAmountLD,
             uint256 gotDeadline,
             uint256 gotMaxBpsToSponsor,
             uint256 gotMaxUserSlippageBps,
@@ -164,10 +166,11 @@ contract SponsoredOFTSrcPeripheryTest is Test {
             bytes memory gotActionData
         ) = abi.decode(
                 spComposeMsg,
-                (bytes32, uint256, uint256, uint256, bytes32, bytes32, uint32, uint8, uint8, bytes)
+                (bytes32, uint256, uint256, uint256, uint256, bytes32, bytes32, uint32, uint8, uint8, bytes)
             );
 
         assertEq(gotNonce, nonce, "nonce mismatch");
+        assertEq(gotAmountLD, SEND_AMOUNT, "amountLD mismatch");
         assertEq(gotDeadline, deadline, "deadline mismatch");
         assertEq(gotMaxBpsToSponsor, 500, "maxBpsToSponsor mismatch");
         assertEq(gotMaxUserSlippageBps, 300, "maxUserSlippageBps mismatch");