test: AllocationManager progress

src/contracts/core/AllocationManager.sol

@@ -282,21 +282,6 @@ contract AllocationManager is
         emit AllocationDelaySet(operator, delay, info.effectTimestamp);
     }
 
-    /**
-     * @param encumberedMagnitude the effective magnitude allocated to all operator sets
-     * for the strategy
-     * @param currentMagnitude the effective current magnitude allocated to a single operator set
-     * for the strategy
-     * @param pendingDiff the pending change in magnitude, if one exists
-     * @param effectTimestamp the time after which `pendingDiff` will take effect
-     */
-    struct PendingMagnitudeInfo {
-        uint64 encumberedMagnitude;
-        uint64 currentMagnitude;
-        int128 pendingDiff;
-        uint32 effectTimestamp;
-    }
-
     /**
      * @dev For an operator set, get the operator's effective allocated magnitude.
      * If the operator set has a pending modification that can be completed at the

src/contracts/interfaces/IAllocationManager.sol

@@ -93,6 +93,22 @@ interface IAllocationManagerTypes {
         uint256 wadToSlash;
         string description;
     }
+
+    /**
+     * @param encumberedMagnitude the effective magnitude allocated to all operator sets
+     * for the strategy
+     * @param currentMagnitude the effective current magnitude allocated to a single operator set
+     * for the strategy
+     * @param pendingDiff the pending change in magnitude, if one exists
+     * @param effectTimestamp the time after which `pendingDiff` will take effect
+     */
+    struct PendingMagnitudeInfo {
+        uint64 encumberedMagnitude;
+        uint64 currentMagnitude;
+        int128 pendingDiff;
+        uint32 effectTimestamp;
+    }
+
 }
 
 interface IAllocationManagerEvents is IAllocationManagerTypes {

src/test/mocks/AVSDirectoryMock.sol

@@ -2,8 +2,32 @@
 pragma solidity ^0.8.9;
 
 import "forge-std/Test.sol";
+import "src/contracts/interfaces/IAVSDirectory.sol";
 
 contract AVSDirectoryMock is Test {
     receive() external payable {}
     fallback() external payable {}
+
+    mapping(address => mapping(bytes32 => bool)) public _isOperatorSlashable;
+    mapping(bytes32 => bool) public _isOperatorSetBatch;
+
+    function isOperatorSlashable(address operator, OperatorSet memory operatorSet) public virtual view returns (bool) {
+        return _isOperatorSlashable[operator][bytes32(abi.encode(operatorSet))];
+    }
+
+    function isOperatorSetBatch(OperatorSet[] memory operatorSets) public virtual view returns (bool) {
+        return _isOperatorSetBatch[keccak256(abi.encode(operatorSets))];
+    }
+
+    function setIsOperatorSlashable(
+        address operator, 
+        OperatorSet memory operatorSet, 
+        bool value
+    ) public virtual {
+        _isOperatorSlashable[operator][bytes32(abi.encode(operatorSet))] = value;
+    }
+
+    function setIsOperatorSetBatch(OperatorSet[] memory operatorSets, bool value) public virtual {
+        _isOperatorSetBatch[keccak256(abi.encode(operatorSets))] = value;
+    }
 }