test increment year errors

beethovenxfi · Dec 12, 2024 · 9a269a7 · 9a269a7
1 parent 12fc2bd
commit 9a269a7
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/BeetsTest.t.sol b/test/BeetsTest.t.sol
@@ -35,6 +35,7 @@ contract BeetsTest is Test {
         assertEq(beetsToken.amountMintedCurrentYear(), 0);
         assertEq(beetsToken.getMaxAllowedSupplyCurrentYear(), 220_000_000 ether);
         assertEq(beetsToken.balanceOf(TOKEN_MINTER_TARGET), INITIAL_SUPPLY);
+        assertEq(beetsToken.getEndTimestampCurrentYear(), block.timestamp + 365 days);
     }
 
     function testConstructorErrorZeroSupply() public {
@@ -150,4 +151,17 @@ contract BeetsTest is Test {
         assertEq(beetsToken.amountMintedCurrentYear(), 0);
         assertEq(beetsToken.startingSupplyCurrentYear(), INITIAL_SUPPLY);
     }
+
+    function testIncrementYearNotEndedError() public {
+        vm.warp(block.timestamp + 364 days);
+
+        vm.prank(TOKEN_MINTER_ADDRESS);
+        vm.expectRevert(abi.encodeWithSelector(Beets.CurrentYearHasNotEnded.selector));
+        beetsToken.incrementYear();
+    }
+
+    function testIncrementYearUnauthorized() public {
+        vm.expectRevert(abi.encodeWithSelector(OwnableUnauthorizedAccount.selector, address(this)));
+        beetsToken.incrementYear();
+    }
 }