add more connector tests

zeta-chain · Aug 7, 2024 · 8c3e4f6 · 8c3e4f6
1 parent 9b0385e
commit 8c3e4f6
Show file tree

Hide file tree

Showing 2 changed files with 75 additions and 0 deletions.
diff --git a/v2/test/ZetaConnectorNative.t.sol b/v2/test/ZetaConnectorNative.t.sol
@@ -40,9 +40,11 @@ contract ZetaConnectorNativeTest is
     address destination;
     address tssAddress;
 
+    error EnforcedPause();
     error AccessControlUnauthorizedAccount(address account, bytes32 neededRole);
 
     bytes32 public constant WITHDRAWER_ROLE = keccak256("WITHDRAWER_ROLE");
+    bytes32 public constant PAUSER_ROLE = keccak256("PAUSER_ROLE");
 
     function setUp() public {
         owner = address(this);
@@ -88,6 +90,41 @@ contract ZetaConnectorNativeTest is
         assertEq(balanceAfter, amount);
     }
 
+    function testWithdrawTogglePause() public {
+        uint256 amount = 100_000;
+        bytes32 internalSendHash = "";
+
+        vm.prank(tssAddress);
+        vm.expectRevert(abi.encodeWithSelector(AccessControlUnauthorizedAccount.selector, tssAddress, PAUSER_ROLE));
+        zetaConnector.pause();
+
+        vm.prank(tssAddress);
+        vm.expectRevert(abi.encodeWithSelector(AccessControlUnauthorizedAccount.selector, tssAddress, PAUSER_ROLE));
+        zetaConnector.unpause();
+
+        vm.prank(owner);
+        zetaConnector.pause();
+
+        vm.expectRevert(EnforcedPause.selector);
+        vm.prank(tssAddress);
+        zetaConnector.withdraw(destination, amount, internalSendHash);
+
+        vm.prank(owner);
+        zetaConnector.unpause();
+
+        uint256 balanceBefore = zetaToken.balanceOf(destination);
+        assertEq(balanceBefore, 0);
+
+        bytes memory data = abi.encodeWithSignature("transfer(address,uint256)", destination, amount);
+        vm.expectCall(address(zetaToken), 0, data);
+        vm.expectEmit(true, true, true, true, address(zetaConnector));
+        emit Withdraw(destination, amount);
+        vm.prank(tssAddress);
+        zetaConnector.withdraw(destination, amount, internalSendHash);
+        uint256 balanceAfter = zetaToken.balanceOf(destination);
+        assertEq(balanceAfter, amount);
+    }
+
     function testWithdrawFailsIfSenderIsNotWithdrawer() public {
         uint256 amount = 100_000;
         bytes32 internalSendHash = "";

diff --git a/v2/test/ZetaConnectorNonNative.t.sol b/v2/test/ZetaConnectorNonNative.t.sol
@@ -42,9 +42,11 @@ contract ZetaConnectorNonNativeTest is
 
     error AccessControlUnauthorizedAccount(address account, bytes32 neededRole);
     error ExceedsMaxSupply();
+    error EnforcedPause();
 
     bytes32 public constant WITHDRAWER_ROLE = keccak256("WITHDRAWER_ROLE");
     bytes32 public constant TSS_ROLE = keccak256("TSS_ROLE");
+    bytes32 public constant PAUSER_ROLE = keccak256("PAUSER_ROLE");
 
     function setUp() public {
         owner = address(this);
@@ -91,6 +93,42 @@ contract ZetaConnectorNonNativeTest is
         assertEq(balanceAfter, amount);
     }
 
+    function testWithdrawTogglePause() public {
+        uint256 amount = 100_000;
+        bytes32 internalSendHash = "";
+
+        vm.prank(tssAddress);
+        vm.expectRevert(abi.encodeWithSelector(AccessControlUnauthorizedAccount.selector, tssAddress, PAUSER_ROLE));
+        zetaConnector.pause();
+
+        vm.prank(tssAddress);
+        vm.expectRevert(abi.encodeWithSelector(AccessControlUnauthorizedAccount.selector, tssAddress, PAUSER_ROLE));
+        zetaConnector.unpause();
+
+        vm.prank(owner);
+        zetaConnector.pause();
+
+        vm.expectRevert(EnforcedPause.selector);
+        vm.prank(tssAddress);
+        zetaConnector.withdraw(destination, amount, internalSendHash);
+
+        vm.prank(owner);
+        zetaConnector.unpause();
+
+        uint256 balanceBefore = zetaToken.balanceOf(destination);
+        assertEq(balanceBefore, 0);
+
+        bytes memory data =
+            abi.encodeWithSignature("mint(address,uint256,bytes32)", destination, amount, internalSendHash);
+        vm.expectCall(address(zetaToken), 0, data);
+        vm.expectEmit(true, true, true, true, address(zetaConnector));
+        emit Withdraw(destination, amount);
+        vm.prank(tssAddress);
+        zetaConnector.withdraw(destination, amount, internalSendHash);
+        uint256 balanceAfter = zetaToken.balanceOf(destination);
+        assertEq(balanceAfter, amount);
+    }
+
     function testWithdrawFailsIfSenderIsNotWithdrawer() public {
         uint256 amount = 100_000;
         bytes32 internalSendHash = "";