GPT

Foundry Test Suite

// SPDX-License-Identifier: MIT pragma solidity ^0.8.24; import {Test, console2} from "forge-std/Test.sol"; import {StakingVault} from "../src/StakingVault.sol"; import {ERC1967Proxy} from "@openzeppelin/contracts/proxy

Prompt text

Original English text. Replace placeholders with your own details.

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

import {Test, console2} from "forge-std/Test.sol";
import {StakingVault} from "../src/StakingVault.sol";
import {ERC1967Proxy} from "@openzeppelin/contracts/proxy/ERC1967/ERC1967Proxy.sol";
import {MockERC20} from "./mocks/MockERC20.sol";

contract StakingVaultTest is Test {
 StakingVault public vault;
 MockERC20 public token;
 address public owner = makeAddr("owner");
 address public alice = makeAddr("alice");
 address public bob = makeAddr("bob");

 uint256 constant LOCK_DURATION = 7 days;
 uint256 constant STAKE_AMOUNT = 1000e18;

 function setUp() public {
 token = new MockERC20("Stake Token", "STK");

 // Deploy behind UUPS proxy
 StakingVault impl = new StakingVault();
 bytes memory initData = abi.encodeCall(
 StakingVault.initialize,
 (address(token), LOCK_DURATION, owner)
 );
 ERC1967Proxy proxy = new ERC1967Proxy(address(impl), initData);
 vault = StakingVault(address(proxy));

 // Fund test accounts
 token.mint(alice, 10_000e18);
 token.mint(bob, 10_000e18);

 vm.prank(alice);
 token.approve(address(vault), type(uint256).max);
 vm.prank(bob);
 token.approve(address(vault), type(uint256).max);
 }

 function test_stake_updatesBalance() public {
 vm.prank(alice);
 vault.stake(STAKE_AMOUNT);

 (uint128 amount,,) = vault.stakes(alice);
 assertEq(amount, STAKE_AMOUNT);
 assertEq(vault.totalStaked(), STAKE_AMOUNT);
 assertEq(token.balanceOf(address(vault)), STAKE_AMOUNT);
 }

 function test_withdraw_revertsBeforeLock() public {
 vm.prank(alice);
 vault.stake(STAKE_AMOUNT);

 vm.prank(alice);
 vm.expectRevert();
 vault.withdraw();
 }

 function test_withdraw_succeedsAfterLock() public {
 vm.prank(alice);
 vault.stake(STAKE_AMOUNT);

 vm.warp(block.timestamp + LOCK_DURATION + 1);

 vm.prank(alice);
 vault.withdraw();

 (uint128 amount,,) = vault.stakes(alice);
 assertEq(amount, 0);
 assertEq(token.balanceOf(alice), 10_000e18);
 }

 function test_stake_revertsWhenPaused() public {
 vm.prank(owner);
 vault.pause();

 vm.prank(alice);
 vm.expectRevert();
 vault.stake(STAKE_AMOUNT);
 }

 function testFuzz_stake_arbitraryAmount(uint128 amount) public {
 vm.assume(amount > 0 && amount <= 10_000e18);

 vm.prank(alice);
 vault.stake(amount);

 (uint128 staked,,) = vault.stakes(alice);
 assertEq(staked, amount);
 }
}

Source & attribution

Awesome Prompts · Awesome Prompts

Source-listed license: GPL-3.0. Prompt text is preserved; titles and previews may be shortened for display. Source examples are not NexGateHub test results.

View original source ↗ · Source credits & licenses

Explore more prompts →