assetxContracts/test/YTLp.t.sol

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

import "forge-std/Test.sol";
import "../contracts/ytLp/tokens/USDY.sol";
import "../contracts/ytLp/tokens/YTLPToken.sol";
import "../contracts/ytLp/core/YTPriceFeed.sol";
import "../contracts/ytLp/core/YTVault.sol";
import "../contracts/ytLp/core/YTPoolManager.sol";
import "../contracts/ytLp/core/YTRewardRouter.sol";
import "../contracts/vault/YTAssetVault.sol";
import "../contracts/vault/YTAssetFactory.sol";
import "../contracts/ytLp/tokens/WUSD.sol";

contract YTLpTest is Test {
    address deployer;
    address user1;
    address user2;
    address user3;
    
    USDY usdy;
    YTLPToken ytlp;
    YTPriceFeed priceFeed;
    YTVault vault;
    YTPoolManager poolManager;
    YTRewardRouter router;
    WUSD wusd;
    YTAssetFactory factory;
    
    YTAssetVault ytTokenA;
    YTAssetVault ytTokenB;
    YTAssetVault ytTokenC;
    
    uint256 constant PRICE_PRECISION = 1e30;
    uint256 constant BASIS_POINTS = 10000;
    
    function setUp() public {
        deployer = address(this);
        user1 = address(0x1);
        user2 = address(0x2);
        user3 = address(0x3);
        
        vm.deal(user1, 100 ether);
        vm.deal(user2, 100 ether);
        vm.deal(user3, 100 ether);
        
        // 部署WUSD
        wusd = new WUSD();
        wusd.initialize("Wrapped USD", "WUSD");
        
        // 部署代币（可升级合约）
        usdy = new USDY();
        usdy.initialize();
        
        ytlp = new YTLPToken();
        ytlp.initialize();
        
        // 部署核心合约（可升级合约）
        priceFeed = new YTPriceFeed();
        priceFeed.initialize(address(wusd));
        
        vault = new YTVault();
        vault.initialize(address(usdy), address(priceFeed));
        
        poolManager = new YTPoolManager();
        poolManager.initialize(
            address(vault),
            address(usdy),
            address(ytlp),
            15 * 60
        );
        
        router = new YTRewardRouter();
        router.initialize(
            address(usdy),
            address(ytlp),
            address(poolManager),
            address(vault)
        );
        
        // 部署YTAssetVault实现合约
        YTAssetVault vaultImpl = new YTAssetVault();
        
        // 部署YTAssetFactory
        factory = new YTAssetFactory();
        factory.initialize(
            address(vaultImpl),
            1000000 ether // defaultHardCap
        );
        
        // 通过factory创建YTAssetVault代币
        address ytTokenAAddr = factory.createVault(
            "YT Token A",
            "YT-A",
            deployer, // manager
            1000000 ether, // hardCap
            address(wusd),
            block.timestamp + 365 days, // redemptionTime
            PRICE_PRECISION, // wusdPrice = 1.0
            PRICE_PRECISION  // ytPrice = 1.0
        );
        ytTokenA = YTAssetVault(ytTokenAAddr);
        
        address ytTokenBAddr = factory.createVault(
            "YT Token B",
            "YT-B",
            deployer,
            1000000 ether,
            address(wusd),
            block.timestamp + 365 days,
            PRICE_PRECISION,
            PRICE_PRECISION
        );
        ytTokenB = YTAssetVault(ytTokenBAddr);
        
        address ytTokenCAddr = factory.createVault(
            "YT Token C",
            "YT-C",
            deployer,
            1000000 ether,
            address(wusd),
            block.timestamp + 365 days,
            PRICE_PRECISION,
            PRICE_PRECISION
        );
        ytTokenC = YTAssetVault(ytTokenCAddr);
        
        // 配置权限
        usdy.addVault(address(vault));
        usdy.addVault(address(poolManager));
        ytlp.setMinter(address(poolManager), true);
        vault.setPoolManager(address(poolManager));
        vault.setSwapper(address(router), true);
        poolManager.setHandler(address(router), true);
        
        // 配置参数
        vault.setSwapFees(30, 4, 50, 20); // 0.3%, 0.04%, 动态税率
        vault.setDynamicFees(false); // 先关闭动态费率，便于精确测试
        vault.setMaxSwapSlippageBps(1000);
        priceFeed.setMaxPriceChangeBps(500);
        
        // 设置WUSD价格来源（使用ytTokenA）
        priceFeed.setWusdPriceSource(address(ytTokenA));
        
        // 配置白名单
        vault.setWhitelistedToken(address(ytTokenA), 18, 4000, 45000000 ether, false);
        vault.setWhitelistedToken(address(ytTokenB), 18, 3000, 35000000 ether, false);
        vault.setWhitelistedToken(address(ytTokenC), 18, 2000, 25000000 ether, false);
        
        // 初始化价格 $1.00
        priceFeed.forceUpdatePrice(address(ytTokenA), 1e30);
        priceFeed.forceUpdatePrice(address(ytTokenB), 1e30);
        priceFeed.forceUpdatePrice(address(ytTokenC), 1e30);
        
        // 不设置价差（便于精确计算）
        
        // 为测试用户铸造YT代币（需要先给合约WUSD，再depositYT）
        uint256 initialAmount = 10000 ether;
        
        // 给deployer一些WUSD用于购买YT
        wusd.mint(deployer, 30000 ether);
        
        // Deployer购买YT代币
        wusd.approve(address(ytTokenA), initialAmount);
        ytTokenA.depositYT(initialAmount);
        
        wusd.approve(address(ytTokenB), initialAmount);
        ytTokenB.depositYT(initialAmount);
        
        wusd.approve(address(ytTokenC), initialAmount);
        ytTokenC.depositYT(initialAmount);
        
        // 转账给用户
        ytTokenA.transfer(user1, 5000 ether);
        ytTokenB.transfer(user1, 5000 ether);
        ytTokenC.transfer(user1, 5000 ether);
        
        ytTokenA.transfer(user2, 3000 ether);
        ytTokenB.transfer(user2, 3000 ether);
        
        // 给用户一些WUSD（用于后续可能的操作）
        wusd.mint(user1, 10000 ether);
        wusd.mint(user2, 10000 ether);
        wusd.mint(user3, 10000 ether);
    }
    
    // ==================== 1. 部署和初始化测试 ====================
    
    function test_01_DeployContracts() public view {
        assertEq(usdy.name(), "YT USD");
        assertEq(usdy.symbol(), "USDY");
        assertEq(usdy.decimals(), 18);
        
        assertEq(ytlp.name(), "YT Liquidity Provider");
        assertEq(ytlp.symbol(), "ytLP");
        assertEq(ytlp.decimals(), 18);
        
        assertEq(vault.ytPoolManager(), address(poolManager));
        assertEq(poolManager.ytVault(), address(vault));
    }
    
    function test_02_ConfigurePermissions() public view {
        assertTrue(usdy.vaults(address(vault)));
        assertTrue(usdy.vaults(address(poolManager)));
        assertTrue(ytlp.isMinter(address(poolManager)));
        assertTrue(poolManager.isHandler(address(router)));
        assertTrue(vault.isSwapper(address(router)));
    }
    
    function test_03_ConfigureWhitelist() public view {
        assertTrue(vault.whitelistedTokens(address(ytTokenA)));
        assertTrue(vault.whitelistedTokens(address(ytTokenB)));
        assertTrue(vault.whitelistedTokens(address(ytTokenC)));
        
        assertEq(vault.tokenWeights(address(ytTokenA)), 4000);
        assertEq(vault.tokenWeights(address(ytTokenB)), 3000);
        assertEq(vault.tokenWeights(address(ytTokenC)), 2000);
        assertEq(vault.totalTokenWeights(), 9000);
        
        assertFalse(vault.stableTokens(address(ytTokenA)));
        assertTrue(vault.stableTokens(address(usdy))); // USDY被标记为稳定币
    }
    
    function test_04_ConfigureFees() public view {
        assertEq(vault.swapFeeBasisPoints(), 30); // 0.3%
        assertEq(vault.stableSwapFeeBasisPoints(), 4); // 0.04%
        assertEq(vault.taxBasisPoints(), 50);
        assertEq(vault.stableTaxBasisPoints(), 20);
        assertFalse(vault.hasDynamicFees()); // 已关闭便于测试
    }
    
    function test_05_YTAssetVaultBasics() public view {
        assertEq(ytTokenA.name(), "YT Token A");
        assertEq(ytTokenA.symbol(), "YT-A");
        assertEq(ytTokenA.ytPrice(), PRICE_PRECISION);
        assertEq(ytTokenA.wusdPrice(), PRICE_PRECISION);
    }
    
    // ==================== 2. 添加流动性测试（精确计算）====================
    
    function test_06_FirstAddLiquidity() public {
        uint256 depositAmount = 1000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), depositAmount);
        
        uint256 ytLPBefore = ytlp.balanceOf(user1);
        assertEq(ytLPBefore, 0);
        
        // 添加流动性
        uint256 ytLPReceived = router.addLiquidity(
            address(ytTokenA),
            depositAmount,
            0,
            0
        );
        
        vm.stopPrank();
        
        // 精确计算预期值（无价差，0.3%手续费）
        // 存入: 1000 YT-A @ $1.00
        // 手续费: 1000 × 0.3% = 3 个代币
        // 扣费后: 997 个代币
        // USDY价值: 997 × $1.00 = 997 USDY
        // 首次铸造: ytLP = USDY = 997 ether
        
        uint256 expectedYtLP = 997 ether;
        assertEq(ytLPReceived, expectedYtLP, "ytLP amount incorrect");
        assertEq(ytlp.balanceOf(user1), expectedYtLP, "user1 balance incorrect");
        assertEq(ytlp.totalSupply(), expectedYtLP, "total supply incorrect");
        
        // 验证池子状态
        assertEq(vault.poolAmounts(address(ytTokenA)), depositAmount, "pool amount incorrect");
        assertEq(vault.usdyAmounts(address(ytTokenA)), expectedYtLP, "usdy amount incorrect");
        
        // 验证ytLP价格
        uint256 ytLPPrice = poolManager.getPrice(true);
        // AUM = 1000 (池子有1000个代币) × $1.00 = $1000
        // Supply = 997 ytLP
        // Price = AUM * 1e18 / Supply (带18位精度)
        assertTrue(ytLPPrice > 1 ether, "ytLP price should be > $1");
    }
    
    function test_07_SecondAddLiquidity() public {
        // 用户1先添加
        uint256 firstAmount = 1000 ether;
        vm.startPrank(user1);
        ytTokenA.approve(address(router), firstAmount);
        router.addLiquidity(address(ytTokenA), firstAmount, 0, 0);
        vm.stopPrank();
        
        uint256 user1YtLP = ytlp.balanceOf(user1); // 997 ether
        
        // 用户2添加
        uint256 secondAmount = 2000 ether;
        vm.startPrank(user2);
        ytTokenB.approve(address(router), secondAmount);
        
        uint256 ytLPReceived = router.addLiquidity(
            address(ytTokenB),
            secondAmount,
            0,
            0
        );
        
        vm.stopPrank();
        
        // 精确计算
        uint256 expectedYtLP = 1988.018 ether;
        assertEq(ytLPReceived, expectedYtLP, "second add ytLP amount incorrect");
        assertEq(ytlp.balanceOf(user2), expectedYtLP, "user2 balance incorrect");
        assertEq(ytlp.totalSupply(), user1YtLP + expectedYtLP, "total supply incorrect");
    }
    
    function test_08_AddLiquiditySlippageProtection() public {
        uint256 amount = 1000 ether;
        uint256 tooHighMinYtLP = 1500 ether; // 设置过高的最小值
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        
        vm.expectRevert(abi.encodeWithSignature("InsufficientOutput()"));
        router.addLiquidity(
            address(ytTokenA),
            amount,
            0,
            tooHighMinYtLP
        );
        vm.stopPrank();
    }
    
    // ==================== 3. 移除流动性测试 ====================
    
    function test_09_RemoveLiquidity() public {
        // 先添加流动性
        uint256 addAmount = 1000 ether;
        vm.startPrank(user1);
        ytTokenA.approve(address(router), addAmount);
        router.addLiquidity(address(ytTokenA), addAmount, 0, 0);
        
        uint256 ytLPBalance = ytlp.balanceOf(user1); // 997 ether
        
        // 等待冷却期
        vm.warp(block.timestamp + 15 * 60 + 1);
        
        uint256 tokenBalanceBefore = ytTokenA.balanceOf(user1);
        
        // 移除流动性
        uint256 amountOut = router.removeLiquidity(
            address(ytTokenA),
            ytLPBalance,
            0,
            user1
        );
        
        vm.stopPrank();
        
        uint256 expectedOut = 997 ether;
        assertEq(amountOut, expectedOut, "remove liquidity amount incorrect");
        assertEq(ytTokenA.balanceOf(user1), tokenBalanceBefore + expectedOut, "user1 final balance incorrect");
        assertEq(ytlp.balanceOf(user1), 0, "ytLP should be burned");
        assertEq(ytlp.totalSupply(), 0, "ytLP supply should be 0");
    }
    
    function test_10_RemoveLiquidityCooldownProtection() public {
        uint256 amount = 1000 ether;
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        router.addLiquidity(address(ytTokenA), amount, 0, 0);
        
        uint256 ytLPBalance = ytlp.balanceOf(user1);
        
        // 不等待冷却期，直接移除
        vm.expectRevert(abi.encodeWithSignature("CooldownNotPassed()"));
        router.removeLiquidity(address(ytTokenA), ytLPBalance, 0, user1);
        
        vm.stopPrank();
    }
    
    // ==================== 4. Swap测试 ====================
    
    function test_11_SwapYTTokens() public {
        // 先为池子添加流动性
        uint256 liquidityAmount = 2000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), liquidityAmount);
        router.addLiquidity(address(ytTokenA), liquidityAmount, 0, 0);
        
        ytTokenB.approve(address(router), liquidityAmount);
        router.addLiquidity(address(ytTokenB), liquidityAmount, 0, 0);
        vm.stopPrank();
        
        // Swap测试
        uint256 swapAmount = 100 ether;
        
        vm.startPrank(user2);
        ytTokenA.approve(address(router), swapAmount);
        
        uint256 balanceBBefore = ytTokenB.balanceOf(user2);
        
        uint256 amountOut = router.swapYT(
            address(ytTokenA),
            address(ytTokenB),
            swapAmount,
            0,
            user2
        );
        
        vm.stopPrank();
        
        uint256 expectedOut = 99.7 ether;
        assertEq(amountOut, expectedOut, "swap amount incorrect");
        assertEq(ytTokenB.balanceOf(user2), balanceBBefore + expectedOut, "user2 balance incorrect");
    }
    
    function test_12_SwapSameTokenReverts() public {
        uint256 amount = 1000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        router.addLiquidity(address(ytTokenA), amount, 0, 0);
        
        ytTokenA.approve(address(router), 100 ether);
        
        vm.expectRevert(abi.encodeWithSignature("SameToken()"));
        router.swapYT(address(ytTokenA), address(ytTokenA), 100 ether, 0, user1);
        
        vm.stopPrank();
    }
    
    // ==================== 5. 价格测试 ====================
    
    function test_13_PriceWithoutSpread() public view {
        // 未设置价差时
        uint256 price = priceFeed.getPrice(address(ytTokenA), true);
        assertEq(price, 1e30, "price should be $1.00");
        
        uint256 maxPrice = priceFeed.getMaxPrice(address(ytTokenA));
        uint256 minPrice = priceFeed.getMinPrice(address(ytTokenA));
        
        assertEq(maxPrice, 1e30, "maxPrice should equal base price");
        assertEq(minPrice, 1e30, "minPrice should equal base price");
    }
    
    function test_14_PriceWithSpread() public {
        // 设置0.2%价差 (20 bps)
        priceFeed.setSpreadBasisPoints(address(ytTokenA), 20);
        
        uint256 basePrice = 1e30; // $1.00
        uint256 maxPrice = priceFeed.getMaxPrice(address(ytTokenA));
        uint256 minPrice = priceFeed.getMinPrice(address(ytTokenA));
        
        // MaxPrice = $1.00 × 1.002 = $1.002
        uint256 expectedMax = basePrice * 10020 / 10000;
        assertEq(maxPrice, expectedMax, "maxPrice with spread incorrect");
        
        // MinPrice = $1.00 × 0.998 = $0.998
        uint256 expectedMin = basePrice * 9980 / 10000;
        assertEq(minPrice, expectedMin, "minPrice with spread incorrect");
        
        // 清除价差设置
        priceFeed.setSpreadBasisPoints(address(ytTokenA), 0);
    }
    
    function test_15_WUSDPriceFromVault() public view {
            // WUSD价格应该从ytTokenA读取
        uint256 wusdPrice = priceFeed.getPrice(address(wusd), true);

        // 应该等于ytTokenA的wusdPrice
        assertEq(wusdPrice, PRICE_PRECISION, "WUSD price should be 1.0");
    }
    
    // ==================== 6. YTAssetVault特定功能测试 ====================
    
    function test_16_UpdateYTPrices() public {
        uint256 newWusdPrice = 1.01e30; // $1.01
        uint256 newYtPrice = 1.05e30;    // $1.05
        
        // 通过factory更新价格
        factory.updateVaultPrices(address(ytTokenA), newWusdPrice, newYtPrice);
        
        assertEq(ytTokenA.wusdPrice(), newWusdPrice, "wusdPrice update failed");
        assertEq(ytTokenA.ytPrice(), newYtPrice, "ytPrice update failed");
        
        // 重置价格
        factory.updateVaultPrices(address(ytTokenA), PRICE_PRECISION, PRICE_PRECISION);
    }
    
    function test_17_BuyYTWithWUSD() public {
        uint256 wusdAmount = 1000 ether;
        
        vm.startPrank(user1);
        wusd.approve(address(ytTokenA), wusdAmount);
        
        uint256 ytBefore = ytTokenA.balanceOf(user1);
        uint256 ytReceived = ytTokenA.depositYT(wusdAmount);
        uint256 ytAfter = ytTokenA.balanceOf(user1);
        
        vm.stopPrank();
        
        // 价格都是1.0，应该1:1兑换
        assertEq(ytReceived, wusdAmount, "YT amount should equal WUSD amount");
        assertEq(ytAfter - ytBefore, wusdAmount, "YT balance incorrect");
    }
    
    function test_18_HardCapProtection() public {
        // 获取当前totalSupply
        uint256 currentSupply = ytTokenA.totalSupply(); // 10000 ether
        
        // 通过factory设置hardCap为当前供应量 + 500（允许少量增加）
        uint256 newHardCap = currentSupply + 500 ether;
        factory.setHardCap(address(ytTokenA), newHardCap);
        
        vm.startPrank(user1);
        wusd.approve(address(ytTokenA), 1000 ether);
        
        // 尝试存入1000 ether会超过hardCap，应该revert
        vm.expectRevert(abi.encodeWithSignature("HardCapExceeded()"));
        ytTokenA.depositYT(1000 ether);
        
        vm.stopPrank();
        
        // 恢复hardCap
        factory.setHardCap(address(ytTokenA), 1000000 ether);
    }
    
    // ==================== 7. 权限测试 ====================
    
    function test_19_OnlyFactoryCanUpdatePrices() public {
        vm.startPrank(user1);
        
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        ytTokenA.updatePrices(1e30, 1e30);
        
        vm.stopPrank();
    }
    
    function test_20_OnlyGovCanSetWhitelist() public {
        vm.startPrank(user1);
        
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        vault.setWhitelistedToken(address(0x123), 18, 1000, 1000000 ether, false);
        
        vm.stopPrank();
    }
    
    // ==================== 8. 完整流程测试 ====================
    
    function test_21_CompleteFlow() public {
        console.log("=== Complete Flow Test ===");
        
        // 步骤1: User1添加YT-A流动性
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        uint256 ytLP1 = router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        console.log("User1 added 1000 YT-A, received ytLP:", ytLP1);
        assertEq(ytLP1, 997 ether);
        vm.stopPrank();
        
        // 步骤2: User1添加YT-B流动性
        vm.startPrank(user1);
        ytTokenB.approve(address(router), 1000 ether);
        uint256 ytLP1b = router.addLiquidity(address(ytTokenB), 1000 ether, 0, 0);
        console.log("User1 added 1000 YT-B, received ytLP:", ytLP1b);
        assertEq(ytLP1b, 994.009 ether);
        vm.stopPrank();
        
        uint256 totalYtLP = ytlp.balanceOf(user1);
        console.log("User1 total ytLP:", totalYtLP);
        
        // 步骤3: User2执行Swap
        vm.startPrank(user2);
        ytTokenA.approve(address(router), 100 ether);
        uint256 swapOut = router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        console.log("User2 swapped 100 YT-A, received YT-B:", swapOut);
        assertEq(swapOut, 99.7 ether);
        vm.stopPrank();
        
        // 步骤4: 等待冷却期后，User1移除流动性
        vm.warp(block.timestamp + 16 * 60);
        
        vm.startPrank(user1);
        uint256 removeAmount = totalYtLP / 2; // 移除一半
        uint256 tokenOut = router.removeLiquidity(address(ytTokenA), removeAmount, 0, user1);
        console.log("User1 removed half ytLP, received YT-A:", tokenOut);
        vm.stopPrank();
        
        assertTrue(tokenOut > 990 ether && tokenOut < 1000 ether, "token out should be around 997");
    }
    
    // ==================== 9. 手续费测试 ====================
    
    function test_22_SwapFeesAccumulation() public {
        // 添加初始流动性
        uint256 liquidityAmount = 2000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), liquidityAmount);
        router.addLiquidity(address(ytTokenA), liquidityAmount, 0, 0);
        
        ytTokenB.approve(address(router), liquidityAmount);
        router.addLiquidity(address(ytTokenB), liquidityAmount, 0, 0);
        
        uint256 ytLPBefore = ytlp.balanceOf(user1);
        uint256 priceBefore = poolManager.getPrice(true);
        
        vm.stopPrank();
        
        // 执行swap累积手续费（使用user2）
        uint256 swapAmount = 500 ether;
        
        vm.startPrank(user2);
        
        // Swap 1: YT-A → YT-B
        ytTokenA.approve(address(router), swapAmount);
        router.swapYT(address(ytTokenA), address(ytTokenB), swapAmount, 0, user2);
        
        // Swap 2: YT-B → YT-A
        ytTokenB.approve(address(router), swapAmount);
        router.swapYT(address(ytTokenB), address(ytTokenA), swapAmount, 0, user2);
        
        vm.stopPrank();
        
        uint256 priceAfter = poolManager.getPrice(true);
        
        // ytLP价格应该增长（手续费留在池中）
        assertTrue(priceAfter > priceBefore, "ytLP price should increase");
        
        // user1的ytLP数量不变
        assertEq(ytlp.balanceOf(user1), ytLPBefore, "ytLP balance should not change");
    }
    
    function test_23_GetSwapFeeBasisPoints() public {
        uint256 usdyAmount = 1000 ether;
        
        // YT代币之间互换
        uint256 feeBps = vault.getSwapFeeBasisPoints(
            address(ytTokenA),
            address(ytTokenB),
            usdyAmount
        );
        assertEq(feeBps, 30, "YT swap fee should be 30 bps");
        
        // 赎回费率
        uint256 redemptionFeeBps = vault.getRedemptionFeeBasisPoints(
            address(ytTokenA),
            usdyAmount
        );
        assertEq(redemptionFeeBps, 30, "redemption fee should be 30 bps");
    }
    
    // ==================== 10. 白名单管理测试 ====================
    
    function test_24_AddWhitelistToken() public {
        // 通过factory创建新的YTAssetVault
        address ytTokenDAddr = factory.createVault(
            "YT Token D",
            "YT-D",
            deployer,
            1000000 ether,
            address(wusd),
            block.timestamp + 365 days,
            PRICE_PRECISION,
            PRICE_PRECISION
        );
        YTAssetVault ytTokenD = YTAssetVault(ytTokenDAddr);
        
        // 铸造一些YT-D
        wusd.mint(deployer, 1000 ether);
        wusd.approve(address(ytTokenD), 1000 ether);
        ytTokenD.depositYT(1000 ether);
        
        // 添加到白名单
        vault.setWhitelistedToken(address(ytTokenD), 18, 1000, 10000000 ether, false);
        
        // 验证
        assertTrue(vault.whitelistedTokens(address(ytTokenD)), "should be whitelisted");
        assertEq(vault.tokenWeights(address(ytTokenD)), 1000, "weight incorrect");
        assertEq(vault.totalTokenWeights(), 10000, "total weight incorrect");
        
        // 初始化价格
        priceFeed.forceUpdatePrice(address(ytTokenD), 1e30);
        
        // 验证可以添加流动性
        vm.startPrank(deployer);
        ytTokenD.approve(address(router), 100 ether);
        uint256 ytLPReceived = router.addLiquidity(address(ytTokenD), 100 ether, 0, 0);
        vm.stopPrank();
        
        assertEq(ytLPReceived, 99.7 ether, "first liquidity for new token incorrect");
    }
    
    function test_25_RemoveWhitelistToken() public {
        // 确保池子是空的
        assertEq(vault.poolAmounts(address(ytTokenC)), 0, "pool should be empty");
        
        uint256 weightBefore = vault.totalTokenWeights();
        
        // 移除白名单
        vault.clearWhitelistedToken(address(ytTokenC));
        
        // 验证
        assertFalse(vault.whitelistedTokens(address(ytTokenC)), "should not be whitelisted");
        assertEq(vault.tokenWeights(address(ytTokenC)), 0, "weight should be 0");
        assertEq(vault.totalTokenWeights(), weightBefore - 2000, "total weight incorrect");
        
        // 验证无法添加流动性
        vm.startPrank(user1);
        ytTokenC.approve(address(router), 100 ether);
        
        vm.expectRevert(abi.encodeWithSignature("TokenNotWhitelisted()"));
        router.addLiquidity(address(ytTokenC), 100 ether, 0, 0);
        
        vm.stopPrank();
    }
    
    function test_26_UpdateTokenWeight() public {
        uint256 oldWeight = vault.tokenWeights(address(ytTokenA));
        assertEq(oldWeight, 4000);
        
        // 更新权重
        vault.setWhitelistedToken(address(ytTokenA), 18, 5000, 45000000 ether, false);
        
        // 验证
        assertEq(vault.tokenWeights(address(ytTokenA)), 5000, "updated weight incorrect");
        assertEq(vault.totalTokenWeights(), 10000, "total weight after update incorrect");
    }
    
    // ==================== 11. 查询函数测试 ====================
    
    function test_27_GetPoolValue() public {
        vm.startPrank(user1);
        
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        
        ytTokenB.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenB), 2000 ether, 0, 0);
        
        vm.stopPrank();
        
        // 获取池子总价值
        uint256 poolValue = vault.getPoolValue(true);
        
        // 池中有: 1000 YT-A + 2000 YT-B = $3000
        uint256 expectedValue = 3000 ether;
        assertEq(poolValue, expectedValue, "pool value incorrect");
    }
    
    function test_28_GetTargetUsdyAmount() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        uint256 totalUsdy = usdy.totalSupply();
        uint256 targetUsdy = vault.getTargetUsdyAmount(address(ytTokenA));
        
        // YT-A权重 4000, 总权重 9000
        uint256 expectedTarget = totalUsdy * 4000 / 9000;
        assertEq(targetUsdy, expectedTarget, "target usdy amount incorrect");
    }
    
    function test_29_GetAccountValue() public {
        uint256 amount = 1000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        router.addLiquidity(address(ytTokenA), amount, 0, 0);
        vm.stopPrank();
        
        uint256 accountValue = router.getAccountValue(user1);
        
        // 账户价值应该接近1000 USDY
        assertTrue(accountValue >= 995 ether && accountValue <= 1005 ether, "account value should be around 1000");
    }
    
    // ==================== 12. 动态手续费测试 ====================
    
    function test_30_DynamicFeesDisabled() public {
        assertFalse(vault.hasDynamicFees());
        
        uint256 feeBps = vault.getFeeBasisPoints(
            address(ytTokenA),
            1000 ether,
            30,
            50,
            true
        );
        
        assertEq(feeBps, 30, "should return base fee when dynamic disabled");
    }
    
    function test_31_DynamicFeesEnabled() public {
        vault.setDynamicFees(true);
        
        vm.startPrank(user1);
        
        // 大量添加YT-A
        ytTokenA.approve(address(router), 3000 ether);
        router.addLiquidity(address(ytTokenA), 3000 ether, 0, 0);
        
        // 少量添加YT-B
        ytTokenB.approve(address(router), 500 ether);
        router.addLiquidity(address(ytTokenB), 500 ether, 0, 0);
        
        vm.stopPrank();
        
        uint256 usdyAmount = 100 ether;
        
        // YT-A → YT-B (恶化平衡，费率更高)
        uint256 feeHigher = vault.getSwapFeeBasisPoints(
            address(ytTokenA),
            address(ytTokenB),
            usdyAmount
        );
        
        // YT-B → YT-A (改善平衡，费率更低)
        uint256 feeLower = vault.getSwapFeeBasisPoints(
            address(ytTokenB),
            address(ytTokenA),
            usdyAmount
        );
        
        assertTrue(feeHigher > 30, "fee should be higher when worsening balance");
        assertTrue(feeLower < 30, "fee should be lower when improving balance");
        
        vault.setDynamicFees(false);
    }
    
    // ==================== 13. 价格预言机测试 ====================
    
    function test_32_SetSpreadBasisPoints() public {
        uint256 spreadBps = 20;
        
        priceFeed.setSpreadBasisPoints(address(ytTokenA), spreadBps);
        
        assertEq(priceFeed.spreadBasisPoints(address(ytTokenA)), spreadBps);
    }
    
    function test_33_SpreadBasisPointsTooHigh() public {
        uint256 tooHighSpread = 300; // 3% > 最大2%
        
        vm.expectRevert(abi.encodeWithSignature("SpreadTooHigh()"));
        priceFeed.setSpreadBasisPoints(address(ytTokenA), tooHighSpread);
    }
    
    function test_34_BatchSetSpread() public {
        address[] memory tokens = new address[](3);
        tokens[0] = address(ytTokenA);
        tokens[1] = address(ytTokenB);
        tokens[2] = address(ytTokenC);
        
        uint256[] memory spreads = new uint256[](3);
        spreads[0] = 10;
        spreads[1] = 20;
        spreads[2] = 30;
        
        priceFeed.setSpreadBasisPointsForMultiple(tokens, spreads);
        
        assertEq(priceFeed.spreadBasisPoints(address(ytTokenA)), 10);
        assertEq(priceFeed.spreadBasisPoints(address(ytTokenB)), 20);
        assertEq(priceFeed.spreadBasisPoints(address(ytTokenC)), 30);
        
        // 清除
        spreads[0] = 0;
        spreads[1] = 0;
        spreads[2] = 0;
        priceFeed.setSpreadBasisPointsForMultiple(tokens, spreads);
    }
    
    function test_35_PriceProtectionMaxChange() public {
        priceFeed.forceUpdatePrice(address(ytTokenA), 1e30);
        
        uint256 tooHighPrice = 1.06e30; // +6%
        
        priceFeed.forceUpdatePrice(address(ytTokenA), tooHighPrice);
        
        assertEq(priceFeed.maxPriceChangeBps(), 500, "max change should be 5%");
    }
    
    // ==================== 14. AUM计算测试 ====================
    
    function test_36_GetAumWithMaximise() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        uint256 aumMax = poolManager.getAumInUsdy(true);
        uint256 aumMin = poolManager.getAumInUsdy(false);
        
        // 无价差时，两者应该相等
        assertEq(aumMax, aumMin, "aum should be equal without spread");
        assertEq(aumMax, 1000 ether, "aum should be $1000");
    }
    
    function test_37_GetAumWithSpread() public {
        priceFeed.setSpreadBasisPoints(address(ytTokenA), 20); // 0.2%
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        uint256 aumMax = poolManager.getAumInUsdy(true);
        uint256 aumMin = poolManager.getAumInUsdy(false);
        
        assertEq(aumMax, 1002 ether, "aum max with spread incorrect");
        assertEq(aumMin, 998 ether, "aum min with spread incorrect");
        
        priceFeed.setSpreadBasisPoints(address(ytTokenA), 0);
    }
    
    // ==================== 15. 多用户场景测试 ====================
    
    function test_38_MultipleUsersAddLiquidity() public {
        // User1 添加
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        uint256 ytLP1 = router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        // User2 添加
        vm.startPrank(user2);
        ytTokenA.approve(address(router), 2000 ether);
        uint256 ytLP2 = router.addLiquidity(address(ytTokenA), 2000 ether, 0, 0);
        vm.stopPrank();
        
        assertEq(ytLP1, 997 ether, "user1 ytLP incorrect");
        assertEq(ytLP2, 1988.018 ether, "user2 ytLP incorrect");
        
        // 份额比例
        uint256 total = ytlp.totalSupply();
        uint256 user1Share = ytLP1 * 10000 / total;
        uint256 user2Share = ytLP2 * 10000 / total;
        
        assertApproxEqAbs(user1Share, 3340, 1, "user1 share incorrect");
        assertApproxEqAbs(user2Share, 6660, 1, "user2 share incorrect");
    }
    
    function test_39_RemoveLiquidityPartial() public {
        uint256 addAmount = 1000 ether;
        vm.startPrank(user1);
        ytTokenA.approve(address(router), addAmount);
        router.addLiquidity(address(ytTokenA), addAmount, 0, 0);
        
        uint256 ytLPBalance = ytlp.balanceOf(user1);
        uint256 removeAmount = ytLPBalance / 2;
        
        vm.warp(block.timestamp + 15 * 60 + 1);
        
        uint256 amountOut = router.removeLiquidity(
            address(ytTokenA),
            removeAmount,
            0,
            user1
        );
        
        vm.stopPrank();
        
        uint256 expectedOut = 498.5 ether;
        assertEq(amountOut, expectedOut, "partial remove amount incorrect");
        assertEq(ytlp.balanceOf(user1), removeAmount, "remaining ytLP incorrect");
    }
    
    // ==================== 16. 安全功能测试 ====================
    
    function test_40_EmergencyMode() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        vault.setEmergencyMode(true);
        
        vm.startPrank(user2);
        ytTokenA.approve(address(router), 100 ether);
        
        vm.expectRevert(abi.encodeWithSignature("EmergencyMode()"));
        router.addLiquidity(address(ytTokenA), 100 ether, 0, 0);
        
        vm.expectRevert(abi.encodeWithSignature("EmergencyMode()"));
        router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        
        vm.stopPrank();
        
        vault.setEmergencyMode(false);
    }
    
    function test_41_SwapDisabled() public {
        vault.setSwapEnabled(false);
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        
        vm.expectRevert(abi.encodeWithSignature("SwapDisabled()"));
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        
        vm.stopPrank();
        
        vault.setSwapEnabled(true);
    }
    
    function test_42_MaxSwapAmount() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenA), 2000 ether, 0, 0);
        
        ytTokenB.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenB), 2000 ether, 0, 0);
        vm.stopPrank();
        
        vault.setMaxSwapAmount(address(ytTokenA), 50 ether);
        
        vm.startPrank(user2);
        ytTokenA.approve(address(router), 100 ether);
        
        vm.expectRevert(abi.encodeWithSignature("AmountExceedsLimit()"));
        router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        
        vm.stopPrank();
        
        vault.setMaxSwapAmount(address(ytTokenA), 0);
    }
    
    // ==================== 17. 边界条件测试 ====================
    
    function test_43_AddZeroAmountReverts() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 0);
        
        vm.expectRevert(abi.encodeWithSignature("InvalidAmount()"));
        router.addLiquidity(address(ytTokenA), 0, 0, 0);
        
        vm.stopPrank();
    }
    
    function test_44_RemoveZeroAmountReverts() public {
        vm.startPrank(user1);
        
        vm.expectRevert(abi.encodeWithSignature("InvalidAmount()"));
        router.removeLiquidity(address(ytTokenA), 0, 0, user1);
        
        vm.stopPrank();
    }
    
    function test_45_SwapZeroAmountReverts() public {
        vm.startPrank(user1);
        
        vm.expectRevert(abi.encodeWithSignature("InvalidAmount()"));
        router.swapYT(address(ytTokenA), address(ytTokenB), 0, 0, user1);
        
        vm.stopPrank();
    }
    
    function test_46_SwapUnwhitelistedTokenReverts() public {
        // 通过factory创建新的YTAssetVault
        address ytTokenDAddr = factory.createVault(
            "YT Token D",
            "YT-D",
            deployer,
            1000000 ether,
            address(wusd),
            block.timestamp + 365 days,
            PRICE_PRECISION,
            PRICE_PRECISION
        );
        YTAssetVault ytTokenD = YTAssetVault(ytTokenDAddr);
        
        wusd.mint(user1, 500 ether);
        
        vm.startPrank(user1);
        wusd.approve(address(ytTokenD), 500 ether);
        ytTokenD.depositYT(500 ether);
        
        // 添加YT-A流动性
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        
        // 尝试swap未白名单的代币
        ytTokenD.approve(address(router), 100 ether);
        
        vm.expectRevert(abi.encodeWithSignature("TokenNotWhitelisted()"));
        router.swapYT(address(ytTokenD), address(ytTokenA), 100 ether, 0, user1);
        
        vm.stopPrank();
    }
    
    // ==================== 18. 费用精确计算测试 ====================
    
    function test_47_ExactFeeCalculation() public {
        uint256 amount = 1000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        
        uint256 poolAmountBefore = vault.poolAmounts(address(ytTokenA));
        uint256 usdyBefore = vault.usdyAmounts(address(ytTokenA));
        
        router.addLiquidity(address(ytTokenA), amount, 0, 0);
        
        vm.stopPrank();
        
        uint256 poolAmountAfter = vault.poolAmounts(address(ytTokenA));
        uint256 usdyAfter = vault.usdyAmounts(address(ytTokenA));
        
        // 池子应该收到全部代币
        assertEq(poolAmountAfter - poolAmountBefore, amount, "pool should receive full amount");
        
        // USDY债务只记录扣费后的
        assertEq(usdyAfter - usdyBefore, 997 ether, "usdy debt incorrect");
    }
    
    function test_48_RedemptionFeeCalculation() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        
        uint256 ytLPBalance = ytlp.balanceOf(user1);
        
        vm.warp(block.timestamp + 16 * 60);
        
        uint256 poolAmountBefore = vault.poolAmounts(address(ytTokenA));
        
        router.removeLiquidity(address(ytTokenA), ytLPBalance, 0, user1);
        
        vm.stopPrank();
        
        uint256 poolAmountAfter = vault.poolAmounts(address(ytTokenA));
        
        uint256 feeInPool = poolAmountBefore - poolAmountAfter;
        assertEq(feeInPool, 997 ether, "fee should be collected");
        
        assertEq(poolAmountAfter, 3 ether, "remaining pool incorrect");
    }
    
    // ==================== 19. ytLP价格增长测试 ====================
    
    function test_49_YtLPPriceGrowthFromFees() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenA), 2000 ether, 0, 0);
        
        ytTokenB.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenB), 2000 ether, 0, 0);
        vm.stopPrank();
        
        uint256 priceBefore = poolManager.getPrice(true);
        uint256 supplyBefore = ytlp.totalSupply();
        
        console.log("Price before swaps:", priceBefore);
        console.log("Supply:", supplyBefore);
        
        // 执行多次swap累积手续费
        vm.startPrank(user1);
        
        for (uint i = 0; i < 10; i++) {
            ytTokenA.approve(address(router), 100 ether);
            router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
            
            ytTokenB.approve(address(router), 100 ether);
            router.swapYT(address(ytTokenB), address(ytTokenA), 100 ether, 0, user2);
        }
        
        vm.stopPrank();
        
        uint256 priceAfter = poolManager.getPrice(true);
        uint256 supplyAfter = ytlp.totalSupply();
        
        console.log("Price after swaps:", priceAfter);
        
        assertEq(supplyAfter, supplyBefore, "supply should not change");
        assertTrue(priceAfter > priceBefore, "price should increase");
        
        uint256 priceIncrease = (priceAfter - priceBefore) * 10000 / priceBefore;
        console.log("Price increase (bps):", priceIncrease);
        
        assertTrue(priceIncrease >= 10 && priceIncrease <= 30, "price increase should be 10-30 bps");
    }
    
    // ==================== 20. 价格查询测试 ====================
    
    function test_50_GetPriceFromVault() public view {
        uint256 price = vault.getPrice(address(ytTokenA), true);
        assertEq(price, 1e30, "vault price incorrect");
        
        uint256 maxPrice = vault.getMaxPrice(address(ytTokenA));
        uint256 minPrice = vault.getMinPrice(address(ytTokenA));
        
        assertEq(maxPrice, 1e30);
        assertEq(minPrice, 1e30);
    }
    
    function test_51_GetPriceInfo() public view {
        (
            uint256 currentPrice,
            uint256 cachedPrice,
            uint256 maxPrice,
            uint256 minPrice,
            uint256 spread
        ) = priceFeed.getPriceInfo(address(ytTokenA));
        
        assertEq(currentPrice, 1e30, "current price incorrect");
        assertEq(maxPrice, 1e30, "max price incorrect");
        assertEq(minPrice, 1e30, "min price incorrect");
        assertEq(spread, 0, "spread should be 0");
    }
    
    function test_52_YtLPPriceCalculation() public {
        uint256 amount = 1000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        router.addLiquidity(address(ytTokenA), amount, 0, 0);
        vm.stopPrank();
        
        uint256 ytLPPrice = poolManager.getPrice(true);
        
        assertTrue(ytLPPrice > 1 ether, "ytLP price should be > $1");
        assertTrue(ytLPPrice < 1.01 ether, "ytLP price should be < $1.01");
    }
    
    function test_53_AddLiquidityWithSpread() public {
        priceFeed.setSpreadBasisPoints(address(ytTokenA), 20); // 0.2%
        
        uint256 amount = 1000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        
        uint256 ytLPReceived = router.addLiquidity(address(ytTokenA), amount, 0, 0);
        
        vm.stopPrank();
        
        uint256 expectedYtLP = 995.006 ether;
        assertEq(ytLPReceived, expectedYtLP, "ytLP with spread incorrect");
        
        priceFeed.setSpreadBasisPoints(address(ytTokenA), 0);
    }
    
    function test_54_RemoveLiquiditySlippageProtection() public {
        uint256 amount = 1000 ether;
        vm.startPrank(user1);
        ytTokenA.approve(address(router), amount);
        router.addLiquidity(address(ytTokenA), amount, 0, 0);
        
        uint256 ytLPBalance = ytlp.balanceOf(user1);
        
        vm.warp(block.timestamp + 15 * 60 + 1);
        
        uint256 tooHighMinOut = 2000 ether;
        
        vm.expectRevert(abi.encodeWithSignature("InsufficientOutput()"));
        router.removeLiquidity(address(ytTokenA), ytLPBalance, tooHighMinOut, user1);
        
        vm.stopPrank();
    }
    
    function test_55_SwapSlippageProtection() public {
        uint256 liquidityAmount = 2000 ether;
        
        vm.startPrank(user1);
        ytTokenA.approve(address(router), liquidityAmount);
        router.addLiquidity(address(ytTokenA), liquidityAmount, 0, 0);
        
        ytTokenB.approve(address(router), liquidityAmount);
        router.addLiquidity(address(ytTokenB), liquidityAmount, 0, 0);
        vm.stopPrank();
        
        uint256 swapAmount = 100 ether;
        uint256 tooHighMinOut = 150 ether;
        
        vm.startPrank(user2);
        ytTokenA.approve(address(router), swapAmount);
        
        vm.expectRevert(abi.encodeWithSignature("InsufficientOutput()"));
        router.swapYT(address(ytTokenA), address(ytTokenB), swapAmount, tooHighMinOut, user2);
        
        vm.stopPrank();
    }
    
    function test_56_OnlyHandlerCanAddLiquidity() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(poolManager), 1000 ether);
        
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        poolManager.addLiquidityForAccount(
            user1,
            user1,
            address(ytTokenA),
            1000 ether,
            0,
            0
        );
        
        vm.stopPrank();
    }
    
    function test_57_OnlyPoolManagerCanBuyUSDY() public {
        vm.startPrank(user1);
        ytTokenA.approve(address(vault), 1000 ether);
        
        vm.expectRevert(abi.encodeWithSignature("OnlyPoolManager()"));
        vault.buyUSDY(address(ytTokenA), user1);
        
        vm.stopPrank();
    }
    
    function test_58_OnlyGovCanSetFees() public {
        vm.startPrank(user1);
        
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        vault.setSwapFees(40, 5, 60, 25);
        
        vm.stopPrank();
    }
    
    function test_59_OnlyKeeperCanUpdatePrice() public {
        // 非keeper和非gov不能调用updatePrice
        vm.startPrank(user1);
        
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        priceFeed.updatePrice(address(ytTokenA));
        
        vm.stopPrank();
    }
    
    function test_60_SetKeeperPermission() public {
        // 设置user1为keeper
        priceFeed.setKeeper(user1, true);
        assertTrue(priceFeed.isKeeper(user1), "user1 should be keeper");
        
        // keeper可以调用updatePrice
        vm.startPrank(user1);
        uint256 price = priceFeed.updatePrice(address(ytTokenA));
        vm.stopPrank();
        
        assertEq(price, PRICE_PRECISION, "price should be updated");
        
        // 移除keeper权限
        priceFeed.setKeeper(user1, false);
        assertFalse(priceFeed.isKeeper(user1), "user1 should not be keeper");
        
        // 移除后不能调用
        vm.startPrank(user1);
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        priceFeed.updatePrice(address(ytTokenA));
        vm.stopPrank();
    }
    
    function test_61_GovCanAlwaysUpdatePrice() public {
        // gov可以直接调用updatePrice
        uint256 price = priceFeed.updatePrice(address(ytTokenA));
        assertEq(price, PRICE_PRECISION, "gov can update price");
    }
    
    // ==================== 21. YTRewardRouter 暂停功能测试 ====================
    
    function test_62_RouterPauseByGov() public {
        // Gov可以暂停
        router.pause();
        assertTrue(router.paused(), "router should be paused");
        
        // Gov可以恢复
        router.unpause();
        assertFalse(router.paused(), "router should be unpaused");
    }
    
    function test_63_OnlyGovCanPauseRouter() public {
        // User不能暂停
        vm.startPrank(user1);
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        router.pause();
        vm.stopPrank();
        
        // User不能恢复
        router.pause(); // 由deployer暂停
        
        vm.startPrank(user1);
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        router.unpause();
        vm.stopPrank();
        
        router.unpause(); // 恢复
    }
    
    function test_64_CannotAddLiquidityWhenRouterPaused() public {
        // 暂停router
        router.pause();
        
        // 尝试添加流动性应该失败
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        
        vm.expectRevert(abi.encodeWithSignature("EnforcedPause()"));
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        // 恢复后应该可以添加
        router.unpause();
        
        vm.startPrank(user1);
        uint256 ytLPReceived = router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        assertEq(ytLPReceived, 997 ether, "add liquidity should work after unpause");
    }
    
    function test_65_CannotRemoveLiquidityWhenRouterPaused() public {
        // 先添加流动性
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        
        uint256 ytLPBalance = ytlp.balanceOf(user1);
        
        // 等待冷却期
        vm.warp(block.timestamp + 15 * 60 + 1);
        vm.stopPrank();
        
        // 暂停router
        router.pause();
        
        // 尝试移除流动性应该失败
        vm.startPrank(user1);
        vm.expectRevert(abi.encodeWithSignature("EnforcedPause()"));
        router.removeLiquidity(address(ytTokenA), ytLPBalance, 0, user1);
        vm.stopPrank();
        
        // 恢复后应该可以移除
        router.unpause();
        
        vm.startPrank(user1);
        uint256 amountOut = router.removeLiquidity(address(ytTokenA), ytLPBalance, 0, user1);
        vm.stopPrank();
        
        assertEq(amountOut, 997 ether, "remove liquidity should work after unpause");
    }
    
    function test_66_CannotSwapWhenRouterPaused() public {
        // 先添加流动性
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenA), 2000 ether, 0, 0);
        
        ytTokenB.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenB), 2000 ether, 0, 0);
        vm.stopPrank();
        
        // 暂停router
        router.pause();
        
        // 尝试swap应该失败
        vm.startPrank(user2);
        ytTokenA.approve(address(router), 100 ether);
        
        vm.expectRevert(abi.encodeWithSignature("EnforcedPause()"));
        router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        vm.stopPrank();
        
        // 恢复后应该可以swap
        router.unpause();
        
        vm.startPrank(user2);
        uint256 amountOut = router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        vm.stopPrank();
        
        assertEq(amountOut, 99.7 ether, "swap should work after unpause");
    }
    
    function test_67_QueryFunctionsWorkWhenRouterPaused() public {
        // 先添加流动性
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        vm.stopPrank();
        
        // 暂停router
        router.pause();
        
        // 查询函数应该仍然可用
        uint256 ytLPPrice = router.getYtLPPrice();
        assertTrue(ytLPPrice > 0, "getYtLPPrice should work when paused");
        
        uint256 accountValue = router.getAccountValue(user1);
        assertTrue(accountValue > 0, "getAccountValue should work when paused");
        
        // 验证返回的值是合理的
        assertTrue(ytLPPrice > 1 ether, "ytLP price should be > $1");
        assertTrue(accountValue >= 995 ether && accountValue <= 1005 ether, "account value should be around 1000");
    }
    
    function test_68_PauseRouterDoesNotAffectVaultDirectly() public {
        // 暂停router不影响直接通过vault操作
        router.pause();
        
        vm.startPrank(user1);
        ytTokenA.approve(address(vault), 1000 ether);
        
        // 直接通过poolManager添加流动性仍然失败（因为user1不是handler）
        vm.expectRevert(abi.encodeWithSignature("Forbidden()"));
        poolManager.addLiquidityForAccount(user1, user1, address(ytTokenA), 1000 ether, 0, 0);
        
        vm.stopPrank();
        
        router.unpause();
    }
    
    function test_69_CompleteFlowWithPauseResume() public {
        console.log("=== Complete Flow With Pause/Resume Test ===");
        
        // 步骤1: 添加流动性
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 1000 ether);
        uint256 ytLP1 = router.addLiquidity(address(ytTokenA), 1000 ether, 0, 0);
        console.log("Added liquidity, received ytLP:", ytLP1);
        vm.stopPrank();
        
        // 步骤2: 暂停router
        router.pause();
        console.log("Router paused");
        
        // 步骤3: 验证所有操作都被阻止
        vm.startPrank(user1);
        ytTokenB.approve(address(router), 1000 ether);
        
        vm.expectRevert(abi.encodeWithSignature("EnforcedPause()"));
        router.addLiquidity(address(ytTokenB), 1000 ether, 0, 0);
        console.log("Add liquidity blocked during pause");
        vm.stopPrank();
        
        // 步骤4: 恢复router
        router.unpause();
        console.log("Router unpaused");
        
        // 步骤5: 继续正常操作
        vm.startPrank(user1);
        uint256 ytLP2 = router.addLiquidity(address(ytTokenB), 1000 ether, 0, 0);
        console.log("Added liquidity after unpause, received ytLP:", ytLP2);
        vm.stopPrank();
        
        // 验证总余额
        uint256 totalYtLP = ytlp.balanceOf(user1);
        console.log("Total ytLP:", totalYtLP);
        assertEq(totalYtLP, ytLP1 + ytLP2, "total ytLP should be sum of both additions");
    }
    
    function test_70_EmergencyScenarioPauseEverything() public {
        console.log("=== Emergency Scenario: Pause Everything ===");
        
        // 先建立一些状态
        vm.startPrank(user1);
        ytTokenA.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenA), 2000 ether, 0, 0);
        
        ytTokenB.approve(address(router), 2000 ether);
        router.addLiquidity(address(ytTokenB), 2000 ether, 0, 0);
        vm.stopPrank();
        
        console.log("Initial liquidity added");
        
        // 模拟紧急情况：暂停router
        router.pause();
        console.log("Router paused for emergency");
        
        // 同时暂停vault (通过设置紧急模式)
        vault.setEmergencyMode(true);
        console.log("Vault emergency mode activated");
        
        // 验证所有操作都被阻止
        vm.startPrank(user2);
        ytTokenA.approve(address(router), 100 ether);
        
        // Router暂停阻止操作
        vm.expectRevert(abi.encodeWithSignature("EnforcedPause()"));
        router.addLiquidity(address(ytTokenA), 100 ether, 0, 0);
        
        vm.expectRevert(abi.encodeWithSignature("EnforcedPause()"));
        router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        
        vm.stopPrank();
        
        console.log("All operations blocked during emergency");
        
        // 恢复系统
        router.unpause();
        vault.setEmergencyMode(false);
        console.log("System recovered from emergency");
        
        // 验证系统恢复正常
        vm.startPrank(user2);
        uint256 swapOut = router.swapYT(address(ytTokenA), address(ytTokenB), 100 ether, 0, user2);
        vm.stopPrank();
        
        assertEq(swapOut, 99.7 ether, "swap should work after recovery");
        console.log("System operational after recovery");
    }
    
    receive() external payable {}
}