Skip to content

Commit

Permalink
boost coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
pblivin0x committed Jul 3, 2024
1 parent 0697f39 commit 61b8409
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions test/staking/SnapshotStakingPool.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,19 @@ contract SnapshotStakingPoolTest is Test {
assertEq(snapshotStakingPool.balanceOf(bob.addr), amount);
assertEq(snapshotStakingPool.nextClaimId(bob.addr), 1);
assertEq(snapshotStakingPool.getCurrentSnapshotId(), 0);

_snapshot(1 ether);
_snapshot(1 ether);

vm.prank(owner);
stakeToken.transfer(alice.addr, amount);
vm.prank(alice.addr);
stakeToken.approve(address(snapshotStakingPool), amount);

vm.prank(alice.addr);
snapshotStakingPool.stake(amount);

assertEq(snapshotStakingPool.nextClaimId(alice.addr), snapshotStakingPool.getCurrentSnapshotId());
}

function testUnstake() public {
Expand Down Expand Up @@ -295,6 +308,19 @@ contract SnapshotStakingPoolTest is Test {
assertEq(snapshotStakingPool.canAccrue(), true);
}

function testGetTimeUntilNextSnapshot() public {
vm.warp(block.timestamp + snapshotDelay + 1);
assertEq(snapshotStakingPool.getTimeUntilNextSnapshot(), 0);

_stake(bob.addr, 1 ether);
_snapshot(1 ether);

assertEq(snapshotStakingPool.getTimeUntilNextSnapshot(), snapshotDelay);

vm.warp(block.timestamp + snapshotDelay - 10);
assertEq(snapshotStakingPool.getTimeUntilNextSnapshot(), 10);
}

function testClaim() public {
vm.prank(bob.addr);
vm.expectRevert(SnapshotStakingPool.InvalidSnapshotId.selector);
Expand Down

0 comments on commit 61b8409

Please sign in to comment.