fix: ignore __vi_import__ and support Vitest.test

Closes #638

Author: sheremet-va

src/worker/collect.ts

@@ -99,8 +99,13 @@ export function astParseFile(filepath: string, code: string) {
       ) {
         return callee.object?.name
       }
-      // direct call as `__vite_ssr_exports_0__.test()`
-      if (callee.object?.name?.startsWith('__vite_ssr_')) {
+      if (
+        // direct call as `__vite_ssr_exports_0__.test()`
+        callee.object?.name?.startsWith('__vite_ssr_')
+        // call as `__vite_ssr_exports_0__.Vitest.test`,
+        // this is a special case for using Vitest namespaces popular in Effect
+        || (callee.object?.object?.name?.startsWith('__vite_ssr_') && callee.object?.property?.name === 'Vitest')
+      ) {
         return getName(callee.property)
       }
       // call as `__vite_ssr__.test.skip()`
@@ -167,7 +172,11 @@ export function astParseFile(filepath: string, code: string) {
         message = message.slice(2)
       }
 
-      message = message.replace(/__vite_ssr_import_\d+__\./g, '')
+      message = message
+        // Vite SSR injects these
+        .replace(/__vite_ssr_import_\d+__\./g, '')
+        // Vitest module mocker injects these
+        .replace(/__vi_import_\d+__/g, '')
 
       // cannot statically analyze, so we always skip it
       if (mode === 'skipIf' || mode === 'runIf') {

test/e2e/discovery.test.ts

@@ -4,10 +4,13 @@ import { describe, expect, it, onTestFinished } from 'vitest'
 import { createVitest } from 'vitest/node'
 import { astCollectTests } from '../../src/worker/collect'
 
+const variableFixture = 'test-from-vitest-variable.ts'
+
 describe('can discover tests', () => {
   it.for([
     'todo-import-suite.ts',
     'todo-globals-suite.ts',
+    variableFixture,
   ])('can discover todo tests inside a suite in %s', async (fixture) => {
     const vitest = await createVitest('test', { config: false })
     onTestFinished(() => vitest.close())
@@ -34,7 +37,7 @@ describe('can discover tests', () => {
     expect(testTask.mode).toBe('run')
     expect(testTask.location).toMatchObject({
       line: 4,
-      column: 31, // TODO: should it be 5 instead?
+      column: variableFixture === fixture ? 38 : 31, // TODO: should it be 5 instead? since we only care about "line", ignore for now
     })
 
     expect(suiteTask.name).toBe('Drafts')

test/e2e/fixtures/collect/test-from-vitest-variable.ts

@@ -0,0 +1,14 @@
+import { Vitest } from './vitest'
+
+Vitest.describe('TicketDetailBottomBar', () => {
+  Vitest.it.each(['submit', 'discard'])(
+    'emits %s event when button is clicked',
+    async (eventName) => {
+    },
+  )
+
+  Vitest.describe('Drafts', () => {
+    Vitest.it.todo('should not display draft information if ticket has no draft')
+    Vitest.it.todo('should display draft information if ticket has a draft')
+  })
+})

test/e2e/fixtures/collect/vitest.ts

@@ -0,0 +1 @@
+export * as Vitest from 'vitest'