使用Foundry开发第一个ERC20合约

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

import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
import "@openzeppelin/contracts/access/Ownable.sol";

/**
 * @title MyToken
 * @dev 一个基于OpenZeppelin的ERC20代币合约示例，继承了ERC20标准和Ownable权限管理
 */
contract MyToken is ERC20, Ownable {

    /**
     * @dev constructor, initializes the token name and symbol, and pre-mints a certain amount of tokens for the deployer
     * @param name token
     * @param symbol of the token
     */
    constructor(string memory name, string memory symbol) ERC20(name, symbol) Ownable(msg.sender) {
        // 在合约部署时铸造100万个代币，假设每个代币有18个小数位
        _mint(msg.sender, 1000000 * 10 ** decimals());
    }

    /**
     * @dev 合约拥有者调用的函数，可以冻结账户
     * @param account 需要冻结的账户地址
     * @notice 仅合约拥有者（部署者）可以调用此函数
     */
    function freezeAccount(address account) external onlyOwner {
        // 这里可以实现冻结账户的逻辑
        // 比如将账户标记为冻结状态，以禁止转账或其他操作
    }

    // 可根据需要添加更多的功能
}

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

import {Test} from "forge-std/Test.sol";
import "@openzeppelin/contracts/access/Ownable.sol";
import "../src/MyToken.sol";

contract MyTokenTest is Test {
    MyToken public myToken;
    address public owner;
    address public addr1;
    uint256 public initialBalance = 1000000 * 10 ** 18;

    // 在每个测试之前执行
    function setUp() public {
        // 部署合约，并为部署者铸造初始代币
        owner = address(this);
        addr1 = address(0x123);

        myToken = new MyToken("MyToken", "MTK");

        // 验证初始余额
        assertEq(myToken.balanceOf(owner), initialBalance);
    }

    // 测试代币的名称和符号
    function testTokenNameAndSymbol() public view {
        assertEq(myToken.name(), "MyToken");
        assertEq(myToken.symbol(), "MTK");
    }

    // 测试代币的总供应量
    function testTotalSupply() public view {
        assertEq(myToken.totalSupply(), initialBalance);
    }

    // 测试转账功能
    function testTransfer() public {
        uint256 transferAmount = 100 * 10 ** 18; // 转账100个代币
        myToken.transfer(addr1, transferAmount);

        // 验证余额变化
        assertEq(myToken.balanceOf(owner), initialBalance - transferAmount);
        assertEq(myToken.balanceOf(addr1), transferAmount);
    }

    // 测试代币铸造功能，确认部署时代币已铸造
    function testInitialMint() public view {
        assertEq(myToken.balanceOf(owner), initialBalance);
    }

    // call `freezeAccount`
    function testOnlyOwnerCanFreeze() public {
        vm.startPrank(addr1);
        // Only owner can freeze account
        vm.expectRevert(
            abi.encodeWithSelector(
                Ownable.OwnableUnauthorizedAccount.selector,
                addr1
            )
        );
        myToken.freezeAccount(addr1);
        vm.stopPrank();
    }

    // 测试其他非拥有者无法调用 `freezeAccount`
    function testNonOwnerCannotFreeze() public {
        address nonOwner = address(0x456);
        vm.startPrank(nonOwner);
        vm.expectRevert(
            abi.encodeWithSelector(
                Ownable.OwnableUnauthorizedAccount.selector,
                nonOwner
            )
        );
        myToken.freezeAccount(addr1); // 只有拥有者可以冻结账户
        vm.stopPrank();
    }
}