fix(coverage): await profiler calls (#7763)

AriPerkkio · web-flow · commit 795a64336330 · 2025-04-04T12:10:06.000+02:00
diff --git a/packages/coverage-v8/src/index.ts b/packages/coverage-v8/src/index.ts
@@ -10,19 +10,21 @@ const session = new inspector.Session()
 let enabled = false
 
 const mod: CoverageProviderModule = {
-  startCoverage({ isolate }) {
+  async startCoverage({ isolate }) {
     if (isolate === false && enabled) {
       return
     }
 
     enabled = true
 
     session.connect()
-    session.post('Profiler.enable')
-    session.post('Profiler.startPreciseCoverage', {
-      callCount: true,
-      detailed: true,
-    })
+    await new Promise(resolve => session.post('Profiler.enable', resolve))
+    await new Promise(resolve =>
+      session.post(
+        'Profiler.startPreciseCoverage',
+        { callCount: true, detailed: true },
+        resolve,
+      ))
   },
 
   takeCoverage(options): Promise<{ result: ScriptCoverageWithOffset[] }> {
@@ -32,14 +34,19 @@ const mod: CoverageProviderModule = {
           return reject(error)
         }
 
-        const result = coverage.result
-          .filter(filterResult)
-          .map(res => ({
-            ...res,
-            startOffset: options?.moduleExecutionInfo?.get(fileURLToPath(res.url))?.startOffset || 0,
-          }))
+        try {
+          const result = coverage.result
+            .filter(filterResult)
+            .map(res => ({
+              ...res,
+              startOffset: options?.moduleExecutionInfo?.get(fileURLToPath(res.url))?.startOffset || 0,
+            }))
 
-        resolve({ result })
+          resolve({ result })
+        }
+        catch (e) {
+          reject(e)
+        }
       })
 
       if (provider === 'stackblitz') {
@@ -48,13 +55,13 @@ const mod: CoverageProviderModule = {
     })
   },
 
-  stopCoverage({ isolate }) {
+  async stopCoverage({ isolate }) {
     if (isolate === false) {
       return
     }
 
-    session.post('Profiler.stopPreciseCoverage')
-    session.post('Profiler.disable')
+    await new Promise(resolve => session.post('Profiler.stopPreciseCoverage', resolve))
+    await new Promise(resolve => session.post('Profiler.disable', resolve))
     session.disconnect()
   },
 
