test: splitting two time-dependent tests

valory-xyz · Jul 12, 2024 · 660e67c · 660e67c
1 parent a423fb7
commit 660e67c
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/test/Tokenomics.js b/test/Tokenomics.js
@@ -1041,7 +1041,7 @@ describe("Tokenomics", async () => {
             await snapshot.restore();
         });
 
-        it("Get to the epoch before the end of the OLAS year and try to change maxBond or epochLen", async () => {
+        it("Get to two epochs before the end of the OLAS year and try to change epochLen", async () => {
             // Take a snapshot of the current state of the blockchain
             const snapshot = await helpers.takeSnapshot();
 
@@ -1051,7 +1051,7 @@ describe("Tokenomics", async () => {
             const yearChangeTime = timeLaunch + oneYear;
 
             // Get to the time of more than one epoch length before the year change (1.5 epoch length)
-            let timeEpochBeforeYearChange = yearChangeTime - epochLen - epochLen / 2;
+            const timeEpochBeforeYearChange = yearChangeTime - epochLen - epochLen / 2;
             await helpers.time.increaseTo(timeEpochBeforeYearChange);
             await tokenomics.checkpoint();
 
@@ -1062,12 +1062,23 @@ describe("Tokenomics", async () => {
             await helpers.time.increase(epochLen);
             await tokenomics.checkpoint();
             expect(await tokenomics.epochLen()).to.equal(2 * epochLen);
-            // Restore the state of the blockchain back to the time half of the epoch before one epoch left for the current year
-            snapshotInternal.restore();
+
+            // Restore to the state of the snapshot
+            await snapshot.restore();
+        });
+
+        it("Get to the epoch before the end of the OLAS year and try to change epochLen", async () => {
+            // Take a snapshot of the current state of the blockchain
+            const snapshot = await helpers.takeSnapshot();
+
+            // OLAS starting time
+            const timeLaunch = Number(await tokenomics.timeLaunch());
+            // One year time from the launch
+            const yearChangeTime = timeLaunch + oneYear;
 
             // Get to the time of the half epoch length before the year change
             // Meaning that the year does not change yet during the current epoch, but it will during the next one
-            timeEpochBeforeYearChange += epochLen;
+            const timeEpochBeforeYearChange = yearChangeTime - epochLen / 2;
             await helpers.time.increaseTo(timeEpochBeforeYearChange);
             await tokenomics.checkpoint();