@argos-ci/storybook 5.0.4 → 5.0.5

package/dist/vitest-plugin.js

@@ -252,25 +252,34 @@ var createArgosScreenshotCommand = (pluginOptions) => {
         ...testContext,
         playwrightLibraries: ["@storybook/addon-vitest"],
         setViewportSize: async (size) => {
-          await ctx.page.evaluate((size2) => {
-            const iframe = document.querySelector('iframe[data-vitest="true"]');
-            if (!(iframe instanceof HTMLIFrameElement)) {
-              throw new Error("Vitest iframe not found");
-            }
-            if (size2 === "default") {
-              if (iframe.dataset.defaultWidth && iframe.dataset.defaultHeight) {
-                iframe.style.width = iframe.dataset.defaultWidth;
-                iframe.style.height = iframe.dataset.defaultHeight;
+          await ctx.page.evaluate(
+            ({ size: size2, fullPage }) => {
+              const iframe = document.querySelector(
+                'iframe[data-vitest="true"]'
+              );
+              if (!(iframe instanceof HTMLIFrameElement)) {
+                throw new Error("Vitest iframe not found");
               }
-            } else {
-              if (!iframe.dataset.defaultWidth && !iframe.dataset.defaultHeight) {
-                iframe.dataset.defaultWidth = iframe.style.width;
-                iframe.dataset.defaultHeight = iframe.style.height;
+              if (!iframe.contentDocument) {
+                throw new Error("Vitest iframe contentDocument not found");
               }
-              iframe.style.width = `${size2.width}px`;
-              iframe.style.height = `${size2.height}px`;
-            }
-          }, size);
+              if (size2 === "default") {
+                if (iframe.dataset.defaultWidth && iframe.dataset.defaultHeight) {
+                  iframe.style.width = iframe.dataset.defaultWidth;
+                  iframe.style.height = iframe.dataset.defaultHeight;
+                }
+              } else {
+                if (!iframe.dataset.defaultWidth && !iframe.dataset.defaultHeight) {
+                  iframe.dataset.defaultWidth = iframe.style.width;
+                  iframe.dataset.defaultHeight = iframe.style.height;
+                }
+                iframe.style.height = "auto";
+                iframe.style.width = `${size2.width}px`;
+                iframe.style.height = fullPage ? `${iframe.contentDocument.body.offsetHeight}px` : `${size2.height}px`;
+              }
+            },
+            { size, fullPage: screenshotOptions.fullPage ?? !fitToContent }
+          );
         }
       },
       applyFitToContent(screenshotOptions, fitToContent)

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@argos-ci/storybook",
   "description": "Visual testing for Storybook test runner.",
-  "version": "5.0.4",
+  "version": "5.0.5",
   "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": "867a93c5ffc01d55c7385d3ac3903c4c3ab270b6"
+  "gitHead": "9437efd6cefa301411b44bc7955ae8df1a14e9f7"
 }