@argos-ci/storybook 6.0.16 → 6.0.18

package/dist/vitest-plugin.mjs

@@ -270,6 +270,25 @@ const createArgosScreenshotCommand = (pluginOptions) => {
 		const frame = await ctx.frame();
 		const fitToContent = getFitToContentFromParameters(testContext.story.parameters);
 		const after = await before(ctx);
+		const options = applyFitToContent(screenshotOptions, fitToContent);
+		const userBeforeScreenshot = options?.beforeScreenshot;
+		const optionsWithFit = {
+			...options,
+			beforeScreenshot: async (api) => {
+				await userBeforeScreenshot?.(api);
+				await ctx.page.evaluate(({ fitWidth }) => {
+					const iframe = document.querySelector("iframe[data-vitest=\"true\"]");
+					if (!(iframe instanceof HTMLIFrameElement) || !iframe.contentDocument) return;
+					const { body, documentElement } = iframe.contentDocument;
+					const contentHeight = Math.max(body.scrollHeight, body.offsetHeight, documentElement.scrollHeight);
+					if (contentHeight > iframe.clientHeight) iframe.style.height = `${contentHeight}px`;
+					if (fitWidth) {
+						const contentWidth = Math.max(body.scrollWidth, body.offsetWidth, documentElement.scrollWidth);
+						if (contentWidth > iframe.clientWidth) iframe.style.width = `${contentWidth}px`;
+					}
+				}, { fitWidth: Boolean(fitToContent) });
+			}
+		};
 		const attachments = await storybookArgosScreenshot(frame, {
 			...testContext,
 			playwrightLibraries: ["@storybook/addon-vitest"],
@@ -304,7 +323,7 @@ const createArgosScreenshotCommand = (pluginOptions) => {
 					fullPage: screenshotOptions.fullPage ?? !fitToContent
 				});
 			}
-		}, applyFitToContent(screenshotOptions, fitToContent));
+		}, optionsWithFit);
 		await after();
 		return attachments;
 	};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@argos-ci/storybook",
   "description": "Visual testing for Storybook test runner.",
-  "version": "6.0.16",
+  "version": "6.0.18",
   "author": "Smooth Code",
   "license": "MIT",
   "repository": {
@@ -98,5 +98,5 @@
     "check-format": "prettier --check --ignore-unknown --ignore-path=./.gitignore --ignore-path=../../.gitignore --ignore-path=../../.prettierignore .",
     "lint": "eslint ."
   },
-  "gitHead": "7db853496fdaad253d005fb30c4b47269972b7b0"
+  "gitHead": "caaf0f8f6389199e4b94ea43cd014ff72becec94"
 }