Foundry с 0. Часть 21

Эту неделю мы начнем с разговора о работе со временем в наших тестах.

Напишем простой контракт:

pragma solidity 0.8.18;

contract Auction {
    uint256 public startAt = block.timestamp + 1 days;
    uint256 public endAt = block.timestamp + 2 days;

    function bid() external {
        require(
            block.timestamp >= startAt && block.timestamp < endAt, "cannot bid"
        );
    }

    function end() external {
        require(block.timestamp >= endAt, "cannot end");
    }
}

В Foundry я встречал 4 самых популярных читкода для манипуляцией временем:

1. vm.warp - устанавливает block.timestamp на конкретное значение времени
2. vm.roll - устанавливает block.number
3. skip - увеличивает текущий timestamp
4. rewind - уменьшает текущий timestamp

Какие же тесты можно тут написать?

Например, что bid() вызывается в определенный период времени:

function testBid() public {
  vm.warp(startAt + 1 days);
  auction.bid();
}

или так:

function testTimestamp() public {
   uint256 t = block.timestamp;
   
   // set block.timestamp to t + 100
   skip(100);
   assertEq(block.timestamp, t + 100);

    // set block.timestamp to t + 100 - 100;
    rewind(100);
    assertEq(block.timestamp, t);
}

Обратите внимание на последовательность вызовов! Если в одном тесте вы добавите время, то при вычитании другого значения позже, действие будет происходить с обновленным значением после первой манипуляции!

Ну, и простой тест для block.number:

function testBlockNumber() public {
   uint256 b = block.number;
   // set block number to 11
   vm.roll(11);
   assertEq(block.number, 11);
}

Такие читкоды бывают полезны, когда вы пишите тесты для голосований ДАО или проверки возможности вывода депозита пользователя только в определенный промежуток времени.

В прочем, ничего сложного тут нет

#foundry #lesson21