can update signer wallet

DoG-Protocol · Mar 7, 2024 · 67bf288 · 67bf288
1 parent d234934
commit 67bf288
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/contracts/DoGClaim.sol b/contracts/DoGClaim.sol
@@ -84,6 +84,13 @@ contract DoGClaim is AccessControlUpgradeable {
         emit BalanceLoaded(_msgSender(), amount, _balance);
     }
 
+    function updateSignerWallet(address newSignerWallet) public onlyRole(ADMIN_ROLE) {
+        if (newSignerWallet == address(0)) {
+            revert InvalidAddress(newSignerWallet);
+        }
+        signer = newSignerWallet;
+    }
+
     function updateFeeWallet(address newFeeWallet) public onlyRole(ADMIN_ROLE) {
         if (newFeeWallet == address(0)) {
             revert InvalidAddress(newFeeWallet);

diff --git a/test/DoGClaim.t.sol b/test/DoGClaim.t.sol
@@ -424,6 +424,21 @@ contract DoGClaimTest is Test {
         dogClaim.updateFeeWallet(sender);
     }
 
+    function test_updateSignerWallet() public {
+        address sender = vm.addr(2);
+
+        vm.prank(sender);
+        vm.expectRevert();
+        dogClaim.updateSignerWallet(sender);
+
+        vm.prank(admin);
+        vm.expectRevert();
+        dogClaim.updateSignerWallet(address(0));
+
+        vm.prank(admin);
+        dogClaim.updateSignerWallet(sender);
+    }
+
     function test_updateFeeRate() public {
         address sender = vm.addr(2);