vitest-plugin-vis 1.0.3 → 1.0.5

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

@@ -1 +1 @@
-{"version":3,"file":"match_image_snapshot.d.ts","sourceRoot":"","sources":["../../../src/server/commands/match_image_snapshot.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAyB,MAAM,aAAa,CAAA;AAIxE,OAAO,KAAK,EACX,2BAA2B,EAC3B,sBAAsB,EACtB,2BAA2B,EAC3B,MAAM,uBAAuB,CAAA;AAM9B,MAAM,WAAW,yBAAyB;IACzC,kBAAkB,EAAE,CACnB,MAAM,EAAE,MAAM,GAAG,SAAS,EAC1B,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,yBAAyB,GAAG,SAAS,KAC3C,OAAO,CAAC,IAAI,CAAC,CAAA;CAClB;AAED,MAAM,WAAW,yBAChB,SAAQ,2BAA2B,EAClC,sBAAsB,EACtB,2BAA2B;IAC5B;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;CACnC;AAED,eAAO,MAAM,kBAAkB,EAAE,cAAc,CAC9C;IAAC,MAAM,EAAE,MAAM;IAAE,OAAO,EAAE,MAAM;IAAE,OAAO,CAAC,EAAE,yBAAyB,GAAG,SAAS;CAAC,CA6DlF,CAAA"}
+{"version":3,"file":"match_image_snapshot.d.ts","sourceRoot":"","sources":["../../../src/server/commands/match_image_snapshot.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAyB,MAAM,aAAa,CAAA;AAIxE,OAAO,KAAK,EACX,2BAA2B,EAC3B,sBAAsB,EACtB,2BAA2B,EAC3B,MAAM,uBAAuB,CAAA;AAM9B,MAAM,WAAW,yBAAyB;IACzC,kBAAkB,EAAE,CACnB,MAAM,EAAE,MAAM,GAAG,SAAS,EAC1B,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,yBAAyB,GAAG,SAAS,KAC3C,OAAO,CAAC,IAAI,CAAC,CAAA;CAClB;AAED,MAAM,WAAW,yBAChB,SAAQ,2BAA2B,EAClC,sBAAsB,EACtB,2BAA2B;IAC5B;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;CACnC;AAED,eAAO,MAAM,kBAAkB,EAAE,cAAc,CAC9C;IAAC,MAAM,EAAE,MAAM;IAAE,OAAO,EAAE,MAAM;IAAE,OAAO,CAAC,EAAE,yBAAyB,GAAG,SAAS;CAAC,CAiElF,CAAA"}

package/esm/server/commands/match_image_snapshot.js

@@ -22,10 +22,10 @@ export const matchImageSnapshot = async (context, taskId, subject, options) => {
     const baselineBuffer = await file.tryReadFile(info.baselinePath);
     if (!baselineBuffer) {
         if (isBase64String(subject)) {
-            await writeSnapshot(subject, info.baselinePath);
+            await writeSnapshot(info.baselinePath, subject);
         }
         else {
-            await takeSnapshotByBrowser(context, subject, info.baselinePath, options);
+            await takeSnapshotByBrowser(context, info.baselinePath, subject, options);
         }
         return;
     }
@@ -45,8 +45,11 @@ export const matchImageSnapshot = async (context, taskId, subject, options) => {
 				Expected:   ${resolve(context.project.runner.root, info.baselinePath)}
 				Actual:     ${resolve(context.project.runner.root, info.resultPath)}`);
     }
-    const diffBase64 = PNG.sync.write(diffImage).toString('base64');
-    await writeSnapshot(diffBase64, info.diffPath);
+    if (context.project.config.snapshotOptions.updateSnapshot === 'all') {
+        await writeSnapshotBuffer(info.baselinePath, resultImage.data);
+        return;
+    }
+    await writeSnapshotBuffer(info.diffPath, diffImage.data);
     throw new Error(dedent `Snapshot \`${taskId}\` mismatched
 
 			${options?.failureThreshold
@@ -63,21 +66,25 @@ export const matchImageSnapshot = async (context, taskId, subject, options) => {
 };
 async function takeSnapshot(context, subject, filePath, options) {
     if (isBase64String(subject)) {
-        return Buffer.from(await writeSnapshot(subject, filePath), 'base64');
+        await writeSnapshot(filePath, subject);
+        return Buffer.from(subject, 'base64');
     }
-    return takeSnapshotByBrowser(context, subject, filePath, options);
+    return takeSnapshotByBrowser(context, filePath, subject, options);
 }
-async function takeSnapshotByBrowser(context, subject, filePath, options) {
+async function takeSnapshotByBrowser(context, filePath, subject, options) {
     await mkdirp(dirname(filePath));
     const browser = browserApi(context);
     return browser.takeScreenshot(filePath, subject, {
         timeout: options?.timeout,
     });
 }
-async function writeSnapshot(subject, filePath) {
+async function writeSnapshot(filePath, subject) {
+    await mkdirp(dirname(filePath));
+    await file.writeFile(filePath, subject, { encoding: 'base64' });
+}
+async function writeSnapshotBuffer(filePath, subject) {
     await mkdirp(dirname(filePath));
-    await file.writeFileBase64(filePath, subject);
-    return subject;
+    await file.writeFile(filePath, subject);
 }
 function alignImageSizes(baseline, result) {
     if (isSameSize(baseline, result))

package/esm/server/file.d.ts

@@ -1,6 +1,7 @@
+import { writeFile } from 'node:fs/promises';
 export declare const file: {
     tryReadFile(filePath: string): Promise<Buffer | undefined>;
-    writeFileBase64(filePath: string, data: string): Promise<void>;
+    writeFile: typeof writeFile;
     existFile(filePath: string): Promise<boolean>;
 };
 //# sourceMappingURL=file.d.ts.map

package/esm/server/file.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"file.d.ts","sourceRoot":"","sources":["../../src/server/file.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,IAAI;0BACY,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;8BAGtC,MAAM,QAAQ,MAAM;wBAG1B,MAAM;CAK1B,CAAA"}
+{"version":3,"file":"file.d.ts","sourceRoot":"","sources":["../../src/server/file.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAE5D,eAAO,MAAM,IAAI;0BACY,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;;wBAI5C,MAAM;CAK1B,CAAA"}

package/esm/server/file.js

@@ -3,9 +3,7 @@ export const file = {
     async tryReadFile(filePath) {
         return readFile(filePath).catch(() => undefined);
     },
-    writeFileBase64(filePath, data) {
-        return writeFile(filePath, data, { encoding: 'base64' });
-    },
+    writeFile,
     existFile(filePath) {
         return stat(filePath)
             .then((s) => s.isFile())

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vitest-plugin-vis",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "Vitest visual testing plugin",
   "keywords": [
     "image-snapshot",

package/src/server/commands/match_image_snapshot.ts

@@ -50,9 +50,9 @@ export const matchImageSnapshot: BrowserCommand<
 	const baselineBuffer = await file.tryReadFile(info.baselinePath)
 	if (!baselineBuffer) {
 		if (isBase64String(subject)) {
-			await writeSnapshot(subject, info.baselinePath)
+			await writeSnapshot(info.baselinePath, subject)
 		} else {
-			await takeSnapshotByBrowser(context, subject, info.baselinePath, options)
+			await takeSnapshotByBrowser(context, info.baselinePath, subject, options)
 		}
 		return
 	}
@@ -77,8 +77,12 @@ export const matchImageSnapshot: BrowserCommand<
 		)
 	}
 
-	const diffBase64 = PNG.sync.write(diffImage).toString('base64')
-	await writeSnapshot(diffBase64, info.diffPath)
+	if (context.project.config.snapshotOptions.updateSnapshot === 'all') {
+		await writeSnapshotBuffer(info.baselinePath, resultImage.data)
+		return
+	}
+
+	await writeSnapshotBuffer(info.diffPath, diffImage.data)
 
 	throw new Error(
 		dedent`Snapshot \`${taskId}\` mismatched
@@ -106,15 +110,16 @@ async function takeSnapshot(
 	options: ImageSnapshotTimeoutOptions | undefined,
 ) {
 	if (isBase64String(subject)) {
-		return Buffer.from(await writeSnapshot(subject, filePath), 'base64')
+		await writeSnapshot(filePath, subject)
+		return Buffer.from(subject, 'base64')
 	}
-	return takeSnapshotByBrowser(context, subject, filePath, options)
+	return takeSnapshotByBrowser(context, filePath, subject, options)
 }
 
 async function takeSnapshotByBrowser(
 	context: BrowserCommandContext,
-	subject: string,
 	filePath: string,
+	subject: string,
 	options: ImageSnapshotTimeoutOptions | undefined,
 ) {
 	await mkdirp(dirname(filePath))
@@ -124,10 +129,13 @@ async function takeSnapshotByBrowser(
 	})
 }
 
-async function writeSnapshot(subject: string, filePath: string) {
+async function writeSnapshot(filePath: string, subject: string) {
+	await mkdirp(dirname(filePath))
+	await file.writeFile(filePath, subject, { encoding: 'base64' })
+}
+async function writeSnapshotBuffer(filePath: string, subject: Buffer) {
 	await mkdirp(dirname(filePath))
-	await file.writeFileBase64(filePath, subject)
-	return subject
+	await file.writeFile(filePath, subject)
 }
 
 function alignImageSizes(baseline: PNG, result: PNG) {

package/src/server/file.ts

@@ -4,9 +4,7 @@ export const file = {
 	async tryReadFile(filePath: string): Promise<Buffer | undefined> {
 		return readFile(filePath).catch(() => undefined)
 	},
-	writeFileBase64(filePath: string, data: string) {
-		return writeFile(filePath, data, { encoding: 'base64' })
-	},
+	writeFile,
 	existFile(filePath: string) {
 		return stat(filePath)
 			.then((s) => s.isFile())