feat: utility for removing shares from a strategy

Recon-Fuzz · May 31, 2024 · 7de3c59 · 7de3c59
1 parent 8d0c93b
commit 7de3c59
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/test/recon/EigenLayerSetupV2.sol b/src/test/recon/EigenLayerSetupV2.sol
@@ -289,6 +289,12 @@ contract EigenLayerSetupV2 {
         strategyManager.addStrategiesToDepositWhitelist(deployedStrategiestemp, thirdPartyTransfers);
     }
 
+    // @notice utility for removing shares from a strategy because IStrategy interface used on Renzo causes conflicts when trying to call removeShares directly
+    function _removeSharesFromStrategyManager(address staker, address strategy, uint256 sharesToRemove) internal {
+        IStrategy strategyToRemoveFrom = IStrategy(strategy);
+        strategyManager.removeShares(staker, strategyToRemoveFrom, sharesToRemove);
+    }
+
     // @audit this function sets contract addresses with those deployed on mainnet
     function _setAddresses() internal {
         // eigenLayerProxyAdminAddress = stdJson.readAddress(config, ".addresses.eigenLayerProxyAdmin");