Initial commit

test/ccip/CCIPErrorHandling.t.sol

@@ -0,0 +1,115 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.19;
+
+import {Test, console} from "forge-std/Test.sol";
+import {CCIPSender} from "../../contracts/ccip/CCIPSender.sol";
+import {CCIPReceiver} from "../../contracts/ccip/CCIPReceiver.sol";
+import {IRouterClient} from "../../contracts/ccip/IRouterClient.sol";
+
+contract CCIPErrorHandlingTest is Test {
+    CCIPSender public sender;
+    CCIPReceiver public receiver;
+    address public mockRouter;
+    address public linkToken;
+    
+    uint64 constant TARGET_CHAIN_SELECTOR = 5009297550715157269;
+    
+    function setUp() public {
+        mockRouter = address(new MockRouter());
+        linkToken = address(new MockLinkToken());
+        address oracleAggregator = address(this); // Use test contract as aggregator
+        
+        sender = new CCIPSender(mockRouter, oracleAggregator, linkToken);
+        receiver = new CCIPReceiver(mockRouter, address(0));
+        
+        MockLinkToken(linkToken).mint(address(sender), 1000e18);
+    }
+    
+    function testInvalidMessageFormat() public {
+        bytes memory invalidData = "invalid";
+        
+        IRouterClient.Any2EVMMessage memory message = IRouterClient.Any2EVMMessage({
+            messageId: keccak256("test"),
+            sourceChainSelector: 138,
+            sender: abi.encode(address(sender)),
+            data: invalidData,
+            tokenAmounts: new IRouterClient.TokenAmount[](0)
+        });
+        
+        vm.prank(mockRouter);
+        // Should handle invalid format gracefully
+        try receiver.ccipReceive(message) {
+            // If it doesn't revert, that's also acceptable if error handling is implemented
+        } catch {
+            // Expected to revert on invalid format
+        }
+    }
+    
+    function testUnauthorizedSender() public {
+        bytes memory messageData = abi.encode(uint256(25000000000), uint256(1), uint256(block.timestamp));
+        
+        IRouterClient.Any2EVMMessage memory message = IRouterClient.Any2EVMMessage({
+            messageId: keccak256("test"),
+            sourceChainSelector: 138,
+            sender: abi.encode(address(0x123)), // Unauthorized sender
+            data: messageData,
+            tokenAmounts: new IRouterClient.TokenAmount[](0)
+        });
+        
+        vm.prank(mockRouter);
+        // Should reject unauthorized sender
+        receiver.ccipReceive(message);
+    }
+    
+    function testRouterOnlyAccess() public {
+        bytes memory messageData = abi.encode(uint256(25000000000), uint256(1), uint256(block.timestamp));
+        
+        IRouterClient.Any2EVMMessage memory message = IRouterClient.Any2EVMMessage({
+            messageId: keccak256("test"),
+            sourceChainSelector: 138,
+            sender: abi.encode(address(sender)),
+            data: messageData,
+            tokenAmounts: new IRouterClient.TokenAmount[](0)
+        });
+        
+        // Try to call from non-router address
+        vm.expectRevert("CCIPReceiver: only router");
+        receiver.ccipReceive(message);
+    }
+    
+    function testInsufficientLinkBalance() public {
+        // Add destination first
+        sender.addDestination(TARGET_CHAIN_SELECTOR, address(receiver));
+        
+        // Drain LINK balance
+        MockLinkToken(linkToken).transfer(address(0xdead), 1000e18);
+        
+        bytes memory messageData = abi.encode(uint256(25000000000), uint256(1), uint256(block.timestamp));
+        
+        // Should revert due to insufficient balance
+        vm.expectRevert();
+        sender.sendOracleUpdate(TARGET_CHAIN_SELECTOR, 25000000000, 1, block.timestamp);
+    }
+}
+
+contract MockRouter {
+    function send(uint64, bytes memory) external pure returns (bytes32) {
+        return bytes32(0);
+    }
+}
+
+contract MockLinkToken {
+    mapping(address => uint256) public balanceOf;
+    
+    function mint(address to, uint256 amount) external {
+        balanceOf[to] += amount;
+    }
+    
+    function transfer(address to, uint256 amount) external returns (bool) {
+        require(balanceOf[msg.sender] >= amount, "Insufficient balance");
+        balanceOf[msg.sender] -= amount;
+        balanceOf[to] += amount;
+        return true;
+    }
+}
+