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