fix(coverage): exclude injected functions without mapping

packages/coverage-v8/src/provider.ts

@@ -23,6 +23,7 @@ import v8ToIstanbul from 'v8-to-istanbul'
 import { cleanUrl } from 'vite-node/utils'
 
 import { BaseCoverageProvider } from 'vitest/coverage'
+import { offsetToPosition, originalPositionFor, TraceMap } from 'vitest/utils'
 import { version } from '../package.json' with { type: 'json' }
 
 type TransformResults = Map<string, FetchResult>
@@ -349,6 +350,29 @@ export class V8CoverageProvider extends BaseCoverageProvider<ResolvedCoverageOpt
           // If file was executed by vite-node we'll need to add its wrapper
           const wrapperLength = sources.isExecuted ? WRAPPER_LENGTH : 0
 
+          // filter out functions without mappings,
+          // for example, "export getter" injected by Vite ssr transform.
+          // https://github.com/vitest-dev/vitest/issues/7130
+          if (sources.isExecuted && sources.sourceMap) {
+            const traceMap = new TraceMap(sources.sourceMap.sourcemap)
+            functions = functions.filter((f) => {
+              if (f.ranges.length === 1) {
+                const start = f.ranges[0].startOffset - wrapperLength
+                const end = f.ranges[0].endOffset - wrapperLength - 1
+                if ([start, end].every(offset => offset >= 0 && offset < sources.source.length)) {
+                  const startPos = offsetToPosition(sources.source, start)
+                  const endPos = offsetToPosition(sources.source, end)
+                  const startSourcePos = originalPositionFor(traceMap, startPos)
+                  const endSourcePos = originalPositionFor(traceMap, endPos)
+                  if (startSourcePos.line === null && endSourcePos.line === null) {
+                    return false
+                  }
+                }
+              }
+              return true
+            })
+          }
+
           const converter = v8ToIstanbul(
             url,
             wrapperLength,

packages/utils/src/index.ts

@@ -33,6 +33,7 @@ export { nanoid } from './nanoid'
 export {
   lineSplitRE,
   offsetToLineNumber,
+  offsetToPosition,
   positionToOffset,
 } from './offset'
 export { shuffle } from './random'

packages/utils/src/offset.ts

@@ -21,6 +21,13 @@ export function positionToOffset(
 }
 
 export function offsetToLineNumber(source: string, offset: number): number {
+  return offsetToPosition(source, offset).line
+}
+
+export function offsetToPosition(
+  source: string,
+  offset: number,
+): { line: number; column: number } {
   if (offset > source.length) {
     throw new Error(
       `offset is longer than source length! offset ${offset} > length ${source.length}`,
@@ -30,13 +37,15 @@ export function offsetToLineNumber(source: string, offset: number): number {
   const nl = /\r\n/.test(source) ? 2 : 1
   let counted = 0
   let line = 0
+  let column = 0
   for (; line < lines.length; line++) {
     const lineLength = lines[line].length + nl
     if (counted + lineLength >= offset) {
+      column = offset - counted
       break
     }
 
     counted += lineLength
   }
-  return line + 1
+  return { line: line + 1, column }
 }

packages/vitest/src/utils/source-map.ts

@@ -1,6 +1,7 @@
 export {
   lineSplitRE,
   offsetToLineNumber,
+  offsetToPosition,
   positionToOffset,
 } from '@vitest/utils'
 export {

test/coverage-test/fixtures/configs/vitest.config.injected-functions.ts

@@ -0,0 +1,21 @@
+import MagicString from 'magic-string'
+import { defineConfig } from 'vitest/config'
+
+export default defineConfig({
+  plugins: [
+    {
+      name: 'repro',
+      transform(code, id, _options) {
+        if (id.endsWith('injected-functions.ts')) {
+          const output = new MagicString(code)
+          output.prepend(`;function prepended(){};`)
+          output.append(`;function appended(){};`)
+          return {
+            code: output.toString(),
+            map: output.generateMap({ hires: 'boundary' }),
+          }
+        }
+      },
+    },
+  ],
+})

test/coverage-test/fixtures/src/injected-functions.ts

@@ -0,0 +1,4 @@
+function original() {
+  console.log("hello")
+}
+original()

test/coverage-test/fixtures/test/injected-functions.test.ts

@@ -0,0 +1,4 @@
+import { test } from 'vitest'
+import "../src/injected-functions";
+
+test('basic', () => {})

test/coverage-test/test/injected-functions.test.ts

@@ -0,0 +1,81 @@
+import { expect } from 'vitest'
+import { isV8Provider, readCoverageMap, runVitest, test } from '../utils'
+
+test('filter out functions injected by plugin', async () => {
+  const { stderr } = await runVitest({
+    include: ['fixtures/test/injected-functions.test.ts'],
+    coverage: {
+      reporter: ['json', 'html'],
+      include: ['fixtures/src/injected-functions.ts'],
+    },
+    config: 'fixtures/configs/vitest.config.injected-functions.ts',
+  })
+  expect(stderr).toBe('')
+
+  const coverageMap = await readCoverageMap()
+  const fileCoverage = coverageMap.fileCoverageFor('<process-cwd>/fixtures/src/injected-functions.ts')
+  if (isV8Provider()) {
+    expect(fileCoverage.fnMap).toMatchInlineSnapshot(`
+      {
+        "0": {
+          "decl": {
+            "end": {
+              "column": 1,
+              "line": 3,
+            },
+            "start": {
+              "column": 0,
+              "line": 1,
+            },
+          },
+          "line": 1,
+          "loc": {
+            "end": {
+              "column": 1,
+              "line": 3,
+            },
+            "start": {
+              "column": 0,
+              "line": 1,
+            },
+          },
+          "name": "original",
+        },
+      }
+    `)
+  }
+  else {
+    expect(fileCoverage.fnMap).toMatchInlineSnapshot(`
+      {
+        "0": {
+          "decl": {
+            "end": {
+              "column": 20,
+              "line": 1,
+            },
+            "start": {
+              "column": 9,
+              "line": 1,
+            },
+          },
+          "loc": {
+            "end": {
+              "column": null,
+              "line": 3,
+            },
+            "start": {
+              "column": 20,
+              "line": 1,
+            },
+          },
+          "name": "original",
+        },
+      }
+    `)
+  }
+  expect(fileCoverage.f).toMatchInlineSnapshot(`
+    {
+      "0": 1,
+    }
+  `)
+})