chore: check if tokenAmount length is 1

OriginProtocol · Apr 23, 2024 · aab6bcf · aab6bcf
1 parent b4a3b94
commit aab6bcf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/contracts/contracts/mocks/MockCCIPRouter.sol b/contracts/contracts/mocks/MockCCIPRouter.sol
@@ -10,6 +10,7 @@ contract MockCCIPRouter {
         uint64 destinationChainSelector,
         Client.EVM2AnyMessage calldata message
     ) external payable returns (bytes32) {
+        require(message.tokenAmounts.length == 1, "INVALID_TOKEN_COUNT");
         // Make sure tokens can be pulled from the zapper contract
         IERC20(message.tokenAmounts[0].token).transferFrom(
             msg.sender,