smom-dbis-138/test/bridge/trustless/EdgeCases.t.sol

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.19;

import {Test, console} from "forge-std/Test.sol";
import "../../../contracts/bridge/trustless/BondManager.sol";
import "../../../contracts/bridge/trustless/ChallengeManager.sol";
import "../../../contracts/bridge/trustless/InboxETH.sol";
import "../../../contracts/bridge/trustless/LiquidityPoolETH.sol";
import "../../../contracts/bridge/trustless/EnhancedSwapRouter.sol";
import "@openzeppelin/contracts/token/ERC20/ERC20.sol";

contract MockERC20 is ERC20 {
    constructor(string memory name, string memory symbol) ERC20(name, symbol) {
        _mint(msg.sender, 1000000 ether);
    }
}

contract EdgeCasesTest is Test {
    BondManager public bondManager;
    ChallengeManager public challengeManager;
    LiquidityPoolETH public liquidityPool;
    InboxETH public inbox;
    EnhancedSwapRouter public swapRouter;
    
    MockERC20 public weth;
    MockERC20 public usdt;
    MockERC20 public usdc;
    MockERC20 public dai;

    address public deployer = address(0xDE0001);
    address public relayer = address(0x1111);
    address public user = address(0x2222);

    uint256 public constant BOND_MULTIPLIER = 1.1e18;
    uint256 public constant MIN_BOND = 1 ether;
    uint256 public constant CHALLENGE_WINDOW = 30 minutes;

    // Mock protocol addresses
    address public uniswapV3Router = address(0x1111111111111111111111111111111111111111);
    address public curve3Pool = address(0x2222222222222222222222222222222222222222);
    address public dodoexRouter = address(0x3333333333333333333333333333333333333333);
    address public balancerVault = address(0x4444444444444444444444444444444444444444);
    address public oneInchRouter = address(0x5555555555555555555555555555555555555555);

    function setUp() public {
        vm.startPrank(deployer);

        weth = new MockERC20("Wrapped Ether", "WETH");
        usdt = new MockERC20("Tether USD", "USDT");
        usdc = new MockERC20("USD Coin", "USDC");
        dai = new MockERC20("Dai Stablecoin", "DAI");

        bondManager = new BondManager(BOND_MULTIPLIER, MIN_BOND);
        challengeManager = new ChallengeManager(address(bondManager), CHALLENGE_WINDOW);
        liquidityPool = new LiquidityPoolETH(address(weth), 5, 11000);
        inbox = new InboxETH(address(bondManager), address(challengeManager), address(liquidityPool));
        
        swapRouter = new EnhancedSwapRouter(
            uniswapV3Router,
            curve3Pool,
            dodoexRouter,
            balancerVault,
            oneInchRouter,
            address(weth),
            address(usdt),
            address(usdc),
            address(dai)
        );

        liquidityPool.authorizeRelease(address(inbox));
        swapRouter.grantRole(swapRouter.ROUTING_MANAGER_ROLE(), deployer);

        vm.deal(relayer, 100 ether);
        vm.deal(user, 100 ether);
        vm.warp(1000);

        vm.stopPrank();
    }

    function testEdgeCase_ZeroAmountDeposit() public {
        vm.prank(relayer);
        vm.expectRevert();
        inbox.submitClaim{value: MIN_BOND}(
            1,
            address(0),
            0, // Zero amount
            user,
            ""
        );
    }

    function testEdgeCase_MaxUint256Amount() public {
        uint256 maxAmount = type(uint256).max;
        uint256 requiredBond = bondManager.getRequiredBond(maxAmount);
        
        // Should handle large amounts (may revert if bond exceeds available ETH)
        if (requiredBond <= address(relayer).balance) {
            vm.prank(relayer);
            // This might revert due to overflow in calculations, which is expected
            try inbox.submitClaim{value: requiredBond}(
                1,
                address(0),
                maxAmount,
                user,
                ""
            ) {
                // If it succeeds, verify bond was posted
                (address relayer, uint256 bondAmount, , , ) = bondManager.bonds(1);
                assertTrue(bondAmount > 0);
            } catch {
                // Revert is acceptable for extreme values
                assertTrue(true);
            }
        }
    }

    function testEdgeCase_ConcurrentClaims() public {
        uint256 amount = 1 ether;
        uint256 bond = bondManager.getRequiredBond(amount);
        
        // Submit multiple claims concurrently
        for (uint256 i = 1; i <= 5; i++) {
            vm.prank(relayer);
            inbox.submitClaim{value: bond}(
                i,
                address(0),
                amount,
                user,
                ""
            );
        }

        // Verify all bonds were posted
        for (uint256 i = 1; i <= 5; i++) {
            (, uint256 bondAmount, , , ) = bondManager.bonds(i);
            assertTrue(bondAmount > 0);
        }
    }

    function testEdgeCase_ProviderToggleDuringSwap() public {
        // Toggle provider off
        vm.prank(deployer);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.UniswapV3, false);
        
        assertFalse(swapRouter.providerEnabled(EnhancedSwapRouter.SwapProvider.UniswapV3));
        
        // Toggle back on
        vm.prank(deployer);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.UniswapV3, true);
        
        assertTrue(swapRouter.providerEnabled(EnhancedSwapRouter.SwapProvider.UniswapV3));
    }

    function testEdgeCase_EmptyRoutingConfig() public {
        // Set empty routing config (should revert)
        EnhancedSwapRouter.SwapProvider[] memory emptyProviders = new EnhancedSwapRouter.SwapProvider[](0);
        
        vm.prank(deployer);
        vm.expectRevert();
        swapRouter.setRoutingConfig(0, emptyProviders);
    }

    function testEdgeCase_InvalidSizeCategory() public {
        EnhancedSwapRouter.SwapProvider[] memory providers = new EnhancedSwapRouter.SwapProvider[](1);
        providers[0] = EnhancedSwapRouter.SwapProvider.UniswapV3;
        
        vm.prank(deployer);
        vm.expectRevert();
        swapRouter.setRoutingConfig(999, providers); // Invalid category
    }

    function testEdgeCase_AllProvidersDisabled() public {
        // Disable all providers
        vm.startPrank(deployer);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.UniswapV3, false);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.Dodoex, false);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.Balancer, false);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.Curve, false);
        swapRouter.setProviderEnabled(EnhancedSwapRouter.SwapProvider.OneInch, false);
        vm.stopPrank();

        // Verify all are disabled
        assertFalse(swapRouter.providerEnabled(EnhancedSwapRouter.SwapProvider.UniswapV3));
        assertFalse(swapRouter.providerEnabled(EnhancedSwapRouter.SwapProvider.Dodoex));
        assertFalse(swapRouter.providerEnabled(EnhancedSwapRouter.SwapProvider.Balancer));
    }

    function testEdgeCase_RepeatedBondRelease() public {
        uint256 depositId = 1;
        uint256 amount = 1 ether;
        uint256 bond = bondManager.getRequiredBond(amount);
        
        // Submit claim
        vm.prank(relayer);
        inbox.submitClaim{value: bond}(depositId, address(0), amount, user, "");
        
        // Wait for challenge window
        vm.warp(block.timestamp + CHALLENGE_WINDOW + 1);
        
        // Finalize
        challengeManager.finalizeClaim(depositId);
        
        // Try to release bond twice
        bondManager.releaseBond(depositId);
        
        vm.expectRevert();
        bondManager.releaseBond(depositId); // Should revert
    }

    function testEdgeCase_InvalidStablecoin() public {
        // Try to swap to invalid stablecoin
        address invalidToken = address(0x9999);
        
        vm.expectRevert();
        swapRouter.swapToStablecoin(
            LiquidityPoolETH.AssetType.WETH,
            invalidToken,
            1 ether,
            0,
            EnhancedSwapRouter.SwapProvider.UniswapV3
        );
    }

    function testEdgeCase_ZeroSlippageTolerance() public {
        // Swap with zero slippage tolerance (very strict)
        // This should work but might fail if price moves
        try swapRouter.swapToStablecoin(
            LiquidityPoolETH.AssetType.WETH,
            address(usdt),
            1 ether,
            1 ether, // 100% slippage tolerance (essentially no protection)
            EnhancedSwapRouter.SwapProvider.UniswapV3
        ) {
            // If it succeeds, that's fine
            assertTrue(true);
        } catch {
            // Expected to fail without actual DEX integration
            assertTrue(true);
        }
    }

    function testEdgeCase_MultipleBalancerPools() public {
        bytes32 pool1 = keccak256("pool1");
        bytes32 pool2 = keccak256("pool2");
        
        vm.prank(deployer);
        swapRouter.setBalancerPoolId(address(weth), address(usdt), pool1);
        
        vm.prank(deployer);
        swapRouter.setBalancerPoolId(address(weth), address(usdc), pool2);
        
        assertEq(swapRouter.balancerPoolIds(address(weth), address(usdt)), pool1);
        assertEq(swapRouter.balancerPoolIds(address(weth), address(usdc)), pool2);
    }
}