Slashing: Add additional happy path AM test cases (#828)

* test: happy path AM tests * chore: format
Layr-Labs · Oct 16, 2024 · 3f6adfb · 3f6adfb
1 parent 08cf29d
commit 3f6adfb
Show file tree

Hide file tree

Showing 3 changed files with 563 additions and 37 deletions.
diff --git a/src/contracts/interfaces/IStrategyManager.sol b/src/contracts/interfaces/IStrategyManager.sol
@@ -138,7 +138,6 @@ interface IStrategyManager is IStrategyManagerErrors, IStrategyManagerEvents, IS
         IStrategy[] calldata strategiesToRemoveFromWhitelist
     ) external;
 
-
     /// @notice Returns bool for whether or not `strategy` is whitelisted for deposit
     function strategyIsWhitelistedForDeposit(
         IStrategy strategy

diff --git a/src/test/mocks/DelegationManagerMock.sol b/src/test/mocks/DelegationManagerMock.sol
@@ -32,6 +32,17 @@ contract DelegationManagerMock is Test {
         return withdrawalRoot;
     }
 
+    function getOperatorsShares(address[] memory operators, IStrategy[] memory strategies) external view returns (uint256[][] memory) {
+        uint256[][] memory operatorSharesArray = new uint256[][](operators.length);
+        for (uint256 i = 0; i < operators.length; i++) {
+            operatorSharesArray[i] = new uint256[](strategies.length);
+            for (uint256 j = 0; j < strategies.length; j++) {
+                operatorSharesArray[i][j] = operatorShares[operators[i]][strategies[j]];
+            }
+        }
+        return operatorSharesArray;
+    }
+
     function operatorDetails(address operator) external pure returns (IDelegationManagerTypes.OperatorDetails memory) {
         IDelegationManagerTypes.OperatorDetails memory returnValue = IDelegationManagerTypes.OperatorDetails({
             __deprecated_earningsReceiver: operator,