feat: add pause/unpause methods
This commit is contained in:
@@ -11,4 +11,6 @@ contract Constants is Test {
|
||||
// dummy contracts
|
||||
address DUMMY = makeAddr("dummy");
|
||||
address FEE_RECEIVER = makeAddr("feeReceiver");
|
||||
address PAUSER = makeAddr("pauser");
|
||||
address UNPAUSER = makeAddr("unpauser");
|
||||
}
|
||||
|
||||
@@ -206,4 +206,29 @@ contract TychoRouterTest is TychoRouterTestSetup {
|
||||
tychoRouter.setFeeReceiver(FEE_RECEIVER);
|
||||
vm.stopPrank();
|
||||
}
|
||||
|
||||
function testPause() public {
|
||||
vm.startPrank(PAUSER);
|
||||
assertEq(tychoRouter.paused(), false);
|
||||
tychoRouter.pause();
|
||||
assertEq(tychoRouter.paused(), true);
|
||||
vm.stopPrank();
|
||||
|
||||
vm.startPrank(UNPAUSER);
|
||||
tychoRouter.unpause();
|
||||
assertEq(tychoRouter.paused(), false);
|
||||
vm.stopPrank();
|
||||
}
|
||||
|
||||
function testPauseFailures() public {
|
||||
vm.startPrank(BOB);
|
||||
vm.expectRevert();
|
||||
tychoRouter.pause();
|
||||
vm.stopPrank();
|
||||
|
||||
vm.startPrank(UNPAUSER);
|
||||
vm.expectRevert();
|
||||
tychoRouter.unpause();
|
||||
vm.stopPrank();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -17,6 +17,8 @@ contract TychoRouterTestSetup is Test, Constants {
|
||||
tychoRouter.grantRole(keccak256("EXECUTOR_SETTER_ROLE"), BOB);
|
||||
tychoRouter.grantRole(keccak256("FUND_RESCUER_ROLE"), FUND_RESCUER);
|
||||
tychoRouter.grantRole(keccak256("FEE_SETTER_ROLE"), FEE_SETTER);
|
||||
tychoRouter.grantRole(keccak256("PAUSER_ROLE"), PAUSER);
|
||||
tychoRouter.grantRole(keccak256("UNPAUSER_ROLE"), UNPAUSER);
|
||||
executorSetter = BOB;
|
||||
deployDummyContract();
|
||||
vm.stopPrank();
|
||||
|
||||
Reference in New Issue
Block a user