[pull] main from vitest-dev:main

packages/vitest/src/runtime/moduleRunner/utils.ts

@@ -1,5 +1,8 @@
-// copied from vite/src/shared/utils.ts
+// copied from vite
+// https://github.com/vitejs/vite/blob/4417b4f305623b2850bd6ae6553834c017694672/packages/vite/src/shared/utils.ts
+// https://github.com/vitejs/vite/blob/4417b4f305623b2850bd6ae6553834c017694672/packages/vite/src/node/utils.ts
 const postfixRE = /[?#].*$/
+const trailingSeparatorRE = /[?&]$/
 
 function cleanUrl(url: string): string {
   return url.replace(postfixRE, '')
@@ -16,6 +19,6 @@ export function injectQuery(url: string, queryToInject: string): string {
 
 export function removeQuery(url: string, queryToRemove: string): string {
   return url
-    .replace(new RegExp(`[?&]${queryToRemove}(?=[&#]|$)`), '')
-    .replace(/\?$/, '')
+    .replace(new RegExp(`([?&])${queryToRemove}(?:&|$)`), '$1')
+    .replace(trailingSeparatorRE, '')
 }

test/e2e/deps/dep-simple2/index.js

@@ -0,0 +1 @@
+export default 'test-dep-simple2'

test/e2e/deps/dep-simple2/package.json

@@ -0,0 +1,6 @@
+{
+  "name": "test-dep-simple2",
+  "type": "module",
+  "private": true,
+  "exports": "./index.js"
+}

test/e2e/fixtures/optimize-deps-mock/basic.test.ts

@@ -0,0 +1,40 @@
+import { expect, test, vi } from 'vitest'
+// @ts-expect-error no type
+import * as dep from "@test/test-dep-url"
+// @ts-expect-error no type
+import * as simple from "test-dep-simple"
+// @ts-expect-error no type
+import * as simple2 from "test-dep-simple2"
+
+// mock + optimized
+vi.mock('@test/test-dep-url', async (importOriginal) => {
+  const original = await importOriginal<any>()
+  return { ...original,  mocked: "ok" };
+})
+
+// mock + not optimized + no external
+vi.mock('test-dep-simple', async (importOriginal) => {
+  const original = await importOriginal<any>()
+  return { ...original,  mocked: "ok" };
+})
+
+// mock + not optimized + external
+vi.mock('test-dep-simple2', async (importOriginal) => {
+  const original = await importOriginal<any>()
+  return { ...original,  mocked: "ok" };
+})
+
+test('basic', () => {
+  expect.soft({...dep}).toEqual({
+    mocked: 'ok',
+    importMetaUrl: expect.stringContaining('/node_modules/.vite/vitest/')
+  })
+  expect({...simple}).toEqual({
+    mocked: 'ok',
+    default: 'test-dep-simple',
+  })
+  expect({...simple2}).toEqual({
+    mocked: 'ok',
+    default: 'test-dep-simple2',
+  })
+})

test/e2e/fixtures/optimize-deps-mock/vitest.config.ts

@@ -0,0 +1,19 @@
+import { defineConfig } from 'vitest/config'
+
+export default defineConfig({
+  ssr: {
+    noExternal: ["test-dep-simple"],
+    optimizeDeps: {
+      include: ["@test/test-dep-url"],
+    },
+  },
+  test: {
+    deps: {
+      optimizer: {
+        ssr: {
+          enabled: true,
+        },
+      },
+    },
+  },
+})

test/e2e/package.json

@@ -40,6 +40,7 @@
     "test-dep-conditions": "file:./deps/test-dep-conditions",
     "test-dep-invalid": "link:./deps/dep-invalid",
     "test-dep-simple": "file:./deps/dep-simple",
+    "test-dep-simple2": "file:./deps/dep-simple2",
     "tinyexec": "^1.0.2",
     "tinyspy": "catalog:",
     "typescript": "catalog:",

test/e2e/test/optimize-deps-mock.test.ts

@@ -0,0 +1,21 @@
+import { expect, test } from 'vitest'
+import { runVitest } from '../../test-utils'
+
+test.for(['forks', 'threads', 'vmThreads', 'vmForks'])(
+  'optimize deps and mock - %s',
+  async (pool) => {
+    const { errorTree, stderr } = await runVitest({
+      root: './fixtures/optimize-deps-mock',
+      pool,
+    })
+
+    expect(stderr).toBe('')
+    expect(errorTree()).toMatchInlineSnapshot(`
+      {
+        "basic.test.ts": {
+          "basic": "passed",
+        },
+      }
+    `)
+  },
+)

test/tsconfig.json

@@ -2,7 +2,8 @@
   "extends": "../tsconfig.base.json",
   "compilerOptions": {
     "types": [
-      "node"
+      "node",
+      "vite/client"
     ]
   },
   "include": [

test/unit/test/mocking/import-actual-query-target.ts

@@ -0,0 +1 @@
+export default 'import-actual-query-target'

test/unit/test/mocking/import-actual-query.test.ts

@@ -0,0 +1,20 @@
+import { expect, test, vi } from 'vitest'
+import * as target from './import-actual-query-target?raw'
+
+vi.mock(import('./import-actual-query-target?raw'), async (importOriginal) => {
+  const original = await importOriginal<any>()
+  return {
+    ...original,
+    mocked: 'ok',
+  }
+})
+
+test('importOriginal preserves query parameters', () => {
+  expect({ ...target }).toMatchInlineSnapshot(`
+    {
+      "default": "export default 'import-actual-query-target'
+    ",
+      "mocked": "ok",
+    }
+  `)
+})