From aab6bcfe096efde10a61eb36aa9cbfff6318eecf Mon Sep 17 00:00:00 2001
From: PraneshASP <praneshas2000@gmail.com>
Date: Tue, 23 Apr 2024 12:03:02 +0530
Subject: [PATCH] chore: check if tokenAmount length is 1

---
 contracts/contracts/mocks/MockCCIPRouter.sol | 1 +
 1 file changed, 1 insertion(+)

diff --git a/contracts/contracts/mocks/MockCCIPRouter.sol b/contracts/contracts/mocks/MockCCIPRouter.sol
index 1bf23f3f2d..73fa2d1985 100644
--- 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,