refactor: getoperatorsetcount

Layr-Labs · Nov 26, 2024 · c24a46b · c24a46b
1 parent fca8750
commit c24a46b
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/src/contracts/core/AllocationManager.sol b/src/contracts/core/AllocationManager.sol
@@ -537,7 +537,7 @@ contract AllocationManager is
      */
 
     /// @inheritdoc IAllocationManager
-    function getAVSSetCount(
+    function getOperatorSetCount(
         address avs
     ) external view returns (uint256) {
         return _operatorSets[avs].length();

diff --git a/src/contracts/core/RewardsCoordinator.sol b/src/contracts/core/RewardsCoordinator.sol
@@ -228,7 +228,7 @@ contract RewardsCoordinator is
     /// @inheritdoc IRewardsCoordinator
     function setClaimerFor(address earner, address claimer) external checkCanCall(earner) {
         // Require that the earner is an operator or AVS
-        require(delegationManager.isOperator(earner) || allocationManager.getAVSSetCount(earner) > 0, InvalidEarner());
+        require(delegationManager.isOperator(earner) || allocationManager.getOperatorSetCount(earner) > 0, InvalidEarner());
         _setClaimer(earner, claimer);
     }
 

diff --git a/src/contracts/interfaces/IAllocationManager.sol b/src/contracts/interfaces/IAllocationManager.sol
@@ -334,7 +334,7 @@ interface IAllocationManager is ISignatureUtils, IAllocationManagerErrors, IAllo
      * @notice Returns the number of operator sets for the AVS
      * @param avs the AVS to query
      */
-    function getAVSSetCount(
+    function getOperatorSetCount(
         address avs
     ) external view returns (uint256);
 

diff --git a/src/test/mocks/AllocationManagerMock.sol b/src/test/mocks/AllocationManagerMock.sol
@@ -11,7 +11,7 @@ contract AllocationManagerMock is Test {
     receive() external payable {}
     fallback() external payable {}
 
-    mapping(address avs => uint256) public getAVSSetCount;
+    mapping(address avs => uint256) public getOperatorSetCount;
     mapping(address => mapping(IStrategy => Snapshots.DefaultWadHistory)) internal _maxMagnitudeHistory;
 
     function setMaxMagnitudes(
@@ -72,6 +72,6 @@ contract AllocationManagerMock is Test {
     }
 
     function setAVSSetCount(address avs, uint256 numSets) external {
-        getAVSSetCount[avs] = numSets;
+        getOperatorSetCount[avs] = numSets;
     }
 }
diff --git a/src/test/unit/AllocationManagerUnit.t.sol b/src/test/unit/AllocationManagerUnit.t.sol
@@ -3216,7 +3216,7 @@ contract AllocationManagerUnitTests_createOperatorSets is AllocationManagerUnitT
             }
         }
 
-        assertEq(createSetParams.length, allocationManager.getAVSSetCount(avs), "should be correct number of sets");
+        assertEq(createSetParams.length, allocationManager.getOperatorSetCount(avs), "should be correct number of sets");
     }
 }