RACE.t.sol

// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.20;

import "forge-std/Test.sol";
import {CrossChainMessenger} from "src/RACE.sol";

contract Receiver {
    bool public shouldRevert;
    uint256 public lastReceived;

    function setShouldRevert(bool _shouldRevert) external {
        shouldRevert = _shouldRevert;
    }

    function receiveMessage(uint256 _lastReceived) external {
        require(!shouldRevert);
        lastReceived = _lastReceived;
    }
}

contract CrossChainMessengerTest is Test {
    CrossChainMessenger messenger;
    Receiver receiver;

    function setUp() public {
        messenger = new CrossChainMessenger();
        receiver = new Receiver();
    }

    function test_forceReplayMessagesOnlyExecutesFirstMessage() public {
        CrossChainMessenger.Message[] memory messages = new CrossChainMessenger.Message[](2);

        messages[0] = CrossChainMessenger.Message({
            sender: address(this),
            to: address(receiver),
            gas: 100_000,
            data: abi.encodeCall(Receiver.receiveMessage, (1))
        });

        messages[1] = CrossChainMessenger.Message({
            sender: address(this),
            to: address(receiver),
            gas: 100_000,
            data: abi.encodeCall(Receiver.receiveMessage, (2))
        });

        receiver.setShouldRevert(true);
        messenger.relayMessage(messages[0]);
        messenger.relayMessage(messages[1]);
        assertEq(receiver.lastReceived(), 0);

        receiver.setShouldRevert(false);
        messenger.forceReplayMessages(messages);
        assertEq(receiver.lastReceived(), 1);
    }
}