vacp2p/staking-reward-streamer
1
0
Fork 0
mirror of https://github.com/vacp2p/staking-reward-streamer.git synced 2026-01-08 20:48:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(tests): use UndefinedError in some cases

Browse Source
This commit is contained in:
Ricardo Guilherme Schmidt
2025-03-21 13:55:58 -03:00
committed by r4bbit
parent 400923aed0
commit 0385522821
3 changed files with 83 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
.gas-report
View File

@@ -176,7 +176,7 @@
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| MIN_LOCKUP_PERIOD | 309 | 309 | 309 | 309 | 15 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| compound | 32664 | 33959 | 32664 | 116059 | 514 |
| compound | 32664 | 33940 | 32664 | 116059 | 514 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| compoundAccount | 369662 | 369662 | 369662 | 369662 | 1 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
@@ -192,7 +192,7 @@
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| getStakedBalance | 2643 | 2643 | 2643 | 2643 | 1 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| getVault | 2139 | 7731 | 2139 | 20139 | 4665 |
| getVault | 2139 | 7738 | 2139 | 20139 | 4666 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| initialize | 92753 | 92753 | 92753 | 92753 | 95 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
@@ -200,7 +200,7 @@
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| leave | 95106 | 95106 | 95106 | 95106 | 1 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| lock | 7062 | 38789 | 59606 | 110806 | 1034 |
| lock | 7062 | 38963 | 59606 | 110806 | 1034 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| migrateToVault | 13505 | 73541 | 15712 | 191406 | 3 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
@@ -218,7 +218,7 @@
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| rewardStartTime | 430 | 1430 | 1430 | 2430 | 2 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| rewardsBalanceOf | 2340 | 3562 | 3953 | 6340 | 268 |
| rewardsBalanceOf | 2340 | 3544 | 3953 | 6340 | 268 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| rewardsBalanceOfAccount | 10242 | 10242 | 10242 | 10242 | 1 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
@@ -228,29 +228,29 @@
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| setTrustedCodehash | 24194 | 24194 | 24194 | 24194 | 95 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| stake | 2639 | 151049 | 82444 | 268529 | 2667 |
| stake | 2639 | 151014 | 82444 | 268529 | 2667 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalMP | 805 | 1257 | 1257 | 1710 | 6 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalMPAccrued | 385 | 982 | 385 | 2385 | 4671 |
| totalMPAccrued | 385 | 983 | 385 | 2385 | 4672 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalMPStaked | 429 | 1025 | 429 | 2429 | 4674 |
| totalMPStaked | 429 | 1026 | 429 | 2429 | 4675 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalMaxMP | 407 | 1004 | 407 | 2407 | 4671 |
| totalMaxMP | 407 | 1005 | 407 | 2407 | 4672 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalRewardsAccrued | 429 | 429 | 429 | 429 | 3 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalRewardsSupply | 1020 | 1654 | 1814 | 6759 | 290 |
| totalRewardsSupply | 1020 | 1646 | 1814 | 6759 | 290 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalShares | 662 | 662 | 662 | 662 | 6 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| totalStaked | 430 | 1026 | 430 | 2430 | 4678 |
| totalStaked | 430 | 1027 | 430 | 2430 | 4679 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| unstake | 39196 | 42270 | 39222 | 80792 | 269 |
| unstake | 39196 | 42183 | 39222 | 80792 | 269 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| updateGlobalState | 15820 | 17253 | 15820 | 29516 | 1020 |
| updateGlobalState | 15820 | 17250 | 15820 | 29516 | 1021 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| updateVaultMP | 22753 | 24225 | 22753 | 36507 | 1020 |
| updateVaultMP | 22753 | 24222 | 22753 | 36507 | 1021 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
| upgradeTo | 10235 | 10857 | 10235 | 12723 | 4 |
|------------------------------------------------------+-----------------+--------+--------+--------+---------|
@@ -278,7 +278,7 @@
|----------------------------------------+-----------------+--------+--------+--------+---------|
| leave | 12161 | 128077 | 69651 | 360844 | 4 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| lock | 12091 | 54214 | 75072 | 126269 | 1035 |
| lock | 12091 | 54388 | 75072 | 126269 | 1035 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| migrateToVault | 28993 | 100223 | 31200 | 240476 | 3 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
@@ -286,13 +286,13 @@
|----------------------------------------+-----------------+--------+--------+--------+---------|
| register | 12674 | 78057 | 78600 | 78600 | 374 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| stake | 12071 | 185366 | 97937 | 324097 | 2668 |
| stake | 12071 | 185300 | 97937 | 324097 | 2668 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| stakeManager | 369 | 369 | 369 | 369 | 373 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| trustStakeManager | 7580 | 7580 | 7580 | 7580 | 1 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| unstake | 12048 | 58640 | 54665 | 111814 | 270 |
| unstake | 12048 | 58554 | 54665 | 111814 | 270 |
|----------------------------------------+-----------------+--------+--------+--------+---------|
| withdraw | 20754 | 20754 | 20754 | 20754 | 1 |
╰----------------------------------------+-----------------+--------+--------+--------+---------╯
@@ -308,7 +308,7 @@
|----------------------------------------------------+-----------------+------+--------+--------+---------|
| Function Name | Min | Avg | Median | Max | # Calls |
|----------------------------------------------------+-----------------+------+--------+--------+---------|
| fallback | 708 | 8135 | 2629 | 396039 | 27207 |
| fallback | 708 | 8138 | 2629 | 396039 | 27214 |
|----------------------------------------------------+-----------------+------+--------+--------+---------|
| implementation | 346 | 2345 | 2346 | 2346 | 4355 |
╰----------------------------------------------------+-----------------+------+--------+--------+---------╯
@@ -332,7 +332,7 @@
+===============================================================================================================================================+
| Deployment Cost | Deployment Size | | | | |
|------------------------------------------------------------------------------------------+-----------------+-------+--------+-------+---------|
| 1204853 | 6015 | | | | |
| 1204853 | 6207 | | | | |
|------------------------------------------------------------------------------------------+-----------------+-------+--------+-------+---------|
| | | | | | |
|------------------------------------------------------------------------------------------+-----------------+-------+--------+-------+---------|
@@ -412,11 +412,11 @@
|---------------------------------------------+-----------------+-------+--------+-------+---------|
| Function Name | Min | Avg | Median | Max | # Calls |
|---------------------------------------------+-----------------+-------+--------+-------+---------|
| approve | 29075 | 31548 | 29183 | 46259 | 2676 |
| approve | 29075 | 31547 | 29183 | 46259 | 2676 |
|---------------------------------------------+-----------------+-------+--------+-------+---------|
| balanceOf | 561 | 1266 | 561 | 2561 | 5465 |
| balanceOf | 561 | 1266 | 561 | 2561 | 5466 |
|---------------------------------------------+-----------------+-------+--------+-------+---------|
| mint | 33964 | 37193 | 34072 | 68248 | 2685 |
| mint | 33964 | 37192 | 34072 | 68248 | 2685 |
╰---------------------------------------------+-----------------+-------+--------+-------+---------╯
╭-----------------------------------------------------------------------------+-----------------+--------+--------+--------+---------╮

72
.gas-snapshot
View File

@@ -1,22 +1,22 @@
CompoundTest:test_CompoundForAccount() (gas: 2493561)
CompoundTest:test_RevertWhenInsufficientMPBalance() (gas: 382040)
CompoundTest:test_CompoundForAccount() (gas: 2493583)
CompoundTest:test_RevertWhenInsufficientMPBalance() (gas: 382062)
EmergencyExitTest:test_CannotEnableEmergencyModeTwice() (gas: 93420)
EmergencyExitTest:test_CannotLeaveBeforeEmergencyMode() (gas: 355381)
EmergencyExitTest:test_EmergencyExitBasic() (gas: 450301)
EmergencyExitTest:test_EmergencyExitMultipleUsers() (gas: 776038)
EmergencyExitTest:test_CannotLeaveBeforeEmergencyMode() (gas: 355403)
EmergencyExitTest:test_EmergencyExitBasic() (gas: 450256)
EmergencyExitTest:test_EmergencyExitMultipleUsers() (gas: 776060)
EmergencyExitTest:test_EmergencyExitToAlternateAddress() (gas: 455763)
EmergencyExitTest:test_EmergencyExitWithLock() (gas: 473147)
EmergencyExitTest:test_EmergencyExitWithLock() (gas: 473103)
EmergencyExitTest:test_EmergencyExitWithRewards() (gas: 438555)
EmergencyExitTest:test_OnlyOwnerCanEnableEmergencyMode() (gas: 39109)
FuzzTests:testFuzz_AccrueMP(uint128,uint64,uint64) (runs: 1023, μ: 433295, ~: 407385)
FuzzTests:testFuzz_AccrueMP_Relock(uint128,uint64,uint64,uint64) (runs: 1023, μ: 535458, ~: 509184)
FuzzTests:testFuzz_Compound(uint128,uint64,uint64) (runs: 1023, μ: 521411, ~: 494881)
FuzzTests:testFuzz_EmergencyExit(uint256,uint256) (runs: 1007, μ: 525759, ~: 507538)
FuzzTests:testFuzz_Lock(uint256,uint64) (runs: 1023, μ: 772579, ~: 793973)
FuzzTests:testFuzz_Relock(uint256,uint64,uint64) (runs: 1023, μ: 410341, ~: 392608)
FuzzTests:testFuzz_Rewards(uint256,uint256,uint256,uint16,uint16) (runs: 1000, μ: 634354, ~: 636257)
FuzzTests:testFuzz_Stake(uint256,uint64) (runs: 1023, μ: 312467, ~: 290859)
FuzzTests:testFuzz_Unstake(uint128,uint64,uint16,uint128) (runs: 1023, μ: 620437, ~: 599309)
FuzzTests:testFuzz_AccrueMP(uint128,uint64,uint64) (runs: 1024, μ: 434256, ~: 407334)
FuzzTests:testFuzz_AccrueMP_Relock(uint128,uint64,uint64,uint64) (runs: 1024, μ: 533518, ~: 509224)
FuzzTests:testFuzz_Compound(uint128,uint64,uint64) (runs: 1024, μ: 522387, ~: 494820)
FuzzTests:testFuzz_EmergencyExit(uint256,uint256) (runs: 1007, μ: 528440, ~: 509976)
FuzzTests:testFuzz_Lock(uint256,uint64) (runs: 1025, μ: 772594, ~: 793906)
FuzzTests:testFuzz_Relock(uint256,uint64,uint64) (runs: 1025, μ: 408965, ~: 392534)
FuzzTests:testFuzz_Rewards(uint256,uint256,uint256,uint16,uint16) (runs: 1001, μ: 636843, ~: 638758)
FuzzTests:testFuzz_Stake(uint256,uint64) (runs: 1025, μ: 314299, ~: 290763)
FuzzTests:testFuzz_Unstake(uint128,uint64,uint16,uint128) (runs: 1024, μ: 621954, ~: 599327)
IntegrationTest:testStakeFoo() (gas: 1441361)
KarmaMintAllowanceTest:testAddKarmaDistributorOnlyOwner() (gas: 364802)
KarmaMintAllowanceTest:testBalanceOf() (gas: 441486)
@@ -61,19 +61,19 @@ KarmaTest:testTransfersNotAllowed() (gas: 40241)
LeaveTest:test_LeaveShouldProperlyUpdateAccounting() (gas: 9899202)
LeaveTest:test_RevertWhenStakeManagerIsTrusted() (gas: 352518)
LeaveTest:test_TrustNewStakeManager() (gas: 9949960)
LockTest:test_LockFailsWithInvalidPeriod(uint256) (runs: 1024, μ: 387855, ~: 366605)
LockTest:test_LockFailsWithInvalidPeriod(uint256) (runs: 1025, μ: 388844, ~: 366617)
LockTest:test_LockFailsWithNoStake() (gas: 109665)
LockTest:test_LockFailsWithZero() (gas: 362624)
LockTest:test_LockFailsWithZero() (gas: 362602)
LockTest:test_LockMultipleTimesExceedMaxLock() (gas: 674397)
LockTest:test_LockWithPriorLock() (gas: 621342)
LockTest:test_LockWithoutPriorLock() (gas: 493744)
LockTest:test_RevertWhenVaultToLockIsEmpty() (gas: 109623)
MaliciousUpgradeTest:test_UpgradeStackOverflowStakeManager() (gas: 2018869)
MathTest:test_CalcAbsoluteMaxTotalMP() (gas: 5218)
MathTest:test_CalcAbsoluteMaxTotalMP() (gas: 5240)
MathTest:test_CalcAccrueMP() (gas: 8599)
MathTest:test_CalcBonusMP() (gas: 19178)
MathTest:test_CalcInitialMP() (gas: 5836)
MathTest:test_CalcMaxAccruedMP() (gas: 4929)
MathTest:test_CalcBonusMP() (gas: 19200)
MathTest:test_CalcInitialMP() (gas: 5794)
MathTest:test_CalcMaxAccruedMP() (gas: 4886)
MathTest:test_CalcMaxTotalMP() (gas: 19940)
MultipleVaultsStakeTest:test_StakeMultipleVaults() (gas: 860071)
NFTMetadataGeneratorSVGTest:testGenerateMetadata() (gas: 92580)
@@ -82,49 +82,49 @@ NFTMetadataGeneratorSVGTest:testSetImageStringsRevert() (gas: 35891)
NFTMetadataGeneratorURLTest:testGenerateMetadata() (gas: 108341)
NFTMetadataGeneratorURLTest:testSetBaseURL() (gas: 50631)
NFTMetadataGeneratorURLTest:testSetBaseURLRevert() (gas: 36066)
RewardsStreamerMP_RewardsTest:testRewardsBalanceOf() (gas: 1326353)
RewardsStreamerMP_RewardsTest:testRewardsBalanceOf() (gas: 1326310)
RewardsStreamerMP_RewardsTest:testSetRewards() (gas: 224746)
RewardsStreamerMP_RewardsTest:testSetRewards_RevertsBadAmount() (gas: 61324)
RewardsStreamerMP_RewardsTest:testSetRewards_RevertsBadDuration() (gas: 101082)
RewardsStreamerMP_RewardsTest:testSetRewards_RevertsBadAmount() (gas: 61282)
RewardsStreamerMP_RewardsTest:testSetRewards_RevertsBadDuration() (gas: 101104)
RewardsStreamerMP_RewardsTest:testSetRewards_RevertsNotAuthorized() (gas: 39344)
RewardsStreamerMP_RewardsTest:testTotalRewardsSupply() (gas: 761342)
RewardsStreamerMP_RewardsTest:testTotalRewardsSupply() (gas: 761364)
StakeTest:test_RevertWhenStakeMultipleTimesExceedsMaxLockUpTime() (gas: 774528)
StakeTest:test_RevertWhenStakeMultipleTimesWithGapsExceedsMaxLockUpTime() (gas: 826669)
StakeTest:test_StakeMultipleAccounts() (gas: 601877)
StakeTest:test_StakeMultipleAccountsAndRewards() (gas: 610415)
StakeTest:test_StakeMultipleAccountsMPIncreasesMaxMPDoesNotChange() (gas: 1039884)
StakeTest:test_StakeMultipleAccountsWithMinLockUp() (gas: 630238)
StakeTest:test_StakeMultipleAccountsWithMinLockUp() (gas: 630260)
StakeTest:test_StakeMultipleAccountsWithRandomLockUp() (gas: 674876)
StakeTest:test_StakeMultipleTimesWithLockIncreaseAtSameBlock() (gas: 623415)
StakeTest:test_StakeMultipleTimesWithLockZeroAfterMaxLock() (gas: 739368)
StakeTest:test_StakeOneAccount() (gas: 339156)
StakeTest:test_StakeOneAccountAndRewards() (gas: 347734)
StakeTest:test_StakeOneAccountMPIncreasesMaxMPDoesNotChange() (gas: 615073)
StakeTest:test_StakeOneAccountReachingMPLimit() (gas: 603204)
StakeTest:test_StakeOneAccountAndRewards() (gas: 347756)
StakeTest:test_StakeOneAccountMPIncreasesMaxMPDoesNotChange() (gas: 615095)
StakeTest:test_StakeOneAccountReachingMPLimit() (gas: 603226)
StakeTest:test_StakeOneAccountWithMaxLockUp() (gas: 377449)
StakeTest:test_StakeOneAccountWithMinLockUp() (gas: 378045)
StakeTest:test_StakeOneAccountWithRandomLockUp() (gas: 378090)
StakeVaultMigrationTest:testMigrateToVault() (gas: 958888)
StakeVaultMigrationTest:testMigrateToVault() (gas: 958866)
StakeVaultMigrationTest:test_RevertWhenMigrationVaultNotEmpty() (gas: 648499)
StakeVaultMigrationTest:test_RevertWhenNotOwnerOfMigrationVault() (gas: 68044)
StakeVaultTest:testOwner() (gas: 15262)
StakingTokenTest:testOwner() (gas: 15262)
StakingTokenTest:testStakeToken() (gas: 13144)
TrustedCodehashAccessTest:test_RevertWhenProxyCloneCodehashNotTrusted() (gas: 2023424)
UnstakeTest:test_RevertWhenStakeMultipleTimesExceedsMaxLockUpTime() (gas: 774550)
UnstakeTest:test_RevertWhenStakeMultipleTimesExceedsMaxLockUpTime() (gas: 774528)
UnstakeTest:test_RevertWhenStakeMultipleTimesWithGapsExceedsMaxLockUpTime() (gas: 826713)
UnstakeTest:test_StakeMultipleAccounts() (gas: 601834)
UnstakeTest:test_StakeMultipleAccountsAndRewards() (gas: 610370)
UnstakeTest:test_StakeMultipleAccountsAndRewards() (gas: 610437)
UnstakeTest:test_StakeMultipleAccountsMPIncreasesMaxMPDoesNotChange() (gas: 1039928)
UnstakeTest:test_StakeMultipleAccountsWithMinLockUp() (gas: 630304)
UnstakeTest:test_StakeMultipleAccountsWithRandomLockUp() (gas: 674920)
UnstakeTest:test_StakeMultipleTimesWithLockIncreaseAtSameBlock() (gas: 623437)
UnstakeTest:test_StakeMultipleTimesWithLockIncreaseAtSameBlock() (gas: 623415)
UnstakeTest:test_StakeMultipleTimesWithLockZeroAfterMaxLock() (gas: 739368)
UnstakeTest:test_StakeOneAccount() (gas: 339156)
UnstakeTest:test_StakeOneAccountAndRewards() (gas: 347733)
UnstakeTest:test_StakeOneAccountMPIncreasesMaxMPDoesNotChange() (gas: 615072)
UnstakeTest:test_StakeOneAccountReachingMPLimit() (gas: 603184)
UnstakeTest:test_StakeOneAccountWithMaxLockUp() (gas: 377471)
UnstakeTest:test_StakeOneAccountWithMaxLockUp() (gas: 377449)
UnstakeTest:test_StakeOneAccountWithMinLockUp() (gas: 378045)
UnstakeTest:test_StakeOneAccountWithRandomLockUp() (gas: 378156)
UnstakeTest:test_UnstakeBonusMPAndAccuredMP() (gas: 655966)
@@ -134,8 +134,8 @@ UnstakeTest:test_UnstakeOneAccount() (gas: 575842)
UnstakeTest:test_UnstakeOneAccountAndAccruedMP() (gas: 600078)
UnstakeTest:test_UnstakeOneAccountAndRewards() (gas: 496119)
UnstakeTest:test_UnstakeOneAccountWithLockUpAndAccruedMP() (gas: 642865)
UpgradeTest:test_RevertWhenNotOwner() (gas: 3706859)
UpgradeTest:test_UpgradeStakeManager() (gas: 9786172)
UpgradeTest:test_RevertWhenNotOwner() (gas: 3706881)
UpgradeTest:test_UpgradeStakeManager() (gas: 9786130)
VaultRegistrationTest:test_VaultRegistration() (gas: 63226)
WithdrawTest:testOwner() (gas: 15296)
WithdrawTest:test_CannotWithdrawStakedFunds() (gas: 368209)

67
test/RewardsStreamerMP.t.sol
View File

@@ -163,15 +163,6 @@ contract RewardsStreamerMPTest is StakeMath, Test {
vault.stake(amount, lockupTime);
}
function _stake(address account, uint256 amount, uint256 lockupTime, bytes4 _expectedRevert) public virtual {
StakeVault vault = StakeVault(vaults[account]);
vm.prank(account);
if (_expectedRevert != bytes4(0)) {
vm.expectRevert(_expectedRevert);
}
vault.stake(amount, lockupTime);
}
function _compound(address account) public {
StakeVault vault = StakeVault(vaults[account]);
streamer.compound(address(vault));
@@ -3018,36 +3009,38 @@ contract FuzzTests is RewardsStreamerMPTest {
);
}
function _stake(address account, uint256 amount, uint256 lockPeriod, bytes4 _expectedRevert) public override {
function _stake(address account, uint256 amount, uint256 lockPeriod, bytes4 _expectedRevert) internal {
stakingToken.mint(account, amount);
vm.prank(account);
StakeVault vault = StakeVault(vaults[account]);
vm.startPrank(account);
stakingToken.approve(vaults[account], amount);
super._stake(account, amount, lockPeriod, _expectedRevert);
_expectRevert(_expectedRevert);
vault.stake(amount, lockPeriod);
vm.stopPrank();
expectedRevert = FuzzTests__UndefinedError.selector;
}
function _stake(address account, uint256 amount, uint256 lockPeriod) public override {
stakingToken.mint(account, amount);
vm.prank(account);
stakingToken.approve(vaults[account], amount);
super._stake(account, amount, lockPeriod);
}
function _lock(address account, uint256 lockPeriod, bytes4 _expectedRevert) internal {
StakeVault vault = StakeVault(vaults[account]);
vm.prank(account);
if (_expectedRevert != NO_REVERT) {
vm.expectRevert(_expectedRevert);
}
_expectRevert(_expectedRevert);
vault.lock(lockPeriod);
expectedRevert = FuzzTests__UndefinedError.selector;
}
function _expectRevert(bytes4 _expectedRevert) internal {
if (_expectedRevert != NO_REVERT) {
if (_expectedRevert == FuzzTests__UndefinedError.selector) {
vm.expectRevert();
} else {
vm.expectRevert(_expectedRevert);
}
}
}
function _compound(address account, bytes4 _expectedRevert) internal {
StakeVault vault = StakeVault(vaults[account]);
if (_expectedRevert != NO_REVERT) {
vm.expectRevert(_expectedRevert);
}
_expectRevert(_expectedRevert);
streamer.compound(address(vault));
}
@@ -3062,9 +3055,7 @@ contract FuzzTests is RewardsStreamerMPTest {
function _unstake(address account, uint256 amount, bytes4 _expectedRevert) internal {
StakeVault vault = StakeVault(vaults[account]);
vm.prank(account);
if (_expectedRevert != NO_REVERT) {
vm.expectRevert(_expectedRevert);
}
_expectRevert(_expectedRevert);
vault.unstake(amount);
expectedRevert = FuzzTests__UndefinedError.selector;
}
@@ -3165,11 +3156,7 @@ contract FuzzTests is RewardsStreamerMPTest {
expectedSystemState.totalMaxMP += expectedMaxTotalMP;
}
} else {
if (lockUpPeriod > MAX_LOCKUP_PERIOD) {
expectedRevert = IStakeManager.StakingManager__InvalidLockPeriod.selector;
} else {
expectedRevert = StakeMath.StakeMath__InvalidLockingPeriod.selector;
}
expectedRevert = FuzzTests__UndefinedError.selector;
return;
}
}
@@ -3194,11 +3181,7 @@ contract FuzzTests is RewardsStreamerMPTest {
) + lockUpPeriod;
uint256 calcLockUpPeriod = calcLockEnd - vm.getBlockTimestamp();
if (!(calcLockUpPeriod >= MIN_LOCKUP_PERIOD && calcLockUpPeriod <= MAX_LOCKUP_PERIOD)) {
if (calcLockUpPeriod > MAX_LOCKUP_PERIOD) {
expectedRevert = IStakeManager.StakingManager__InvalidLockPeriod.selector;
} else {
expectedRevert = StakeMath.StakeMath__InvalidLockingPeriod.selector;
}
expectedRevert = FuzzTests__UndefinedError.selector;
return;
}
if (expectedVaultLockState[expectedAccountParams.account].totalLockUp + lockUpPeriod > MAX_LOCKUP_PERIOD) {
@@ -3357,7 +3340,9 @@ contract FuzzTests is RewardsStreamerMPTest {
? Math.mulDiv(accountRewardPeriod, rewardAmount, rewardPeriod)
: rewardAmount;
_stake(alice, stakeAmount, lockUpPeriod);
expectedRevert = NO_REVERT;
_stake(alice, stakeAmount, lockUpPeriod, expectedRevert);
_setRewards(rewardAmount, rewardPeriod);
@@ -3376,8 +3361,8 @@ contract FuzzTests is RewardsStreamerMPTest {
uint256 aliceInitialBalance = stakingToken.balanceOf(alice);
uint256 expectedBonusMP = _bonusMP(stakeAmount, lockUpPeriod);
uint256 expectedMaxTotalMP = _maxTotalMP(stakeAmount, lockUpPeriod);
_stake(alice, stakeAmount, lockUpPeriod);
expectedRevert = NO_REVERT;
_stake(alice, stakeAmount, lockUpPeriod, expectedRevert);
vm.prank(admin);
streamer.enableEmergencyMode();