fix: add sourcemap support for setup files (#50)

index.d.ts

@@ -616,6 +616,7 @@ declare module 'jest-allure2-reporter' {
   /** @inheritDoc */
   export interface AllureTestRunMetadata extends AllureTestCaseMetadata {
     config: unknown;
+    loadedFiles: string[];
     sourceLocation?: never;
     transformedCode?: never;
   }

src/environment/listener.ts

@@ -1,3 +1,4 @@
+import _ from 'lodash';
 import type { AllureTestItemSourceLocation } from 'jest-allure2-reporter';
 import type { Circus } from '@jest/types';
 import type {
@@ -6,15 +7,17 @@ import type {
   TestEnvironmentSetupEvent,
 } from 'jest-environment-emit';
 import * as StackTrace from 'stacktrace-js';
+import type { JestEnvironmentConfig } from '@jest/environment';
 
 import * as api from '../api';
 import realm from '../realms';
 import { autoIndent, getStatusDetails, isJestAssertionError, isLibraryPath } from '../utils';
 
 const listener: EnvironmentListenerFn = (context) => {
   context.testEvents
-    .on('test_environment_setup', injectGlobals)
-    .on('test_environment_setup', setWorkerId)
+    .once('test_environment_setup', injectGlobals)
+    .once('test_environment_setup', setWorkerId)
+    .once('test_environment_setup', reportSetupFiles(context.config))
     .on('add_hook', addHookType)
     .on('add_hook', addSourceLocation)
     .on('add_hook', addSourceCode)
@@ -86,6 +89,16 @@ function setWorkerId() {
   }
 }
 
+function reportSetupFiles(config: JestEnvironmentConfig) {
+  return () => {
+    const { setupFilesAfterEnv = [], setupFiles = [] } = config.projectConfig ?? {};
+    const globalMetadata = realm.runtimeContext.getGlobalMetadata();
+    const loadedFiles = globalMetadata.get('loadedFiles', []);
+    const files = _.difference([...setupFiles, ...setupFilesAfterEnv], loadedFiles);
+    globalMetadata.push('loadedFiles', files);
+  };
+}
+
 function addHookType({ event }: TestEnvironmentCircusEvent<Circus.Event & { name: 'add_hook' }>) {
   const metadata = realm.runtimeContext.getCurrentMetadata();
   metadata.set('hookType', event.hookType);

src/reporter/JestAllure2Reporter.ts

@@ -155,15 +155,16 @@ export class JestAllure2Reporter extends JestMetadataReporter {
   async onTestFileStart(test: Test) {
     super.onTestFileStart(test);
 
-    await FileNavigatorCache.instance.scanSourcemap(test.path);
     const execute = this.#onTestFileStart.bind(this, test);
     const attempt = this.#attemptSync.bind(this, 'onTestFileStart()', execute);
     const testPath = path.relative(this._globalConfig.rootDir, test.path);
     log.trace.begin(__TID(test), testPath);
-    log.trace.complete(__TID(test), 'onTestFileStart', attempt);
+    await log.trace.complete(__TID(test), 'onTestFileStart', attempt);
   }
 
-  #onTestFileStart(test: Test) {
+  async #onTestFileStart(test: Test) {
+    await FileNavigatorCache.instance.scanSourcemap(test.path);
+
     const rawMetadata = JestAllure2Reporter.query.test(test);
     const testFileMetadata = new AllureMetadataProxy<AllureTestFileMetadata>(rawMetadata);
 
@@ -199,6 +200,11 @@ export class JestAllure2Reporter extends JestMetadataReporter {
     const rawMetadata = JestAllure2Reporter.query.test(test);
     const testFileMetadata = new AllureMetadataProxy<AllureTestFileMetadata>(rawMetadata);
     const globalMetadataProxy = this._globalMetadataProxy;
+
+    for (const loadedFile of globalMetadataProxy.get('loadedFiles', [])) {
+      await FileNavigatorCache.instance.scanSourcemap(loadedFile);
+    }
+
     const allureWriter = this._writer;
 
     fallbacks.onTestFileResult(test, testFileMetadata);

src/utils/FileNavigatorCache.ts

@@ -8,6 +8,7 @@ import { FileNavigator } from './index';
 
 export class FileNavigatorCache {
   #cache = new Map<string, Promise<FileNavigator | undefined>>();
+  #scannedSourcemaps = new Set<string>();
 
   resolve(filePath: string): Promise<FileNavigator | undefined> {
     const absolutePath = path.isAbsolute(filePath) ? filePath : path.resolve(filePath);
@@ -21,6 +22,9 @@ export class FileNavigatorCache {
   async scanSourcemap(filePath: string): Promise<void> {
     const sourceMapPath = `${filePath}.map`;
 
+    if (this.#scannedSourcemaps.has(sourceMapPath)) return;
+    this.#scannedSourcemaps.add(sourceMapPath);
+
     const doesNotExist = await fs.access(sourceMapPath).catch(() => true);
     if (doesNotExist) return;