feat: add pause/unpause methods

This commit is contained in:
royvardhan
2025-01-24 18:12:56 +05:30
parent 70509f547d
commit c982ed99e8
4 changed files with 49 additions and 2 deletions

View File

@@ -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");
}

View File

@@ -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();
}
}

View File

@@ -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();