模糊测试(Fuzz Testing)

Forge 支持基于属性的测试。

基于属性的测试是一种测试一般行为而不是孤立场景的方法。

让我们通过编写单元测试来检查这意味着什么,找到我们正在测试的一般属性,并将其转换为基于属性的测试:

pragma solidity 0.8.10;

import "forge-std/Test.sol";

contract Safe {
    receive() external payable {}

    function withdraw() external {
        payable(msg.sender).transfer(address(this).balance);
    }
}

contract SafeTest is Test {
    Safe safe;

    // Needed so the test contract itself can receive ether
    // when withdrawing
    receive() external payable {}

    function setUp() public {
        safe = new Safe();
    }

    function testWithdraw() public {
        payable(address(safe)).transfer(1 ether);
        uint256 preBalance = address(this).balance;
        safe.withdraw();
        uint256 postBalance = address(this).balance;
        assertEq(preBalance + 1 ether, postBalance);
    }
}

运行测试,我们看到它通过了:

$ forge test
Compiling 6 files with 0.8.10
Solc 0.8.10 finished in 3.78s
Compiler run successful

Running 1 test for test/Safe.t.sol:SafeTest
[PASS] testWithdraw() (gas: 19462)
Test result: ok. 1 passed; 0 failed; finished in 873.70µs

这个单元测试_确实测试_我们可以从我们的保险箱中取出以太币。 但是,谁能说它适用于所有金额,而不仅仅是 1 个以太币?

这里的一般性质是:给定一个安全的余额,当我们提取时,我们应该得到保险箱里的东西。

Forge 将运行任何至少采用一个参数的测试作为基于属性的测试,所以让我们重写:

contract SafeTest is Test {
    // ...

    function testWithdraw(uint256 amount) public {
        payable(address(safe)).transfer(amount);
        uint256 preBalance = address(this).balance;
        safe.withdraw();
        uint256 postBalance = address(this).balance;
        assertEq(preBalance + amount, postBalance);
    }
}

如果我们现在运行测试,我们可以看到 Forge 运行基于属性的测试,但它因 amount 的高值而失败:

$ forge test
Compiling 1 files with 0.8.10
Solc 0.8.10 finished in 1.69s
Compiler run successful

Running 1 test for test/Safe.t.sol:SafeTest
[FAIL. Reason: EvmError: Revert Counterexample: calldata=0x215a2f200000000000000000000000000000000000000001000000000000000000000000, args=[79228162514264337593543950336]] testWithdraw(uint256) (runs: 47, μ: 19554, ~: 19554)
Test result: FAILED. 0 passed; 1 failed; finished in 8.75ms

给测试合约的默认以太币数量是 2**96 wei(在 DappTools 中),所以我们必须将数量类型限制为 uint96 以确保我们不会尝试发送超过 我们有:

    function testWithdraw(uint96 amount) public {

现在它通过了:

$ forge test
Compiling 1 files with 0.8.10
Solc 0.8.10 finished in 1.67s
Compiler run successful

Running 1 test for test/Safe.t.sol:SafeTest
[PASS] testWithdraw(uint96) (runs: 256, μ: 19078, ~: 19654)
Test result: ok. 1 passed; 0 failed; finished in 19.56ms

您可能希望使用 assume cheatcodes排除某些情况。 在这些情况下,模糊器fuzzer将丢弃输入并开始新的模糊测试运行:

function testWithdraw(uint96 amount) public {
    vm.assume(amount > 0.1 ether);
    // snip
}

有多种方法可以运行基于属性的测试,特别是参数测试和模糊测试。 Forge 仅支持模糊测试。

Interpreting results

您可能已经注意到,与单元测试相比,模糊测试的总结略有不同:

  • "runs" 是指模糊器fuzzer测试的场景数量。 默认情况下,模糊器fuzzer将生成 256 个场景,但是,这可以使用 FOUNDRY_FUZZ_RUNS 环境变量进行配置。
  • “μ”(希腊字母 mu)是所有模糊运行中使用的平均gas
  • “~”(波浪号)是所有模糊运行中使用的中值gas