Add options to resolveSnapshotPath

docs/config/index.md

@@ -503,7 +503,7 @@ Format options for snapshot testing. These options are passed down to [`pretty-f
 
 ### resolveSnapshotPath
 
-- **Type**: `(testPath: string, snapExtension: string) => string`
+- **Type**: `(testPath: string, snapExtension: string, options: { context: any }) => string`
 - **Default**: stores snapshot files in `__snapshots__` directory
 
 Overrides default snapshot path. For example, to store snapshots next to test files:

packages/vitest/src/integrations/snapshot/client.ts

@@ -40,7 +40,7 @@ export class SnapshotClient {
           filePath,
           new SnapshotState(
             filePath,
-            await rpc().resolveSnapshotPath(filePath),
+            await rpc().resolveSnapshotPath(filePath, undefined, { context: this.test.context }),
             getWorkerState().config.snapshotOptions,
           ),
         )

packages/vitest/src/integrations/snapshot/manager.ts

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

packages/vitest/src/node/pool.ts

@@ -129,8 +129,8 @@ function createChannel(ctx: Vitest) {
       snapshotSaved(snapshot) {
         ctx.snapshot.add(snapshot)
       },
-      resolveSnapshotPath(testPath: string) {
-        return ctx.snapshot.resolvePath(testPath)
+      resolveSnapshotPath(testPath: string, options: { context: any }) {
+        return ctx.snapshot.resolvePath(testPath, options)
       },
       async getSourceMap(id, force) {
         if (force) {

packages/vitest/src/types/config.ts

@@ -312,7 +312,7 @@ export interface InlineConfig {
   /**
    * Resolve custom snapshot path
    */
-  resolveSnapshotPath?: (path: string, extension: string) => string
+  resolveSnapshotPath?: (path: string, extension: string, options: { context: any }) => string
 
   /**
    * Pass with no tests

packages/vitest/src/types/snapshot.ts

@@ -8,7 +8,7 @@ export interface SnapshotStateOptions {
   updateSnapshot: SnapshotUpdateState
   expand?: boolean
   snapshotFormat?: PrettyFormatOptions
-  resolveSnapshotPath?: (path: string, extension: string) => string
+  resolveSnapshotPath?: (path: string, extension: string, options: { context: any }) => string
 }
 
 export interface SnapshotMatchOptions {

packages/vitest/src/types/worker.ts

@@ -30,7 +30,7 @@ export interface WorkerRPC {
   onTaskUpdate: (pack: TaskResultPack[]) => void
 
   snapshotSaved: (snapshot: SnapshotResult) => void
-  resolveSnapshotPath: (testPath: string) => string
+  resolveSnapshotPath: (testPath: string, options: { context: any }) => string
 }
 
 export interface WorkerGlobalState {