staking/test/StakeVault.t.sol

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

import { ERC20 } from "@openzeppelin/contracts/token/ERC20/ERC20.sol";

import { Test } from "forge-std/Test.sol";
import { Deploy } from "../script/Deploy.s.sol";
import { DeployBroken } from "./script/DeployBroken.s.sol";
import { DeploymentConfig } from "../script/DeploymentConfig.s.sol";
import { StakeManager } from "../contracts/StakeManager.sol";
import { StakeVault } from "../contracts/StakeVault.sol";
import { VaultFactory } from "../contracts/VaultFactory.sol";

contract StakeVaultTest is Test {
    StakeManager internal stakeManager;

    DeploymentConfig internal deploymentConfig;

    VaultFactory internal vaultFactory;

    StakeVault internal stakeVault;

    address internal deployer;

    address internal testUser = makeAddr("testUser");

    address internal stakeToken;

    function setUp() public virtual {
        Deploy deployment = new Deploy();
        (vaultFactory, stakeManager, deploymentConfig) = deployment.run();
        (deployer, stakeToken) = deploymentConfig.activeNetworkConfig();

        vm.prank(testUser);
        stakeVault = vaultFactory.createVault();
    }
}

contract StakedTokenTest is StakeVaultTest {
    function setUp() public override {
        StakeVaultTest.setUp();
    }

    function testStakeToken() public {
        assertEq(address(stakeVault.stakedToken()), stakeToken);
    }
}

contract StakeTest is StakeVaultTest {
    function setUp() public override {
        DeployBroken deployment = new DeployBroken();
        (vaultFactory, stakeManager, stakeToken) = deployment.run();

        vm.prank(testUser);
        stakeVault = vaultFactory.createVault();
    }

    function test_RevertWhen_StakeTokenTransferFails() public {
        // ensure user has funds
        deal(stakeToken, testUser, 1000);

        vm.startPrank(address(testUser));
        ERC20(stakeToken).approve(address(stakeVault), 100);
        vm.expectRevert(StakeVault.StakeVault__StakingFailed.selector);
        stakeVault.stake(100, 0);
    }
}