mockCallRevert

签名

function mockCallRevert(address where, bytes calldata data, bytes calldata retdata) external;
function mockCallRevert(
    address where,
    uint256 value,
    bytes calldata data,
    bytes calldata retdata
) external;

描述

如果调用数据严格或宽松匹配 data 并返回 retdata,则将所有对地址 where 的调用 revert。

retdata 可以是原始返回消息或自定义错误。

当对 where 进行调用时,首先检查调用数据是否与 data 完全匹配。 如果不匹配,则检查调用数据是否部分匹配,匹配从调用数据的第一个字节开始。

如果找到匹配项,则调用将被 revert,并返回 retdata

使用第二个签名,我们可以模拟具有特定 msg.value 的调用。在存在歧义的情况下,Calldata 匹配优先于 msg.value

revert 的模拟调用将一直有效,直到调用 clearMockedCalls

ℹ️ 内部调用

此作弊码目前不适用于内部调用。请参见问题 #432

例子

使用原始错误消息 revert 实际调用:

function testMockCallRevert() public {
    vm.mockCallRevert(
        address(0),
        abi.encodeWithSelector(MyToken.balanceOf.selector, address(1)),
        "REVERT_MESSAGE"
    );
    vm.expectRevert("REVERT_MESSAGE");
    IERC20(address(0)).balanceOf(address(1));
}

使用自定义错误 revert 调用:

function testMockCallRevertWithCustomError() public {
    bytes memory customError = abi.encodeWithSelector(TestError.selector, "ERROR_MESSAGE");
    vm.mockCallRevert(
        address(0),
        abi.encodeWithSelector(MyToken.balanceOf.selector, address(1)),
        customError
    );
    vm.expectRevert(customError);
    IERC20(address(0)).balanceOf(address(1));
}

模拟具有给定 msg.value 的调用:

function testMockCallRevertWithValue() public {
    assertEq(example.pay{value: 10}(1), 1);
    assertEq(example.pay{value: 1}(2), 2);
    vm.mockCallRevert(
        address(example),
        10,
        abi.encodeWithSelector(example.pay.selector),
        "ERROR_MESSAGE"
    );
    assertEq(example.pay{value: 1}(2), 2);
    vm.expectRevert("ERROR_MESSAGE");
    assertEq(example.pay{value: 10}(1), 99);
}