Soloseng/epoch-rewards-L2-test (#11270)

* initial tests * fixed EpochRewardsTest_getRewardsMultiplier_L2 tests * using whenL2 utils contract * removed duplicated test * fixed rest of failing tests * renamed function
celo-org · Nov 20, 2024 · c71a412 · c71a412
1 parent 1c68abf
commit c71a412
Show file tree

Hide file tree

Showing 2 changed files with 159 additions and 93 deletions.
diff --git a/packages/protocol/test-sol/unit/common/mocks/MockEpochManager.sol b/packages/protocol/test-sol/unit/common/mocks/MockEpochManager.sol
@@ -36,6 +36,7 @@ contract MockEpochManager is IEpochManager {
   uint256 private currentEpochNumber;
   address[] public electedAccounts;
   address[] public electedSigners;
+  uint256 numberOfElectedAccounts;
   address public epochManagerEnabler;
   bool systemInitialized;
 
@@ -95,6 +96,10 @@ contract MockEpochManager is IEpochManager {
     isProcessingEpoch = _isProcessing;
   }
 
+  function setNumberOfElectedInCurrentSet(uint256 value) external {
+    numberOfElectedAccounts = value;
+  }
+
   function getCurrentEpoch() external view returns (uint256, uint256, uint256, uint256) {
     return getEpochByNumber(currentEpochNumber);
   }
@@ -104,7 +109,7 @@ contract MockEpochManager is IEpochManager {
   }
 
   function numberOfElectedInCurrentSet() external view returns (uint256) {
-    return electedAccounts.length;
+    return numberOfElectedAccounts;
   }
 
   function getElectedAccounts() external view returns (address[] memory) {