feat: allow manager to pause the deposit

alpaca-finance · Jan 19, 2024 · c6647d2 · c6647d2
1 parent b2f6bb8
commit c6647d2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/AutomatedVaultManager.sol b/src/AutomatedVaultManager.sol
@@ -584,6 +584,16 @@ contract AutomatedVaultManager is Initializable, Ownable2StepUpgradeable, Reentr
     }
   }
 
+  function pauseDeposit(address _vaultToken, bool _isPaused) external {
+    if (!isManager[_vaultToken][msg.sender]) {
+      revert AutomatedVaultManager_Unauthorized();
+    }
+
+    vaultInfos[_vaultToken].isDepositPaused = _isPaused;
+
+    emit LogSetIsDepositPaused(_vaultToken, _isPaused);
+  }
+
   function setIsWithdrawPaused(address[] calldata _vaultTokens, bool _isPaused) external onlyOwner {
     uint256 _len = _vaultTokens.length;
     for (uint256 _i; _i < _len;) {

diff --git a/test/automated-vault-manager/AutomatedVaultManager_Deposit.t.sol b/test/automated-vault-manager/AutomatedVaultManager_Deposit.t.sol
@@ -28,6 +28,17 @@ contract AutomatedVaultManagerDepositTest is BaseAutomatedVaultUnitTest {
     vaultManager.deposit(address(this), address(_vaultToken), _depositParams, 0);
   }
 
+  function testRevert_WhenDepositIsEmergencyPausedByManager() public {
+    address _vaultToken = _openDefaultVault();
+
+    vm.prank(MANAGER);
+    vaultManager.pauseDeposit(_vaultToken, true);
+
+    AutomatedVaultManager.TokenAmount[] memory _depositParams = new AutomatedVaultManager.TokenAmount[](0);
+    vm.expectRevert(abi.encodeWithSignature("AutomatedVaultManager_EmergencyPaused()"));
+    vaultManager.deposit(address(this), address(_vaultToken), _depositParams, 0);
+  }
+
   function testRevert_WhenDepositTokenThatIsNotAllowed() public {
     address vaultToken = _openDefaultVault();
     vm.prank(DEPLOYER);