test: yield repo and updated heart

src/test/mocks/MockClearinghouse.sol

@@ -0,0 +1,30 @@
+// SPDX-License-Identifier: AGPL-3.0
+pragma solidity ^0.8.0;
+
+import {ERC20} from "solmate/tokens/ERC20.sol";
+import {ERC4626} from "solmate/mixins/ERC4626.sol";
+
+// mock clearinghouse to use with testing the yieldRepo contract
+// most functions / variables are omitted
+contract MockClearinghouse {
+    ERC20 public reserve;
+    ERC4626 public wrappedReserve;
+    uint256 public principalReceivables;
+
+    constructor(address _reserve, address _wrappedReserve) {
+        reserve = ERC20(_reserve);
+        wrappedReserve = ERC4626(_wrappedReserve);
+    }
+
+    function setPrincipalReceivables(uint256 _principalReceivables) external {
+        principalReceivables = _principalReceivables;
+    }
+
+    function withdrawReserve(uint256 _amount) external {
+        reserve.transfer(msg.sender, _amount);
+    }
+
+    function withdrawWrappedReserve(uint256 _amount) external {
+        wrappedReserve.transfer(msg.sender, _amount);
+    }
+}

src/test/mocks/MockOhm.sol

@@ -14,6 +14,10 @@ contract MockOhm is ERC20 {
         _mint(to, value);
     }
 
+    function burn(uint256 value) public virtual {
+        _burn(msg.sender, value);
+    }
+
     function burnFrom(address from, uint256 value) public virtual {
         uint256 currentAllowance = allowance[from][msg.sender];
         require(currentAllowance >= value, "ERC20: burn amount exceeds allowance");

src/test/mocks/MockYieldRepo.sol

@@ -0,0 +1,29 @@
+// SPDX-License-Identifier: AGPL-3.0
+pragma solidity ^0.8.0;
+
+import {IYieldRepo} from "../../policies/interfaces/IYieldRepo.sol";
+
+contract MockYieldRepo is IYieldRepo {
+    uint48 public epoch;
+    bool public isShutdown;
+
+    function endEpoch() external override {
+        // do nothing
+    }
+
+    function shutdown() external {
+        isShutdown = true;
+    }
+
+    function getReserveBalance() external pure override returns (uint256) {
+        return 0;
+    }
+
+    function getNextYield() external pure override returns (uint256) {
+        return 0;
+    }
+
+    function getOhmBalanceAndBacking() external pure override returns (uint256, uint256) {
+        return (0, 0);
+    }
+}

src/test/policies/Heart.t.sol

@@ -13,15 +13,17 @@ import {ROLESv1} from "modules/ROLES/ROLES.v1.sol";
 import {RolesAdmin} from "policies/RolesAdmin.sol";
 import {ZeroDistributor} from "policies/Distributor/ZeroDistributor.sol";
 import {MockStakingZD} from "test/mocks/MockStakingForZD.sol";
+import {MockYieldRepo} from "test/mocks/MockYieldRepo.sol";
 
 import {FullMath} from "libraries/FullMath.sol";
 
 import "src/Kernel.sol";
 
-import {OlympusHeart} from "policies/Heart.sol";
+import {OlympusHeart, IHeart} from "policies/Heart.sol";
 
 import {IOperator} from "policies/interfaces/IOperator.sol";
 import {IDistributor} from "policies/interfaces/IDistributor.sol";
+import {IYieldRepo} from "policies/interfaces/IYieldRepo.sol";
 
 /**
  * @notice Mock Operator to test Heart
@@ -73,6 +75,8 @@ contract HeartTest is Test {
     MockStakingZD internal staking;
     ZeroDistributor internal distributor;
 
+    MockYieldRepo internal yieldRepo;
+
     uint48 internal constant PRICE_FREQUENCY = uint48(8 hours);
 
     // MINTR
@@ -121,11 +125,15 @@ contract HeartTest is Test {
             distributor = new ZeroDistributor(address(staking));
             staking.setDistributor(address(distributor));
 
+            // Deploy mock yieldRepo
+            yieldRepo = new MockYieldRepo();
+
             // Deploy heart
             heart = new OlympusHeart(
                 kernel,
                 IOperator(address(operator)),
                 IDistributor(address(distributor)),
+                IYieldRepo(address(yieldRepo)),
                 uint256(10e9), // max reward = 10 reward tokens
                 uint48(12 * 50) // auction duration = 5 minutes (50 blocks on ETH mainnet)
             );
@@ -151,6 +159,9 @@ contract HeartTest is Test {
             // Heart ROLES
             rolesAdmin.grantRole("heart_admin", policy);
         }
+
+        // Do initial beat
+        heart.beat();
     }
 
     // ======== SETUP DEPENDENCIES ======= //
@@ -169,6 +180,30 @@ contract HeartTest is Test {
         assertEq(fromKeycode(deps[2]), fromKeycode(expectedDeps[2]));
     }
 
+    function testRevert_configureDependencies_invalidFrequency() public {
+        // Deploy mock staking with different frequency
+        staking = new MockStakingZD(7 hours, 0, block.timestamp);
+        distributor = new ZeroDistributor(address(staking));
+        staking.setDistributor(address(distributor));
+
+        // Deploy heart
+        heart = new OlympusHeart(
+            kernel,
+            IOperator(address(operator)),
+            IDistributor(address(distributor)),
+            IYieldRepo(address(yieldRepo)),
+            uint256(10e9), // max reward = 10 reward tokens
+            uint48(12 * 50) // auction duration = 5 minutes (50 blocks on ETH mainnet)
+        );
+
+        vm.startPrank(address(kernel));
+        // Since the staking frequency is different, the call to configureDependencies reverts
+        bytes memory err = abi.encodeWithSelector(IHeart.Heart_InvalidFrequency.selector);
+        vm.expectRevert(err);
+        heart.configureDependencies();
+        vm.stopPrank();
+    }
+
     function test_requestPermissions() public {
         Permissions[] memory expectedPerms = new Permissions[](3);