chore: update global hook

OriginProtocol · May 17, 2024 · 5708edf · 5708edf
1 parent 2cfb600
commit 5708edf
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 7 deletions.
diff --git a/contracts/test/_global-hooks.js b/contracts/test/_global-hooks.js
@@ -1,6 +1,6 @@
 const mocha = require("mocha");
 
-const { isForkTest, isArbFork, isBaseFork } = require("./helpers");
+const { isMainnetForkTest, isArbFork, isHoleskyFork,isBaseFork } = require("./helpers");
 
 const _chunkId = Number(process.env.CHUNK_ID);
 const _maxChunks = Number(process.env.MAX_CHUNKS);
@@ -34,12 +34,25 @@ mocha.before(function () {
 
   // If you are running unit tests, scrape out all fork tests.
   // For fork tests, scrape out all unit tests.
-  root.suites = root.suites.filter(
-    (s) =>
-      s.file.endsWith(".fork-test.js") == isForkTest &&
-      s.file.endsWith(".arb.fork-test.js") == isArbFork &&
-      s.file.endsWith(".base.fork-test.js") == isBaseFork
-  );
+  root.suites = root.suites.filter((s) => {
+    const isMainnetForkTestFile = s.file.endsWith(".fork-test.js");
+    const isHoleskyTestFile = s.file.endsWith(".holesky-fork-test.js");
+    const isArbTestFile = s.file.endsWith(".arb.fork-test.js");
+    const isBaseTestFile = s.file.endsWith(".base.fork-test.js");
+
+    if (isMainnetForkTest) {
+      return isMainnetForkTestFile;
+    } else if (isHoleskyFork) {
+      return isHoleskyTestFile;
+    } else if (isArbFork) {
+      return isArbTestFile;
+    } else if (isBaseFork) {
+      return isBaseTestFile;
+    } else {
+      // else is unit test
+      return !isMainnetForkTestFile && !isHoleskyTestFile && !isArbTestFile && !isBaseTestFile;
+    }
+  });
 
   if (!runTestsParallely) {
     // When running serially

diff --git a/contracts/test/helpers.js b/contracts/test/helpers.js
@@ -260,6 +260,7 @@ const isTest = process.env.IS_TEST === "true";
 const isSmokeTest = process.env.SMOKE_TEST === "true";
 const isMainnetOrFork = isMainnet || isFork;
 const isForkTest = isFork && isTest;
+const isMainnetForkTest = isForkTest && hre.network.config.chainId == 1;
 const isForkWithLocalNode = isFork && process.env.LOCAL_PROVIDER_URL;
 const isArbitrumOne = hre.network.name == "arbitrumOne";
 const isTestnetSimplifiedDeploy = isHolesky;
@@ -779,6 +780,7 @@ module.exports = {
   isLocalhost,
   isMainnetOrFork,
   isForkTest,
+  isMainnetForkTest,
   isForkWithLocalNode,
   isArbitrumOne,
   isHolesky,