smom-dbis-138/test/dex/DODOPMMIntegration.t.sol

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

import {Test, console} from "forge-std/Test.sol";
import "../../contracts/dex/DODOPMMIntegration.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 MockDodoPool {
    address public immutable baseToken;
    address public immutable quoteToken;

    constructor(address baseToken_, address quoteToken_) {
        baseToken = baseToken_;
        quoteToken = quoteToken_;
    }

    function _BASE_TOKEN_() external view returns (address) {
        return baseToken;
    }

    function _QUOTE_TOKEN_() external view returns (address) {
        return quoteToken;
    }
}

contract DODOPMMIntegrationTest is Test {
    DODOPMMIntegration public integration;
    address public dvm = address(0xdEaD);
    address public dodoApprove = address(0xD0D0);
    MockERC20 public officialUSDT;
    MockERC20 public officialUSDC;
    MockERC20 public compliantUSDT;
    MockERC20 public compliantUSDC;
    address public admin = address(0xAD);

    function setUp() public {
        officialUSDT = new MockERC20("USDT", "USDT");
        officialUSDC = new MockERC20("USDC", "USDC");
        compliantUSDT = new MockERC20("cUSDT", "cUSDT");
        compliantUSDC = new MockERC20("cUSDC", "cUSDC");
        vm.mockCall(
            dvm,
            abi.encodeWithSelector(
                DODOPMMIntegration.createPool.selector
            ),
            abi.encode(address(0x1001))
        );
        integration = new DODOPMMIntegration(
            admin,
            dvm,
            dodoApprove,
            address(officialUSDT),
            address(officialUSDC),
            address(compliantUSDT),
            address(compliantUSDC)
        );
        // admin already has POOL_MANAGER_ROLE from constructor
    }

    function testCreatePoolGeneric() public {
        address baseToken = address(0xB1);
        address quoteToken = address(0xB2);
        address mockPoolAddr = address(0xBeef);
        vm.mockCall(dvm, bytes(""), abi.encode(mockPoolAddr));
        vm.prank(admin);
        address pool = integration.createPool(
            baseToken,
            quoteToken,
            3,
            1e18,
            0.5e18,
            true
        );
        assertEq(pool, mockPoolAddr);
        assertEq(integration.pools(baseToken, quoteToken), mockPoolAddr);
        assertEq(integration.pools(quoteToken, baseToken), mockPoolAddr);
        assertTrue(integration.isRegisteredPool(mockPoolAddr));
    }

    function testCreatePoolRevertsSameToken() public {
        vm.prank(admin);
        vm.expectRevert("DODOPMMIntegration: same token");
        integration.createPool(
            address(officialUSDT),
            address(officialUSDT),
            3,
            1e18,
            0.5e18,
            true
        );
    }

    function testCreatePoolRevertsZeroBase() public {
        vm.prank(admin);
        vm.expectRevert("DODOPMMIntegration: zero base");
        integration.createPool(
            address(0),
            address(officialUSDT),
            3,
            1e18,
            0.5e18,
            true
        );
    }

    function testImportExistingPoolRecordsMappings() public {
        address baseToken = address(officialUSDT);
        address quoteToken = address(compliantUSDC);
        MockDodoPool pool = new MockDodoPool(baseToken, quoteToken);

        vm.prank(admin);
        integration.importExistingPool(
            address(pool),
            baseToken,
            quoteToken,
            3,
            1e18,
            0.5e18,
            false
        );

        assertEq(integration.pools(baseToken, quoteToken), address(pool));
        assertEq(integration.pools(quoteToken, baseToken), address(pool));
        assertTrue(integration.isRegisteredPool(address(pool)));

        DODOPMMIntegration.PoolConfig memory config = integration.getPoolConfig(address(pool));
        assertEq(config.baseToken, baseToken);
        assertEq(config.quoteToken, quoteToken);
    }

    function testImportExistingPoolAcceptsReverseHintAndNormalizes() public {
        address baseToken = address(officialUSDT);
        address quoteToken = address(compliantUSDC);
        MockDodoPool pool = new MockDodoPool(baseToken, quoteToken);

        vm.prank(admin);
        integration.importExistingPool(
            address(pool),
            quoteToken,
            baseToken,
            3,
            1e18,
            0.5e18,
            false
        );

        DODOPMMIntegration.PoolConfig memory config = integration.getPoolConfig(address(pool));
        assertEq(config.baseToken, baseToken);
        assertEq(config.quoteToken, quoteToken);
    }

    function testImportExistingPoolRevertsOnMismatch() public {
        MockDodoPool pool = new MockDodoPool(address(officialUSDT), address(compliantUSDC));

        vm.prank(admin);
        vm.expectRevert("DODOPMMIntegration: pool token mismatch");
        integration.importExistingPool(
            address(pool),
            address(officialUSDT),
            address(compliantUSDT),
            3,
            1e18,
            0.5e18,
            false
        );
    }
}