++ test

celo-org · Sep 24, 2024 · 2379af0 · 2379af0
1 parent b4ddea7
commit 2379af0
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/packages/protocol/test-sol/devchain/e2e/common/EpochManager.t.sol b/packages/protocol/test-sol/devchain/e2e/common/EpochManager.t.sol
@@ -109,6 +109,32 @@ contract E2E_EpochManager_InitializeSystem is E2E_EpochManager {
     assertEq(epochManager.firstKnownEpoch(), 42);
     assertEq(epochManager.getCurrentEpochNumber(), 42);
 
+    assertTrue(epochManager.systemAlreadyInitialized());
+  }
+}
+contract E2E_EpochManager_GetCurrentEpoch is E2E_EpochManager {
+  function setUp() public override {
+    super.setUp();
+    whenL2(vm);
+  }
+
+  function test_Revert_WhenSystemNotInitialized() public {
+    vm.expectRevert("Epoch system not initialized");
+    (
+      uint256 firstBlock,
+      uint256 lastBlock,
+      uint256 startTimestamp,
+      uint256 rewardsBlock
+    ) = epochManager.getCurrentEpoch();
+  }
+
+  function test_ReturnExpectedValues() public {
+    vm.prank(epochManagerEnabler);
+    epochManager.initializeSystem(42, 43, firstElected);
+
+    assertEq(epochManager.firstKnownEpoch(), 42);
+    assertEq(epochManager.getCurrentEpochNumber(), 42);
+
     (
       uint256 firstBlock,
       uint256 lastBlock,