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);
}