feat: allow calling init from test plugins

sandworm-hq · Sep 28, 2022 · e4b196b · e4b196b
1 parent e4057e7
commit e4b196b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/index.js b/src/index.js
@@ -58,11 +58,19 @@ const init = ({
       return Promise.resolve();
     }
 
-    const {name: callerModule} = getCurrentModuleInfo({
+    const {name: initCaller} = getCurrentModuleInfo({
       allowURLs: false,
     });
-    if (!['root', 'mocha>sandworm-mocha'].includes(callerModule)) {
-      logger.warn(`only root may call init (called from ${callerModule})`);
+    const initCallerModules = initCaller.split('>');
+    const rootCallerModule = initCallerModules[0];
+    const directCallerModule = initCallerModules.slice(-1)[0];
+
+    if (
+      initCaller !== 'root' &&
+      (rootCallerModule !== 'mocha' || directCallerModule !== 'sandworm-mocha') &&
+      (rootCallerModule !== 'jest-runner' || directCallerModule !== 'sandworm-jest')
+    ) {
+      logger.warn(`only root may call init (called from ${initCaller})`);
       return Promise.resolve();
     }
     if (typeof devModeOption !== 'boolean') {