feat: add pause/unpause methods

2025-01-24 18:12:56 +05:30
parent 70509f547d
commit c982ed99e8
4 changed files with 49 additions and 2 deletions
--- a/foundry/test/TychoRouter.t.sol
+++ b/foundry/test/TychoRouter.t.sol
@@ -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();
+    }
 }