tests: update tests to add missing functionnalities

AngleProtocol · Oct 22, 2024 · bcda179 · bcda179
1 parent 3bc5b0c
commit bcda179
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/test/fuzz/MultiBlockHarvester.t.sol b/test/fuzz/MultiBlockHarvester.t.sol
@@ -211,13 +211,21 @@ contract MultiBlockHarvestertTest is Fixture, FunctionUtils {
         vm.expectRevert(Errors.NotGovernorOrGuardian.selector);
         harvester.setMaxSlippage(1e9);
 
-        vm.expectRevert(Errors.NotGovernorOrGuardian.selector);
-        harvester.updateLimitExposuresYieldAsset(address(XEVT));
-
         vm.expectRevert(Errors.NotGovernorOrGuardian.selector);
         harvester.toggleTrusted(alice);
     }
 
+    function test_OnlyTrusted_RevertWhen_NotTrusted() public {
+        vm.expectRevert(Errors.NotTrustedOrGuardian.selector);
+        harvester.setTargetExposure(address(EURC), targetExposure);
+
+        vm.expectRevert(Errors.NotTrusted.selector);
+        harvester.harvest(XEVT, 1e9, new bytes(0));
+
+        vm.expectRevert(Errors.NotTrusted.selector);
+        harvester.finalizeRebalance(EURC, 1e6);
+    }
+
     function test_SettersHarvester() public {
         vm.startPrank(governor);
         vm.expectRevert(Errors.InvalidParam.selector);
@@ -314,6 +322,13 @@ contract MultiBlockHarvestertTest is Fixture, FunctionUtils {
         vm.stopPrank();
     }
 
+    function test_SetTargetExposure() public {
+        vm.prank(governor);
+        harvester.setTargetExposure(address(EURC), targetExposure + 1);
+        (, uint64 currentTargetExposure, , , ) = harvester.yieldBearingData(address(EURC));
+        assertEq(currentTargetExposure, targetExposure + 1);
+    }
+
     function test_harvest_TooBigMintedAmount() external {
         _loadReserve(EURC, 1e26);
         _loadReserve(XEVT, 1e6);