fix: resolve file field for suites based on tests when using mocha exports ui

Author: shadowusr

src/test-reader/mocha-reader/tree-builder-decorator.js

@@ -1,5 +1,6 @@
 const { Suite, Test, Hook } = require("../test-object");
 const crypto = require("../../utils/crypto");
+const { computeFile } = require("./utils");
 
 class TreeBuilderDecorator {
     #treeBuilder;
@@ -17,7 +18,9 @@ class TreeBuilderDecorator {
     }
 
     addSuite(mochaSuite) {
-        const { id: mochaId, file } = mochaSuite;
+        const { id: mochaId } = mochaSuite;
+        const file = computeFile(mochaSuite) ?? "unknown-file";
+
         const positionInFile = this.#suiteCounter.get(file) || 0;
         const id = mochaSuite.root ? mochaId : crypto.getShortMD5(file) + positionInFile;
         const suite = this.#mkTestObject(Suite, mochaSuite, { id });

src/test-reader/mocha-reader/utils.js

@@ -0,0 +1,28 @@
+// When using "exports" mocha interface, "file" field is absent on suites, and available on tests only.
+// This helper tries to resolve "file" field for suites, drilling down to child tests and using their file field.
+const computeFile = mochaSuite => {
+    if (mochaSuite.file) {
+        return mochaSuite.file;
+    }
+
+    if (mochaSuite.tests.length > 0 && mochaSuite.tests[0].file) {
+        mochaSuite.file = mochaSuite.tests[0].file;
+
+        return mochaSuite.file;
+    }
+
+    for (const childSuite of mochaSuite.suites) {
+        const computedFile = computeFile(childSuite);
+        if (computedFile) {
+            mochaSuite.file = computedFile;
+
+            return mochaSuite.file;
+        }
+    }
+
+    return null;
+};
+
+module.exports = {
+    computeFile,
+};