govstaking/test/mocks/MockFullEarningPowerCalculator.sol

MockFullEarningPowerCalculator.sol

@@ -0,0 +1,84 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+pragma solidity ^0.8.23;
+
+import {IEarningPowerCalculator} from "src/interfaces/IEarningPowerCalculator.sol";
+
+contract MockFullEarningPowerCalculator is IEarningPowerCalculator {
+  uint256 public multiplierBips = 10_000;
+  bool public useMultiplier;
+  uint256 public fixedReturnValue;
+  bool public useFixedReturn;
+
+  struct StoredEarningPower {
+    uint256 earningPower;
+    bool isQualified;
+    bool isOverride;
+  }
+
+  mapping(address delegatee => StoredEarningPower earningPower) public earningPowerOverrides;
+
+  // Methods implementing the IEarningPowerCalculator interface.
+  function getEarningPower(
+    uint256 _amountStaked,
+    address, // _staker
+    address _delegatee
+  ) external view returns (uint256 _earningPower) {
+    (_earningPower,) = __getEarningPower(_amountStaked, _delegatee);
+  }
+
+  function getNewEarningPower(
+    uint256 _amountStaked,
+    address, // _staker
+    address _delegatee,
+    uint256 // _oldEarningPower
+  ) external view returns (uint256 _newEarningPower, bool _isQualifiedForUpdate) {
+    return __getEarningPower(_amountStaked, _delegatee);
+  }
+
+  // Methods used for configuring the mock during testing.
+  function __setEarningPowerForDelegatee(address _delegatee, uint256 _earningPower) external {
+    earningPowerOverrides[_delegatee] =
+      StoredEarningPower({earningPower: _earningPower, isQualified: true, isOverride: true});
+  }
+
+  function __setEarningPowerAndIsQualifiedForDelegatee(
+    address _delegatee,
+    uint256 _earningPower,
+    bool _isQualified
+  ) external {
+    earningPowerOverrides[_delegatee] =
+      StoredEarningPower({earningPower: _earningPower, isQualified: _isQualified, isOverride: true});
+  }
+
+  function __setMultiplierBips(uint256 _multiplierBips) external {
+    multiplierBips = _multiplierBips;
+    useMultiplier = true;
+  }
+
+  function __disableMultiplier() external {
+    useMultiplier = false;
+  }
+
+  function __setFixedReturn(uint256 _value) external {
+    fixedReturnValue = _value;
+    useFixedReturn = true;
+    useMultiplier = false;
+  }
+
+  function __getEarningPower(uint256 _amountStaked, address _delegatee)
+    internal
+    view
+    returns (uint256 _earningPower, bool _isQualified)
+  {
+    StoredEarningPower memory _storedEarningPower = earningPowerOverrides[_delegatee];
+
+    if (_storedEarningPower.isOverride) {
+      return (_storedEarningPower.earningPower, _storedEarningPower.isQualified);
+    } else {
+      if (useFixedReturn) return (fixedReturnValue, true);
+      uint256 _calculatedAmount =
+        useMultiplier ? (_amountStaked * multiplierBips) / 10_000 : _amountStaked;
+      return (_calculatedAmount, true);
+    }
+  }
+}