chore: duplicate logic into dashboard

lidofinance · Jan 16, 2025 · 765a205 · 765a205
1 parent ffcb109
commit 765a205
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 6 deletions.
diff --git a/contracts/0.8.25/vaults/Dashboard.sol b/contracts/0.8.25/vaults/Dashboard.sol
@@ -409,6 +409,20 @@ contract Dashboard is AccessControlEnumerable {
         _rebalanceVault(_ether);
     }
 
+    /**
+     * @notice Pauses beacon chain deposits on the staking vault.
+     */
+    function pauseBeaconChainDeposits() external virtual onlyRole(DEFAULT_ADMIN_ROLE) {
+        _pauseBeaconChainDeposits();
+    }
+
+    /**
+     * @notice Resumes beacon chain deposits on the staking vault.
+     */
+    function resumeBeaconChainDeposits() external virtual onlyRole(DEFAULT_ADMIN_ROLE) {
+        stakingVault.resumeBeaconChainDeposits();
+    }
+
     // ==================== Internal Functions ====================
 
     /**
@@ -514,6 +528,20 @@ contract Dashboard is AccessControlEnumerable {
         stakingVault.rebalance(_ether);
     }
 
+    /**
+     * @dev Pauses beacon chain deposits on the staking vault.
+     */
+    function _pauseBeaconChainDeposits() internal {
+        stakingVault.pauseBeaconChainDeposits();
+    }
+
+    /**
+     * @dev Resumes beacon chain deposits on the staking vault.
+     */
+    function _resumeBeaconChainDeposits() internal {
+        stakingVault.resumeBeaconChainDeposits();
+    }
+
     // ==================== Events ====================
 
     /// @notice Emitted when the contract is initialized

diff --git a/contracts/0.8.25/vaults/Delegation.sol b/contracts/0.8.25/vaults/Delegation.sol
@@ -345,15 +345,15 @@ contract Delegation is Dashboard {
     /**
      * @notice Pauses deposits to beacon chain from the StakingVault.
      */
-    function pauseBeaconChainDeposits() external onlyRole(CURATOR_ROLE) {
-        IStakingVault(stakingVault).pauseBeaconChainDeposits();
+    function pauseBeaconChainDeposits() external override onlyRole(CURATOR_ROLE) {
+        _pauseBeaconChainDeposits();
     }
 
     /**
      * @notice Resumes deposits to beacon chain from the StakingVault.
      */
-    function resumeBeaconChainDeposits() external onlyRole(CURATOR_ROLE) {
-        IStakingVault(stakingVault).resumeBeaconChainDeposits();
+    function resumeBeaconChainDeposits() external override onlyRole(CURATOR_ROLE) {
+        _resumeBeaconChainDeposits();
     }
 
     /**

diff --git a/test/0.8.25/vaults/dashboard/dashboard.test.ts b/test/0.8.25/vaults/dashboard/dashboard.test.ts
@@ -4,8 +4,7 @@ import { ZeroAddress } from "ethers";
 import { ethers } from "hardhat";
 
 import { HardhatEthersSigner } from "@nomicfoundation/hardhat-ethers/signers";
-import { time } from "@nomicfoundation/hardhat-network-helpers";
-import { setBalance } from "@nomicfoundation/hardhat-network-helpers";
+import { setBalance, time } from "@nomicfoundation/hardhat-network-helpers";
 
 import {
   Dashboard,
@@ -992,4 +991,50 @@ describe("Dashboard", () => {
         .withArgs(amount);
     });
   });
+
+  context("pauseBeaconChainDeposits", () => {
+    it("reverts if the caller is not a curator", async () => {
+      await expect(dashboard.connect(stranger).pauseBeaconChainDeposits()).to.be.revertedWithCustomError(
+        dashboard,
+        "AccessControlUnauthorizedAccount",
+      );
+    });
+
+    it("reverts if the beacon deposits are already paused", async () => {
+      await dashboard.pauseBeaconChainDeposits();
+
+      await expect(dashboard.pauseBeaconChainDeposits()).to.be.revertedWithCustomError(
+        vault,
+        "BeaconChainDepositsResumeExpected",
+      );
+    });
+
+    it("pauses the beacon deposits", async () => {
+      await expect(dashboard.pauseBeaconChainDeposits()).to.emit(vault, "BeaconChainDepositsPaused");
+      expect(await vault.beaconChainDepositsPaused()).to.be.true;
+    });
+  });
+
+  context("resumeBeaconChainDeposits", () => {
+    it("reverts if the caller is not a curator", async () => {
+      await expect(dashboard.connect(stranger).resumeBeaconChainDeposits()).to.be.revertedWithCustomError(
+        dashboard,
+        "AccessControlUnauthorizedAccount",
+      );
+    });
+
+    it("reverts if the beacon deposits are already resumed", async () => {
+      await expect(dashboard.resumeBeaconChainDeposits()).to.be.revertedWithCustomError(
+        vault,
+        "BeaconChainDepositsPauseExpected",
+      );
+    });
+
+    it("resumes the beacon deposits", async () => {
+      await dashboard.pauseBeaconChainDeposits();
+
+      await expect(dashboard.resumeBeaconChainDeposits()).to.emit(vault, "BeaconChainDepositsResumed");
+      expect(await vault.beaconChainDepositsPaused()).to.be.false;
+    });
+  });
 });