feat(snapshot): provide config to resolveSnapshotPath

packages/browser/src/node/rpc.ts

@@ -90,7 +90,9 @@ export function setupBrowserRpc(server: BrowserServer) {
           return ctx.report('onUserConsoleLog', log)
         },
         resolveSnapshotPath(testPath) {
-          return ctx.snapshot.resolvePath(testPath)
+          return ctx.snapshot.resolvePath(testPath, {
+            config: project.getSerializableConfig(),
+          })
         },
         resolveSnapshotRawPath(testPath, rawPath) {
           return ctx.snapshot.resolveRawPath(testPath, rawPath)

packages/snapshot/src/manager.ts

@@ -23,7 +23,7 @@ export class SnapshotManager {
     addSnapshotResult(this.summary, result)
   }
 
-  resolvePath(testPath: string): string {
+  resolvePath(testPath: string, context?: any): string {
     const resolver
       = this.options.resolveSnapshotPath || (() => {
         return join(
@@ -32,7 +32,7 @@ export class SnapshotManager {
         )
       })
 
-    const path = resolver(testPath, this.extension)
+    const path = resolver(testPath, this.extension, context)
     return path
   }
 

packages/snapshot/src/types/index.ts

@@ -20,7 +20,7 @@ export interface SnapshotStateOptions {
   snapshotEnvironment: SnapshotEnvironment
   expand?: boolean
   snapshotFormat?: PrettyFormatOptions
-  resolveSnapshotPath?: (path: string, extension: string) => string
+  resolveSnapshotPath?: (path: string, extension: string, context?: any) => string
 }
 
 export interface SnapshotMatchOptions {

packages/vitest/src/node/pools/rpc.ts

@@ -20,7 +20,9 @@ export function createMethodsRPC(project: WorkspaceProject, options: MethodsOpti
       ctx.snapshot.add(snapshot)
     },
     resolveSnapshotPath(testPath: string) {
-      return ctx.snapshot.resolvePath(testPath)
+      return ctx.snapshot.resolvePath(testPath, {
+        config: project.getSerializableConfig(),
+      })
     },
     async getSourceMap(id, force) {
       if (force) {

packages/vitest/src/node/types/config.ts

@@ -5,6 +5,7 @@ import type { SnapshotStateOptions } from '@vitest/snapshot'
 import type { AliasOptions, ConfigEnv, DepOptimizationConfig, ServerOptions, UserConfig as ViteUserConfig } from 'vite'
 import type { ViteNodeServerOptions } from 'vite-node'
 import type { ChaiConfig } from '../../integrations/chai/config'
+import type { SerializedConfig } from '../../runtime/config'
 import type { EnvironmentOptions } from '../../types/environment'
 import type { Arrayable, ErrorWithDiff, ParsedStack, ProvidedContext } from '../../types/general'
 import type { HappyDOMOptions } from '../../types/happy-dom-options'
@@ -573,7 +574,11 @@ export interface InlineConfig {
   /**
    * Resolve custom snapshot path
    */
-  resolveSnapshotPath?: (path: string, extension: string) => string
+  resolveSnapshotPath?: (
+    path: string,
+    extension: string,
+    context: { config: SerializedConfig }
+  ) => string
 
   /**
    * Path to a custom snapshot environment module that has a default export of `SnapshotEnvironment` object.

test/config/fixtures/snapshot-path-context/__snapshots__/project1/basic.test.ts.snap

@@ -0,0 +1,3 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`basic 1`] = `"hello"`;

test/config/fixtures/snapshot-path-context/__snapshots__/project2/basic.test.ts.snap

@@ -0,0 +1,3 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`basic 1`] = `"hello"`;

test/config/fixtures/snapshot-path-context/basic.test.ts

@@ -0,0 +1,5 @@
+import { test, expect } from 'vitest'
+
+test('basic', () => {
+  expect('hello').toMatchSnapshot()
+})

test/config/fixtures/snapshot-path-context/vitest.config.ts

@@ -0,0 +1,15 @@
+import { join, dirname, basename } from 'node:path';
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    resolveSnapshotPath(path, extension, context) {
+      return join(
+        dirname(path),
+        '__snapshots__',
+        context.config.name ?? 'na',
+        basename(path) + extension
+      );
+    },
+  },
+});

test/config/fixtures/snapshot-path-context/vitest.workspace.ts

@@ -0,0 +1,18 @@
+import { defineWorkspace } from 'vitest/config'
+
+export default defineWorkspace([
+  {
+    extends: './vitest.config.ts',
+    test: {
+      name: 'project1',
+      root: import.meta.dirname,
+    }
+  },
+  {
+    extends: './vitest.config.ts',
+    test: {
+      name: 'project2',
+      root: import.meta.dirname,
+    }
+  }
+])

test/config/test/snapshot.test.ts

@@ -0,0 +1,19 @@
+import { expect, test } from 'vitest'
+import { runVitest } from '../../test-utils'
+
+test('resolveSnapshotPath context', async () => {
+  const { stderr, ctx } = await runVitest({
+    root: './fixtures/snapshot-path-context',
+  })
+  expect(stderr).toBe('')
+  expect(
+    Object.fromEntries(
+      ctx!.state.getFiles().map(f => [`${f.projectName}|${f.name}`, f.result?.state]),
+    ),
+  ).toMatchInlineSnapshot(`
+    {
+      "project1|basic.test.ts": "pass",
+      "project2|basic.test.ts": "pass",
+    }
+  `)
+})