Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 40 additions & 0 deletions src/test/BinaryEligibilityOracleEarningPowerCalculatorTestBase.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
// SPDX-License-Identifier: AGPL-3.0-only

pragma solidity ^0.8.23;

import {BinaryEligibilityOracleEarningPowerCalculator} from
"src/calculators/BinaryEligibilityOracleEarningPowerCalculator.sol";
import {MintRewardNotifier} from "../notifiers/MintRewardNotifier.sol";
import {StakerTestBase} from "./StakerTestBase.sol";
import {Staker} from "../Staker.sol";

/// @title BinaryEligibilityOracleEarningPowerCalculatorTestBase
/// @author [ScopeLift](https://scopelift.co)
/// @notice The base contract for testing `BinaryEligibilityOracleEarningPowerCalculator`. Contains
/// test setup and helper functions for testing the calculator's behavior. This conrtract is meant
/// to be inherited in a concrete implementation with the necessary setUp and virtual methods
/// implemented.
abstract contract BinaryEligibilityOracleEarningPowerCalculatorTestBase is StakerTestBase {
/// @notice The earning power calculator to be tested.
BinaryEligibilityOracleEarningPowerCalculator calculator;

/// @notice A helper function that updates the delegatee score for a given deposit to a random
/// value between 0 and twice the eligibility threshold, facilitating tests for both eligible and
/// ineligible delegatee scenarios.
/// @param _depositId The identifier of the deposit whose delegatee's score is to be updated.
function _updateEarningPower(Staker.DepositIdentifier _depositId) internal virtual override {
uint256 _delegateeeScore = vm.randomUint(0, calculator.delegateeEligibilityThresholdScore() * 2);
Staker.Deposit memory _deposit = _fetchDeposit(_depositId);
vm.startPrank(calculator.scoreOracle());
calculator.updateDelegateeScore(_deposit.delegatee, _delegateeeScore);
vm.stopPrank();
}

/// @notice Bound the mint amount to a realistic value.
/// @dev Override of the base contract's function to set appropriate bounds for this calculator.
/// @param _amount The unbounded mint amount.
/// @return The bounded mint amount.
function _boundMintAmount(uint256 _amount) internal pure virtual override returns (uint256) {
return bound(_amount, 1, 100_000_000e18);
}
}
30 changes: 30 additions & 0 deletions src/test/MintRewardNotifierTestBase.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
// SPDX-License-Identifier: AGPL-3.0-only

pragma solidity ^0.8.23;

import {MintRewardNotifier} from "../notifiers/MintRewardNotifier.sol";
import {StakerTestBase} from "./StakerTestBase.sol";

/// @title MintRewardNotifierTestBase
/// @author [ScopeLift](https://scopelift.co)
/// @notice Base contract for testing a staker contract with a single `MintRewardNotifier`. Extends
/// `StakerTestBase` and implements the reward notification logic.
/// @dev This contract requires an initialized instance of `MintRewardNotifier`. Initialization is
/// typically handled by a deployment script such as
/// `src/script/notifiers/DeployMintRewardNotifier.sol`
abstract contract MintRewardNotifierTestBase is StakerTestBase {
/// @notice The mint reward notifier to test.
MintRewardNotifier mintRewardNotifier;

/// @notice Sets the reward amount, then calls the `notify` function that triggers token minting
/// and reward distribution.
/// @param _amount The amount of reward to notify the staker contract.
function _notifyRewardAmount(uint256 _amount) public override {
Comment thread
alexkeating marked this conversation as resolved.
address _owner = mintRewardNotifier.owner();

vm.prank(_owner);
mintRewardNotifier.setRewardAmount(_amount);

mintRewardNotifier.notify();
}
}
Loading
Loading