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

add explicit emergency mode tests for admin and guardian

Browse Source
This commit is contained in:
Filip Pajic
2025-08-28 12:33:23 +02:00
parent b9809965e9
commit e39b20a959
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/RewardsStreamerMP.t.sol
View File

@@ -31,6 +31,7 @@ contract StakeManagerTest is StakeMath, Test {
address internal bob = makeAddr("bob"); address internal bob = makeAddr("bob");
address internal charlie = makeAddr("charlie"); address internal charlie = makeAddr("charlie");
address internal dave = makeAddr("dave"); address internal dave = makeAddr("dave");
address internal guardian = makeAddr("guardian");
mapping(address owner => address vault) public vaults; mapping(address owner => address vault) public vaults;
@@ -51,6 +52,7 @@ contract StakeManagerTest is StakeMath, Test {
vm.startPrank(admin); vm.startPrank(admin);
karma.addRewardDistributor(address(streamer)); karma.addRewardDistributor(address(streamer));
streamer.setRewardsSupplier(address(karma)); streamer.setRewardsSupplier(address(karma));
streamer.setGuardian(address(guardian));
vm.stopPrank(); vm.stopPrank();
address[4] memory accounts = [alice, bob, charlie, dave]; address[4] memory accounts = [alice, bob, charlie, dave];
@@ -2015,6 +2017,16 @@ contract EmergencyExitTest is StakeManagerTest {
_emergencyExit(alice); _emergencyExit(alice);
} }
function test_OwnerCanEnableEmergencyMode() public {
vm.prank(admin);
streamer.enableEmergencyMode();
}
function test_GuardianCanEnableEmergencyMode() public {
vm.prank(guardian);
streamer.enableEmergencyMode();
}
function test_OnlyOwnerOrGuardianCanEnableEmergencyMode() public { function test_OnlyOwnerOrGuardianCanEnableEmergencyMode() public {
vm.prank(alice); vm.prank(alice);
vm.expectRevert(IStakeManager.StakeManager__Unauthorized.selector); vm.expectRevert(IStakeManager.StakeManager__Unauthorized.selector);