Очень важная строка! Здесь мы говорим, чтобы Foundry работал в тестах только с контрактом Handler, исключая вызовы на WETH.
Дальше мы можем написать небольшой тест:
function invariant_eth_balance() public {
assertGe(address(weth).balance, handler.wethBalance());
console.log("handler num calls", handler.numCalls());
}Он прошел успешно, но все таки некоторое количество откатов может остаться. Происходит это из-за функции fail(), которую мы написали специально для проверки адекватности инвариант теста.
Попробуем исключить fail() из результатов. Да, конечно, можно просто удалить эту функцию из контракта, но есть и другой способ.
В функции setUp() мы можем дополнительно указать те функции, которые хотим использовать.
bytes4[] memory selectors = new bytes4[](3);
selectors[0] = Handler.deposit.selector;
selectors[1] = Handler.withdraw.selector;
selectors[2] = Handler.sendToFallback.selector;
Мы создаем байтовый массив и помещаем туда селекторы необходимых нам для тестов функций.
targetSelector(
FuzzSelector({addr: address(handler), selectors: selectors})
);
А тут мы говорим Foundry, что в своих тестах мы хотим использовать только этот набор.
И есть сейчас мы выполним команду forge test, то количество откатов будет нулевым!
Наш тест стал уже намного лучше, но еще не совсем дотягивающим до нормы. Если подумать, то контрактом WETH будет пользоваться не один пользователь, а непредвиденное количество.
Поэтому в следующем посте мы разберем концепцию ролей в инвариант тестах!
#foundry #lesson29