如何使用Foundry开发并部署一个ERC20智能合约

1
// SPDX-License-Identifier: MIT
2
pragma solidity >=0.8.0;
3

4
import {ERC20} from "openzeppelin-contracts/token/ERC20/ERC20.sol";
5

6
contract MyERC20 is ERC20 {
7
    // 设置代币的名称，符号，和小数位数
8
    constructor() ERC20("My Token", "MYT") {
9
        // 设置代币的总供应量为 10000 个，并分配给部署者
10
        _mint(msg.sender, 10000 * 10 ** decimals());
11
    }
12

13
    // 添加一个自定义的事件，用于记录代币的销毁
14
    event Burn(address indexed from, uint256 amount);
15

16
    // 添加一个自定义的函数，用于销毁代币
17
    function burn(uint256 amount) public {
18
        // 调用内部的 _burn 函数，从发送者的地址中销毁指定数量的代币
19
        _burn(msg.sender, amount);
20
        // 触发 Burn 事件
21
        emit Burn(msg.sender, amount);
22
    }
23
}

1
// SPDX-License-Identifier: MIT
2
pragma solidity >=0.8.0;
3

4
import {console} from "forge-std/console.sol";
5
import {stdStorage, StdStorage, Test} from "forge-std/Test.sol";
6

7
import {MyERC20} from "../MyERC20.sol";
8

9
contract MyERC20Test is MyERC20, Test {
10
    // 设置两个测试账户
11
    address alice = address(0x123);
12
    address bob = address(0x456);
13

14
    // 在每个测试用例之前，给测试账户分配一些代币
15
    function setUp() public override {
16
        _mint(alice, 1000 * 10 ** decimals());
17
        _mint(bob, 500 * 10 ** decimals());
18
    }
19

20
    // 测试代币的转账功能
21
    function testTransfer() public {
22
        // 打印日志
23
        console.log("Testing transfer");
24
        // 记录转账前的余额
25
        uint256 aliceBalanceBefore = balanceOf(alice);
26
        uint256 bobBalanceBefore = balanceOf(bob);
27
        // 以 Alice 的身份，向 Bob 转账 100 个代币
28
        vm.prank(alice);
29
        transfer(bob, 100 * 10 ** decimals());
30
        // 记录转账后的余额
31
        uint256 aliceBalanceAfter = balanceOf(alice);
32
        uint256 bobBalanceAfter = balanceOf(bob);
33
        // 检查余额是否正确变化
34
        assertEq(aliceBalanceAfter, aliceBalanceBefore - 100 * 10 ** decimals(), "Alice balance should decrease");
35
        assertEq(bobBalanceAfter, bobBalanceBefore + 100 * 10 ** decimals(), "Bob balance should increase");
36
    }
37

38
    // 测试代币的销毁功能
39
    function testBurn() public {
40
        // 打印日志
41
        console.log("Testing burn");
42
        // 记录销毁前的余额和总供应量
43
        uint256 aliceBalanceBefore = balanceOf(alice);
44
        uint256 totalSupplyBefore = totalSupply();
45
        // 以 Alice 的身份，销毁 50 个代币
46
        vm.prank(alice);
47
        burn(50 * 10 ** decimals());
48
        // 记录销毁后的余额和总供应量
49
        uint256 aliceBalanceAfter = balanceOf(alice);
50
        uint256 totalSupplyAfter = totalSupply();
51
        // 检查余额和总供应量是否正确变化
52
        assertEq(aliceBalanceAfter, aliceBalanceBefore - 50 * 10 ** decimals(), "Alice balance should decrease");
53
        assertEq(totalSupplyAfter, totalSupplyBefore - 50 * 10 ** decimals(), "Total supply should decrease");
54
    }
55
}