vitest-plugin-vis 2.3.2 → 2.4.0

package/esm/server/vis_context.types.d.ts

@@ -1,5 +1,8 @@
 export type VisState = {
-    projectPath: string;
+    [projectPath: string]: Promise<VisProjectState>;
+};
+export type VisProjectState = {
+    projectRoot: string;
     testTimeout: number;
     hookTimeout: number;
     snapshotRootDir: string;
@@ -27,14 +30,21 @@ export type PartialBrowserCommandContext = {
             testTimeout: number;
             hookTimeout: number;
         };
+        vite: {
+            config: {
+                test?: {
+                    name?: string | undefined;
+                };
+            };
+        };
     };
     provider: {
         name: string;
-        browserName: string;
-        options: {
-            headless: boolean | undefined;
-            screenshotFailures: boolean | undefined;
-            screenshotDirectory: string | undefined;
+        browserName?: string | undefined;
+        options?: {
+            headless?: boolean | undefined;
+            screenshotFailures?: boolean | undefined;
+            screenshotDirectory?: string | undefined;
         };
     };
     testPath: string;

package/esm/server/vis_context.types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vis_context.types.d.ts","sourceRoot":"","sources":["../../src/server/vis_context.types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,QAAQ,GAAG;IACtB,WAAW,EAAE,MAAM,CAAA;IAEnB,WAAW,EAAE,MAAM,CAAA;IACnB,WAAW,EAAE,MAAM,CAAA;IACnB,eAAe,EAAE,MAAM,CAAA;IACvB,mBAAmB,EAAE,MAAM,CAAA;IAC3B,iBAAiB,EAAE,MAAM,CAAA;IACzB,eAAe,EAAE,MAAM,CAAA;IACvB,gBAAgB,EAAE,MAAM,CAAA;IACxB,iBAAiB,EAAE,MAAM,GAAG,SAAS,CAAA;IACrC,MAAM,EAAE,MAAM,CACb,MAAM,EACN;QACC,WAAW,EAAE,MAAM,CAAA;QACnB,SAAS,EAAE,MAAM,CAAA;QACjB,OAAO,EAAE,MAAM,CAAA;QACf,KAAK,EAAE,MAAM,CACZ,MAAM,EACN;YACC,KAAK,EAAE,MAAM,CAAA;SACb,CACD,CAAA;KACD,CACD,CAAA;CACD,CAAA;AACD,MAAM,MAAM,4BAA4B,GAAG;IAC1C,OAAO,EAAE;QACR,MAAM,EAAE;YACP,IAAI,EAAE,MAAM,CAAA;YACZ,eAAe,EAAE;gBAChB,cAAc,EAAE,KAAK,GAAG,KAAK,GAAG,MAAM,CAAA;aACtC,CAAA;YACD,WAAW,EAAE,MAAM,CAAA;YACnB,WAAW,EAAE,MAAM,CAAA;SACnB,CAAA;KACD,CAAA;IACD,QAAQ,EAAE;QACT,IAAI,EAAE,MAAM,CAAA;QACZ,WAAW,EAAE,MAAM,CAAA;QACnB,OAAO,EAAE;YACR,QAAQ,EAAE,OAAO,GAAG,SAAS,CAAA;YAC7B,kBAAkB,EAAE,OAAO,GAAG,SAAS,CAAA;YACvC,mBAAmB,EAAE,MAAM,GAAG,SAAS,CAAA;SACvC,CAAA;KACD,CAAA;IACD,QAAQ,EAAE,MAAM,CAAA;CAChB,CAAA"}
+{"version":3,"file":"vis_context.types.d.ts","sourceRoot":"","sources":["../../src/server/vis_context.types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,QAAQ,GAAG;IACtB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAA;CAC/C,CAAA;AAED,MAAM,MAAM,eAAe,GAAG;IAC7B,WAAW,EAAE,MAAM,CAAA;IACnB,WAAW,EAAE,MAAM,CAAA;IACnB,WAAW,EAAE,MAAM,CAAA;IACnB,eAAe,EAAE,MAAM,CAAA;IACvB,mBAAmB,EAAE,MAAM,CAAA;IAC3B,iBAAiB,EAAE,MAAM,CAAA;IACzB,eAAe,EAAE,MAAM,CAAA;IACvB,gBAAgB,EAAE,MAAM,CAAA;IACxB,iBAAiB,EAAE,MAAM,GAAG,SAAS,CAAA;IACrC,MAAM,EAAE,MAAM,CACb,MAAM,EACN;QACC,WAAW,EAAE,MAAM,CAAA;QACnB,SAAS,EAAE,MAAM,CAAA;QACjB,OAAO,EAAE,MAAM,CAAA;QACf,KAAK,EAAE,MAAM,CACZ,MAAM,EACN;YACC,KAAK,EAAE,MAAM,CAAA;SACb,CACD,CAAA;KACD,CACD,CAAA;CACD,CAAA;AAED,MAAM,MAAM,4BAA4B,GAAG;IAC1C,OAAO,EAAE;QACR,MAAM,EAAE;YACP,IAAI,EAAE,MAAM,CAAA;YACZ,eAAe,EAAE;gBAChB,cAAc,EAAE,KAAK,GAAG,KAAK,GAAG,MAAM,CAAA;aACtC,CAAA;YACD,WAAW,EAAE,MAAM,CAAA;YACnB,WAAW,EAAE,MAAM,CAAA;SACnB,CAAA;QACD,IAAI,EAAE;YACL,MAAM,EAAE;gBACP,IAAI,CAAC,EAAE;oBACN,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;iBACzB,CAAA;aACD,CAAA;SACD,CAAA;KACD,CAAA;IACD,QAAQ,EAAE;QACT,IAAI,EAAE,MAAM,CAAA;QACZ,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;QAChC,OAAO,CAAC,EAAE;YACT,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,CAAA;YAC9B,kBAAkB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAA;YACxC,mBAAmB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;SACxC,CAAA;KACD,CAAA;IACD,QAAQ,EAAE,MAAM,CAAA;CAChB,CAAA"}

package/esm/testing/stubBrowserCommandContext.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stubBrowserCommandContext.d.ts","sourceRoot":"","sources":["../../src/testing/stubBrowserCommandContext.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AACxD,OAAO,KAAK,EAAE,4BAA4B,EAAE,MAAM,gCAAgC,CAAA;AAElF,eAAO,MAAM,yBAAyB,2GAAwC,CAAA;AAE9E,wBAAgB,sCAAsC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,2HAY5G"}
+{"version":3,"file":"stubBrowserCommandContext.d.ts","sourceRoot":"","sources":["../../src/testing/stubBrowserCommandContext.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AACxD,OAAO,KAAK,EAAE,4BAA4B,EAAE,MAAM,gCAAgC,CAAA;AAElF,eAAO,MAAM,yBAAyB,2GAAwC,CAAA;AAE9E,wBAAgB,sCAAsC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,2HAmB5G"}

package/esm/testing/stubBrowserCommandContext.js

@@ -6,6 +6,13 @@ export function createStubPartialBrowserCommandContext({ root, testPath }) {
             config: {
                 root,
             },
+            vite: {
+                config: {
+                    test: {
+                        name: 'subject',
+                    },
+                },
+            },
         },
         provider: {
             options: {},

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vitest-plugin-vis",
-  "version": "2.3.2",
+  "version": "2.4.0",
   "description": "Vitest visual testing plugin",
   "keywords": [
     "image-snapshot",
@@ -107,6 +107,7 @@
     "build": "tsc",
     "build:watch": "tsc -w",
     "clean": "rimraf coverage esm *.tsbuildinfo vitest.*.timestamp*",
+    "clean:vis:local": "rimraf __vis__/local",
     "cov": "cross-env COVERAGE=true vitest run --coverage",
     "coverage": "cross-env COVERAGE=true vitest run --coverage",
     "dev": "run-p build:watch w",

package/src/client/image_snapshot_matcher.ts

@@ -1,8 +1,8 @@
+import type { BrowserCommands } from '@vitest/browser/context'
 import dedent from 'dedent'
 import { resolve } from 'pathe'
 import type { ImageSnapshotNextIndexCommand } from '../commands.ts'
 import type { PrepareImageSnapshotComparisonCommand } from '../server/commands/prepare_image_snapshot_comparison.ts'
-import type { WriteImageSnapshotCommand } from '../server/commands/write_image_snapshot.ts'
 import { isBase64String } from '../shared/base64.ts'
 import { compareImage } from '../shared/compare_image.ts'
 import type { ToMatchImageSnapshotOptions } from '../shared/types.ts'
@@ -15,7 +15,7 @@ import { server } from './vitest_browser_context_proxy.ts'
 import type { CurrentTest } from './vitest_suite_proxy.ts'
 
 export function imageSnapshotMatcher(
-	commands: PrepareImageSnapshotComparisonCommand & WriteImageSnapshotCommand & ImageSnapshotNextIndexCommand,
+	commands: BrowserCommands & PrepareImageSnapshotComparisonCommand & ImageSnapshotNextIndexCommand,
 ) {
 	return async function matchImageSnapshot(
 		test: CurrentTest & {},
@@ -65,11 +65,11 @@ export function imageSnapshotMatcher(
 			return
 		}
 		if (server.config.snapshotOptions.updateSnapshot === 'all' && !options?.expectToFail) {
-			await writeSnapshot(commands, info.baselinePath, resultImage)
+			await writeSnapshot(commands, resolve(info.projectRoot, info.baselinePath), resultImage)
 			return
 		}
 
-		await writeSnapshot(commands, info.diffPath, diffImage)
+		await writeSnapshot(commands, resolve(info.projectRoot, info.diffPath), diffImage)
 
 		throw new Error(
 			dedent`Snapshot \`${taskId}\` mismatched
@@ -103,9 +103,9 @@ export function parseImageSnapshotSubject(subject: any) {
 	)
 }
 
-async function writeSnapshot(commands: WriteImageSnapshotCommand, path: string, image: ImageData) {
+async function writeSnapshot(commands: BrowserCommands, path: string, image: ImageData) {
 	const content = (await toDataURL(image)).split(',')[1]!
-	return commands.writeImageSnapshot(path, content)
+	return commands.writeFile(path, content, { encoding: 'base64' })
 }
 
 async function parseImageSnapshotOptions(

package/src/client/snapshot_options.ts

@@ -51,7 +51,7 @@ function parseArgs<M extends ComparisonMethod>(
 	args: [task: MetaTask, meta: SnapshotMeta<M> | boolean] | [meta: boolean | SnapshotMeta<M>],
 ): [MetaTask | undefined, SnapshotMeta<M>] {
 	return args.length === 1
-		? [ctx.getCurrentTest() ?? (ctx.getCurrentSuite().tasks?.[0] as any)?.file, parseMeta(args[0])]
+		? [ctx.getCurrentTest() ?? (ctx.getCurrentSuite()?.tasks?.[0] as any)?.file, parseMeta(args[0])]
 		: [args[0], parseMeta(args[1])]
 }
 

package/src/commands.ts

@@ -1,6 +1,5 @@
 export * from './server/commands/has_image_snapshot.ts'
 export * from './server/commands/image_snapshot_next_index.ts'
 export * from './server/commands/match_image_snapshot.ts'
-export * from './server/commands/setup_vis_suite.ts'
 export * from './server/commands/prepare_image_snapshot_comparison.ts'
-export * from './server/commands/write_image_snapshot.ts'
+export * from './server/commands/setup_vis_suite.ts'

package/src/config/types.ts

@@ -35,11 +35,11 @@ export type VisOptions<M extends ComparisonMethod = 'pixel'> = ImageSnapshotTime
 			| string
 			| ((context: {
 					ci: boolean
-					browserName: string
+					browserName?: string | undefined
 					providerName: string
 					platform: string
-					screenshotFailures: boolean | undefined
-					screenshotDirectory: string | undefined
+					screenshotFailures?: boolean | undefined
+					screenshotDirectory?: string | undefined
 			  }) => string)
 			| undefined
 		/**

package/src/config/vis.ts

@@ -12,11 +12,13 @@ import type { VisOptions } from './types.ts'
  * which enables the `auto` preset.
  */
 export function vis<M extends ComparisonMethod = 'pixel'>(options: VisOptions<M> = { preset: 'auto' } as any) {
-	visContext.setOptions(options)
-	const preset = options?.preset
 	return {
 		name: NAME,
-		config() {
+		config(userConfig) {
+			const projectName = userConfig?.test?.name
+
+			visContext.setOptions(projectName, options)
+			const preset = options?.preset
 			return {
 				test: {
 					browser: {

package/src/server/commands/browser_command_context.ts

@@ -0,0 +1,9 @@
+import type { BrowserCommandContext } from 'vitest/node'
+
+export function getProjectName(context: { project: { vite: { config: { test?: { name?: string | undefined } } } } }) {
+	return context.project.vite.config.test?.name
+}
+
+export function getBrowserInstanceProjectName(context: BrowserCommandContext) {
+	return context.project.config
+}

package/src/server/commands/extend.ts

@@ -6,7 +6,6 @@ import {
 	prepareImageSnapshotComparison,
 } from './prepare_image_snapshot_comparison.ts'
 import { type SetupVisSuiteCommand, setupVisSuite } from './setup_vis_suite.ts'
-import { type WriteImageSnapshotCommand, writeImageSnapshot } from './write_image_snapshot.ts'
 
 declare module '@vitest/browser/context' {
 	interface BrowserCommands
@@ -14,7 +13,6 @@ declare module '@vitest/browser/context' {
 			ImageSnapshotNextIndexCommand,
 			HasImageSnapshotCommand,
 			PrepareImageSnapshotComparisonCommand,
-			WriteImageSnapshotCommand,
 			MatchImageSnapshotCommand {}
 }
 
@@ -24,5 +22,4 @@ export const commands = {
 	hasImageSnapshot,
 	matchImageSnapshot,
 	prepareImageSnapshotComparison,
-	writeImageSnapshot,
 }

package/src/server/commands/has_image_snapshot.ts

@@ -11,5 +11,5 @@ export const hasImageSnapshot: BrowserCommand<
 > = async (context, taskId, snapshotId, isAutoSnapshot) => {
 	assertTestPathDefined(context, 'hasImageSnapshot')
 
-	return visContext.hasImageSnapshot(context.testPath, taskId, snapshotId, isAutoSnapshot)
+	return visContext.hasImageSnapshot(context as any, taskId, snapshotId, isAutoSnapshot)
 }

package/src/server/commands/image_snapshot_next_index.ts

@@ -12,5 +12,5 @@ export interface ImageSnapshotNextIndexCommand {
 export const imageSnapshotNextIndex: BrowserCommand<[taskId: string]> = async (context, taskId) => {
 	assertTestPathDefined(context, 'imageSnapshotNextIndex')
 
-	return visContext.getTaskCount(context.testPath, taskId)
+	return visContext.getTaskCount(context, taskId)
 }

package/src/server/commands/match_image_snapshot.ts

@@ -43,7 +43,7 @@ export const matchImageSnapshot: BrowserCommand<
 	if (!options) options = {}
 	options.timeout = options.timeout ?? 30000
 
-	const info = visContext.getSnapshotInfo(context.testPath, taskId, isAutoSnapshot, options)
+	const info = await visContext.getSnapshotInfo(context as any, taskId, isAutoSnapshot, options)
 	const baselineBuffer = await file.tryReadFile(info.baselinePath)
 	if (!baselineBuffer) {
 		if (isBase64String(subject)) {

package/src/server/commands/prepare_image_snapshot_comparison.ts

@@ -1,3 +1,4 @@
+import { resolve } from 'pathe'
 import type { BrowserCommand } from 'vitest/node'
 import { isBase64String } from '../../shared/base64.ts'
 import type {
@@ -13,11 +14,29 @@ import { assertTestPathDefined } from './_assertions.ts'
 import type { MatchImageSnapshotOptions } from './match_image_snapshot.ts'
 
 type ImageSnapshotComparisonInfo = {
+	/**
+	 * Path to the project root.
+	 */
 	projectRoot: string
+	/**
+	 * Path to the baseline image relative to the project root.
+	 */
 	baselinePath: string
+	/**
+	 * Path to the result image relative to the project root.
+	 */
 	resultPath: string
+	/**
+	 * Path to the diff image relative to the project root.
+	 */
 	diffPath: string
+	/**
+	 * Base64 encoded baseline image.
+	 */
 	baseline: string
+	/**
+	 * Base64 encoded result image.
+	 */
 	result: string
 } & ImageSnapshotTimeoutOptions &
 	FailureThresholdOptions &
@@ -36,26 +55,26 @@ export const prepareImageSnapshotComparison: BrowserCommand<
 	[taskId: string, snapshotId: string, isAutoSnapshot: boolean, options?: MatchImageSnapshotOptions | undefined]
 > = async (context, taskId, subject, isAutoSnapshot, options) => {
 	assertTestPathDefined(context, 'prepareImageSnapshotComparison')
-
 	// vitest:browser passes in `null` when not defined
 	if (!options) options = {}
 	options.timeout = options.timeout ?? 30000
 
-	const info = visContext.getSnapshotInfo(context.testPath, taskId, isAutoSnapshot, options)
-	const baselineBuffer = await file.tryReadFile(info.baselinePath)
+	const projectRoot = context.project.config.root
+	const info = await visContext.getSnapshotInfo(context as any, taskId, isAutoSnapshot, options)
+	const baselineBuffer = await file.tryReadFile(resolve(projectRoot, info.baselinePath))
 	if (!baselineBuffer) {
 		if (isBase64String(subject)) {
-			await writeSnapshot(info.baselinePath, subject)
+			await writeSnapshot(resolve(projectRoot, info.baselinePath), subject)
 		} else {
-			await takeSnapshotByBrowser(context, info.baselinePath, subject, options)
+			await takeSnapshotByBrowser(context, resolve(projectRoot, info.baselinePath), subject, options)
 		}
 		return
 	}
 
-	const resultBuffer = await takeSnapshot(context, info.resultPath, subject, options)
+	const resultBuffer = await takeSnapshot(context, resolve(projectRoot, info.resultPath), subject, options)
 	return {
 		...info,
-		projectRoot: context.project.config.root,
+		projectRoot,
 		baseline: baselineBuffer.toString('base64'),
 		result: resultBuffer.toString('base64'),
 	}

package/src/server/snapshot_path.ts

@@ -13,8 +13,8 @@ export function resolveSnapshotRootDir(suite: PartialBrowserCommandContext, opti
 		browserName: suite.provider.browserName,
 		providerName: suite.provider.name,
 		platform: process.platform,
-		screenshotFailures: suite.provider.options.screenshotFailures,
-		screenshotDirectory: suite.provider.options.screenshotDirectory,
+		screenshotFailures: suite.provider.options?.screenshotFailures,
+		screenshotDirectory: suite.provider.options?.screenshotDirectory,
 	})
 }
 

package/src/server/vis_context.logic.ts

@@ -1,29 +1,31 @@
-import { join, relative } from 'pathe'
+import { join, relative, resolve } from 'pathe'
 import { pick } from 'type-plus'
 import type { VisOptions } from '../config/types.ts'
 import { BASELINE_DIR, DIFF_DIR, RESULT_DIR } from '../shared/constants.ts'
+import { getProjectName } from './commands/browser_command_context.ts'
 import { file } from './file.ts'
 import { getSnapshotSubpath, resolveSnapshotRootDir } from './snapshot_path.ts'
 import { ctx } from './vis_context.ctx.ts'
-import type { PartialBrowserCommandContext, VisState } from './vis_context.types.ts'
+import type { PartialBrowserCommandContext, VisProjectState, VisState } from './vis_context.types.ts'
 
 export function createVisContext() {
-	let visOptions: VisOptions<any> = {}
-	let state: VisState
-	let globalStateReady: Promise<VisState>
+	let visOptionsRecord: Record<string, VisOptions<any>> = {}
+	let state: VisState = {}
 
 	const context = {
-		setOptions<M extends 'pixel' | 'ssim'>(options: VisOptions<M> = {} as any) {
-			visOptions = options
+		setOptions<M extends 'pixel' | 'ssim'>(projectName: string | undefined, options: VisOptions<M> = {} as any) {
+			visOptionsRecord[projectName ?? '__default'] = options
 		},
-		__test__getOptions() {
-			return visOptions
+		__test__getOptions(projectName: string) {
+			return visOptionsRecord[projectName]
 		},
 		__test__reset() {
-			visOptions = undefined as any
+			visOptionsRecord = {}
+			state = {}
 		},
-		__test__getState() {
-			return state
+		__test__getState(context: PartialBrowserCommandContext) {
+			const projectRoot = getProjectRoot(context)
+			return state[projectRoot]!
 		},
 		/**
 		 * Setup suite is called on each test file's beforeAll hook.
@@ -31,26 +33,33 @@ export function createVisContext() {
 		 * It needs to make sure there is no race condition between the test files.
 		 */
 		async setupSuite(context: PartialBrowserCommandContext) {
-			if (!globalStateReady) {
-				globalStateReady = setupState(context, visOptions)
-				state = await globalStateReady
-			} else {
-				await globalStateReady
+			const projectRoot = getProjectRoot(context)
+
+			if (!state[projectRoot]) {
+				state[projectRoot] = setupState(context, getVisOptions(visOptionsRecord, context) ?? {})
 			}
 
-			const { suiteId, suite } = createSuite(state, context.testPath, visOptions)
-			state.suites[suiteId] = suite
+			const projectState = await state[projectRoot]
+
+			const { suiteId, suite } = createSuite(projectState, context.testPath, visOptionsRecord)
+			projectState.suites[suiteId] = suite
+
 			await Promise.allSettled([ctx.rimraf(suite.diffDir), ctx.rimraf(suite.resultDir)])
-			return pick(state, 'subjectDataTestId')
+			return pick(projectState, 'subjectDataTestId')
 		},
-		getSnapshotInfo(
-			testPath: string,
+		async getSnapshotInfo(
+			browserContext: PartialBrowserCommandContext,
 			name: string,
 			isAutoSnapshot: boolean,
 			options?: { snapshotFileId?: string | undefined },
 		) {
-			const suiteInfo = context.getSuiteInfo(testPath, name)
-			const snapshotFilename = context.getSnapshotFilename(suiteInfo, options?.snapshotFileId, isAutoSnapshot)
+			const suiteInfo = await context.getSuiteInfo(browserContext, name)
+			const snapshotFilename = context.getSnapshotFilename(
+				browserContext,
+				suiteInfo,
+				options?.snapshotFileId,
+				isAutoSnapshot,
+			)
 
 			const { baselineDir, resultDir, diffDir, task } = suiteInfo
 
@@ -60,38 +69,62 @@ export function createVisContext() {
 			const diffPath = join(diffDir, snapshotFilename)
 
 			return {
-				...pick(visOptions, 'comparisonMethod', 'diffOptions', 'failureThreshold', 'failureThresholdType', 'timeout'),
+				...pick(
+					getVisOptions(visOptionsRecord, browserContext),
+					'comparisonMethod',
+					'diffOptions',
+					'failureThreshold',
+					'failureThresholdType',
+					'timeout',
+				),
 				baselinePath,
 				resultPath,
 				diffPath,
 			}
 		},
-		getTaskCount(testPath: string, taskId: string) {
-			return context.getSuiteInfo(testPath, taskId).task.count
+		async getTaskCount(browserContext: PartialBrowserCommandContext, taskId: string) {
+			return (await context.getSuiteInfo(browserContext, taskId)).task.count
 		},
-		hasImageSnapshot(testPath: string, taskId: string, snapshotFileId: string | undefined, isAutoSnapshot: boolean) {
-			const info = context.getSuiteInfo(testPath, taskId)
-			return file.existFile(join(info.baselineDir, context.getSnapshotFilename(info, snapshotFileId, isAutoSnapshot)))
+		async hasImageSnapshot(
+			browserContext: PartialBrowserCommandContext,
+			taskId: string,
+			snapshotFileId: string | undefined,
+			isAutoSnapshot: boolean,
+		) {
+			const info = await context.getSuiteInfo(browserContext, taskId)
+
+			return file.existFile(
+				resolve(
+					info.projectRoot,
+					info.baselineDir,
+					context.getSnapshotFilename(browserContext, info, snapshotFileId, isAutoSnapshot),
+				),
+			)
 		},
 		getSnapshotFilename(
+			browserContext: PartialBrowserCommandContext,
 			info: { taskId: string; task: { count: number } },
-
 			snapshotFileId: string | undefined,
 			isAutoSnapshot: boolean,
 		) {
 			if (snapshotFileId) return `${snapshotFileId}.png`
-			const customizeSnapshotId = visOptions.customizeSnapshotId ?? (({ id, index }) => `${id}-${index}`)
+			const customizeSnapshotId =
+				getVisOptions(visOptionsRecord, browserContext).customizeSnapshotId ?? (({ id, index }) => `${id}-${index}`)
 			return `${customizeSnapshotId({
 				id: info.taskId,
 				index: info.task.count,
 				isAutoSnapshot,
 			})}.png`
 		},
-		getSuiteInfo(testPath: string, taskId: string) {
-			const suiteId = getSuiteId(state, testPath, visOptions)
-			const suite = state.suites[suiteId]!
+		async getSuiteInfo(browserContext: PartialBrowserCommandContext, taskId: string) {
+			const projectRoot = getProjectRoot(browserContext)
+			const projectState = await state[projectRoot]!
+
+			const suiteId = getSuiteId(projectState, browserContext.testPath, visOptionsRecord)
+			const suite = projectState.suites[suiteId]!
 			const task = (suite.tasks[taskId] = suite.tasks[taskId] ?? { count: 1 })
 			return {
+				projectRoot,
 				suiteId,
 				taskId,
 				baselineDir: suite.baselineDir,
@@ -105,21 +138,21 @@ export function createVisContext() {
 }
 
 async function setupState(
-	suite: PartialBrowserCommandContext,
+	browserContext: PartialBrowserCommandContext,
 	visOptions: Pick<VisOptions, 'snapshotRootDir' | 'platform' | 'subjectDataTestId'>,
 ) {
-	const snapshotRootDir = resolveSnapshotRootDir(suite, visOptions)
-	const projectPath = suite.project.config.root
+	const snapshotRootDir = resolveSnapshotRootDir(browserContext, visOptions)
+	const projectRoot = getProjectRoot(browserContext)
 
 	const state = {
-		projectPath,
-		testTimeout: suite.project.config.testTimeout,
-		hookTimeout: suite.project.config.hookTimeout,
+		projectRoot,
+		testTimeout: browserContext.project.config.testTimeout,
+		hookTimeout: browserContext.project.config.hookTimeout,
 		snapshotRootDir,
 		snapshotBaselineDir: join(snapshotRootDir, BASELINE_DIR),
 		snapshotResultDir: join(snapshotRootDir, RESULT_DIR),
 		snapshotDiffDir: join(snapshotRootDir, DIFF_DIR),
-		snapshotRootPath: join(projectPath, snapshotRootDir),
+		snapshotRootPath: join(projectRoot, snapshotRootDir),
 		subjectDataTestId: visOptions.subjectDataTestId,
 		suites: {},
 	}
@@ -127,7 +160,11 @@ async function setupState(
 	return state
 }
 
-export function createSuite(state: VisState, testPath: string, options: Pick<VisOptions, 'customizeSnapshotSubpath'>) {
+export function createSuite(
+	state: VisProjectState,
+	testPath: string,
+	options: Pick<VisOptions, 'customizeSnapshotSubpath'>,
+) {
 	const suiteId = getSuiteId(state, testPath, options)
 	return {
 		suiteId,
@@ -140,6 +177,18 @@ export function createSuite(state: VisState, testPath: string, options: Pick<Vis
 	}
 }
 
-export function getSuiteId(state: VisState, testPath: string, options: Pick<VisOptions, 'customizeSnapshotSubpath'>) {
-	return getSnapshotSubpath(relative(state.projectPath, testPath), options)
+export function getSuiteId(
+	state: VisProjectState,
+	testPath: string,
+	options: Pick<VisOptions, 'customizeSnapshotSubpath'>,
+) {
+	return getSnapshotSubpath(relative(state.projectRoot, testPath), options)
+}
+
+function getVisOptions(visOptionsRecord: Record<string, VisOptions<any>>, context: PartialBrowserCommandContext) {
+	return visOptionsRecord[getProjectName(context) ?? '__default']!
+}
+
+function getProjectRoot(context: PartialBrowserCommandContext) {
+	return context.project.config.root
 }

package/src/server/vis_context.types.ts

@@ -1,6 +1,9 @@
 export type VisState = {
-	projectPath: string
-	// platform: string
+	[projectPath: string]: Promise<VisProjectState>
+}
+
+export type VisProjectState = {
+	projectRoot: string
 	testTimeout: number
 	hookTimeout: number
 	snapshotRootDir: string
@@ -24,6 +27,7 @@ export type VisState = {
 		}
 	>
 }
+
 export type PartialBrowserCommandContext = {
 	project: {
 		config: {
@@ -34,14 +38,21 @@ export type PartialBrowserCommandContext = {
 			testTimeout: number
 			hookTimeout: number
 		}
+		vite: {
+			config: {
+				test?: {
+					name?: string | undefined
+				}
+			}
+		}
 	}
 	provider: {
 		name: string
-		browserName: string
-		options: {
-			headless: boolean | undefined
-			screenshotFailures: boolean | undefined
-			screenshotDirectory: string | undefined
+		browserName?: string | undefined
+		options?: {
+			headless?: boolean | undefined
+			screenshotFailures?: boolean | undefined
+			screenshotDirectory?: string | undefined
 		}
 	}
 	testPath: string

package/src/testing/stubBrowserCommandContext.ts

@@ -10,6 +10,13 @@ export function createStubPartialBrowserCommandContext({ root, testPath }: { roo
 			config: {
 				root,
 			},
+			vite: {
+				config: {
+					test: {
+						name: 'subject',
+					},
+				},
+			},
 		},
 		provider: {
 			options: {},

package/esm/server/commands/write_image_snapshot.d.ts

@@ -1,12 +0,0 @@
-import type { BrowserCommand } from 'vitest/node';
-export interface WriteImageSnapshotCommand {
-    /**
-     * Writes the image snapshot to the file system.
-     * @param relativeFilePath The file path to write the image snapshot to.
-     * It is relative to the project root.
-     * @param imageBase64 The base64 encoded image.
-     */
-    writeImageSnapshot(relativeFilePath: string, imageBase64: string): Promise<void>;
-}
-export declare const writeImageSnapshot: BrowserCommand<[relativeFilePath: string, imageBase64: string]>;
-//# sourceMappingURL=write_image_snapshot.d.ts.map

package/esm/server/commands/write_image_snapshot.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"write_image_snapshot.d.ts","sourceRoot":"","sources":["../../../src/server/commands/write_image_snapshot.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAIjD,MAAM,WAAW,yBAAyB;IACzC;;;;;OAKG;IACH,kBAAkB,CAAC,gBAAgB,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CAChF;AAED,eAAO,MAAM,kBAAkB,EAAE,cAAc,CAAC,CAAC,gBAAgB,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,CAAC,CAS9F,CAAA"}

package/esm/server/commands/write_image_snapshot.js

@@ -1,9 +0,0 @@
-import { mkdirp } from 'mkdirp';
-import { dirname, isAbsolute } from 'pathe';
-import { file } from "../file.js";
-import { assertIsRelativePath } from "./_assertions.js";
-export const writeImageSnapshot = async (_, relativeFilePath, imageBase64) => {
-    assertIsRelativePath(relativeFilePath, 'relativeFilePath');
-    await mkdirp(dirname(relativeFilePath));
-    await file.writeFile(relativeFilePath, imageBase64, { encoding: 'base64' });
-};