AtHeartEngineer/staking-reward-streamer
1
0
Fork 0
mirror of https://github.com/vacp2p/staking-reward-streamer.git synced 2026-01-07 22:43:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(test): add test for withdraw after leave

Browse Source
This commit is contained in:
Ricardo Guilherme Schmidt
2025-08-21 13:19:16 -03:00
parent 1f9e9f678c
commit 33781a53d0
1 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
test/RewardsStreamerMP.t.sol
View File

@@ -2353,6 +2353,33 @@ contract LeaveTest is StakeManagerTest {
assertEq(stakingToken.balanceOf(alice), aliceInitialBalance, "Alice has withdrawn her funds");
}
function test_WithdrawFromVaultCannotHappenBeforeLockupEnd() public {
uint256 aliceInitialBalance = stakingToken.balanceOf(alice);
uint256 stakeAmount = 10e18;
uint256 lockUpPeriod = streamer.MIN_LOCKUP_PERIOD();
_stake(alice, stakeAmount, lockUpPeriod);
assertEq(stakingToken.balanceOf(alice), aliceInitialBalance - stakeAmount, "Alice should have staked tokens");
_leave(alice);
vm.startPrank(alice);
vm.expectRevert(StakeVault.StakeVault__FundsLocked.selector);
StakeVault(vaults[alice]).withdrawFromVault(stakeAmount, alice);
assertEq(stakingToken.balanceOf(alice), aliceInitialBalance - stakeAmount, "Alice should still not have her funds back");
}
function test_WithdrawStakingTokenCannotHappenBeforeLockupEnd() public {
uint256 aliceInitialBalance = stakingToken.balanceOf(alice);
uint256 stakeAmount = 10e18;
uint256 lockUpPeriod = streamer.MIN_LOCKUP_PERIOD();
_stake(alice, stakeAmount, lockUpPeriod);
assertEq(stakingToken.balanceOf(alice), aliceInitialBalance - stakeAmount, "Alice should have staked tokens");
_leave(alice);
vm.prank(alice);
StakeVault(vaults[alice]).withdraw(stakingToken, stakeAmount);
assertEq(stakingToken.balanceOf(alice), aliceInitialBalance - stakeAmount, "Alice should still not have her funds back");
}
}
contract MaliciousUpgradeTest is StakeManagerTest {