fix hooks

OriginProtocol · Jan 9, 2025 · 16a532b · 16a532b
1 parent 5ade3a2
commit 16a532b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/contracts/test/_global-hooks.js b/contracts/test/_global-hooks.js
@@ -6,6 +6,8 @@ const {
   isHoleskyFork,
   isBaseFork,
   isBaseUnitTest,
+  isSonicFork,
+  isSonicUnitTest,
 } = require("./helpers");
 
 const _chunkId = Number(process.env.CHUNK_ID);
@@ -46,6 +48,8 @@ mocha.before(function () {
     const isArbTestFile = s.file.endsWith(".arb.fork-test.js");
     const isBaseTestFile = s.file.endsWith(".base.fork-test.js");
     const isBaseUnitTestFile = s.file.endsWith(".base.js");
+    const isSonicTestFile = s.file.endsWith(".sonic.fork-test.js");
+    const isSonicUnitTestFile = s.file.endsWith(".sonic.js");
     const unitTest = !s.file.endsWith(".fork-test.js") && !isBaseUnitTestFile;
 
     if (isArbFork) {
@@ -54,10 +58,14 @@ mocha.before(function () {
       return isMainnetForkTestFile;
     } else if (isBaseFork) {
       return isBaseTestFile;
+    } else if (isSonicFork) {
+      return isSonicTestFile;
     } else if (isHoleskyFork) {
       return isHoleskyTestFile;
     } else if (isBaseUnitTest) {
       return isBaseUnitTestFile;
+    } else if (isSonicUnitTest) {
+      return isSonicUnitTestFile;
     } else {
       // else is unit test
       return unitTest;