@argos-ci/storybook 5.0.5 → 5.0.7-alpha.0

package/dist/test-runner.js

@@ -111,7 +111,7 @@ async function storybookArgosScreenshot(handler, context, options) {
           }
         });
         allAttachments.push(...attachments);
-        await context.setViewportSize("default");
+        await context.setViewportSize("initial");
         await handler.evaluate(() => {
           delete globalThis.__ARGOS_CURRENT_MODE;
         });
@@ -124,6 +124,7 @@ async function storybookArgosScreenshot(handler, context, options) {
         options: argosOptions,
         globals: context.story.globals ?? {}
       });
+      await context.setViewportSize("initial");
       allAttachments.push(...attachments);
     }
   } else if (context.mode === "manual") {
@@ -241,7 +242,7 @@ async function argosScreenshot(page, context, options) {
       },
       setViewportSize: async (size) => {
         const actualSize = await page.viewportSize();
-        const absoluteSize = size === "default" ? DEFAULT_PLAYWRIGHT_VIEWPORT_SIZE : size;
+        const absoluteSize = size === "default" || size === "initial" ? DEFAULT_PLAYWRIGHT_VIEWPORT_SIZE : size;
         if (!actualSize || actualSize.height !== absoluteSize.height || actualSize.width !== absoluteSize.width) {
           await page.setViewportSize(absoluteSize);
           await page.waitForFunction(

package/dist/vitest-plugin.d.ts

@@ -15,7 +15,7 @@ type StorybookScreenshotContext<Handler extends Page | Frame> = {
     name: string;
     playwrightLibraries: string[];
     test?: MetadataConfig["test"];
-    setViewportSize: (size: ViewportSize | "default") => Promise<void>;
+    setViewportSize: (size: ViewportSize | "default" | "initial") => Promise<void>;
     beforeScreenshot?: (input: {
         handler: Handler;
         globals: StorybookGlobals;

package/dist/vitest-plugin.js

@@ -108,7 +108,7 @@ async function storybookArgosScreenshot(handler, context, options) {
           }
         });
         allAttachments.push(...attachments);
-        await context.setViewportSize("default");
+        await context.setViewportSize("initial");
         await handler.evaluate(() => {
           delete globalThis.__ARGOS_CURRENT_MODE;
         });
@@ -121,6 +121,7 @@ async function storybookArgosScreenshot(handler, context, options) {
         options: argosOptions,
         globals: context.story.globals ?? {}
       });
+      await context.setViewportSize("initial");
       allAttachments.push(...attachments);
     }
   } else if (context.mode === "manual") {
@@ -263,19 +264,26 @@ var createArgosScreenshotCommand = (pluginOptions) => {
               if (!iframe.contentDocument) {
                 throw new Error("Vitest iframe contentDocument not found");
               }
-              if (size2 === "default") {
-                if (iframe.dataset.defaultWidth && iframe.dataset.defaultHeight) {
-                  iframe.style.width = iframe.dataset.defaultWidth;
-                  iframe.style.height = iframe.dataset.defaultHeight;
+              if (size2 === "initial") {
+                if (iframe.dataset.initialWidth && iframe.dataset.initialHeight) {
+                  iframe.style.width = iframe.dataset.initialWidth;
+                  iframe.style.height = iframe.dataset.initialHeight;
                 }
-              } else {
-                if (!iframe.dataset.defaultWidth && !iframe.dataset.defaultHeight) {
-                  iframe.dataset.defaultWidth = iframe.style.width;
-                  iframe.dataset.defaultHeight = iframe.style.height;
-                }
-                iframe.style.height = "auto";
+                return;
+              }
+              if (!iframe.dataset.initialWidth && !iframe.dataset.initialHeight) {
+                iframe.dataset.initialWidth = iframe.style.width;
+                iframe.dataset.initialHeight = iframe.style.height;
+              }
+              if (size2 !== "default") {
                 iframe.style.width = `${size2.width}px`;
-                iframe.style.height = fullPage ? `${iframe.contentDocument.body.offsetHeight}px` : `${size2.height}px`;
+              }
+              if (fullPage) {
+                iframe.style.height = "auto";
+                iframe.style.height = `${iframe.contentDocument.body.offsetHeight}px`;
+              } else if (size2 !== "default") {
+                iframe.style.height = "auto";
+                iframe.style.height = `${size2.height}px`;
               }
             },
             { size, fullPage: screenshotOptions.fullPage ?? !fitToContent }

package/dist/vitest.d.ts

@@ -13,7 +13,7 @@ type StorybookScreenshotContext<Handler extends Page | Frame> = {
     name: string;
     playwrightLibraries: string[];
     test?: MetadataConfig["test"];
-    setViewportSize: (size: ViewportSize | "default") => Promise<void>;
+    setViewportSize: (size: ViewportSize | "default" | "initial") => Promise<void>;
     beforeScreenshot?: (input: {
         handler: Handler;
         globals: StorybookGlobals;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@argos-ci/storybook",
   "description": "Visual testing for Storybook test runner.",
-  "version": "5.0.5",
+  "version": "5.0.7-alpha.0+d1894ac",
   "author": "Smooth Code",
   "license": "MIT",
   "repository": {
@@ -50,12 +50,12 @@
     "node": ">=20.0.0"
   },
   "dependencies": {
-    "@argos-ci/playwright": "6.1.8",
-    "@argos-ci/util": "3.1.1"
+    "@argos-ci/playwright": "6.1.9-alpha.4+d1894ac",
+    "@argos-ci/util": "3.1.2-alpha.4+d1894ac"
   },
   "devDependencies": {
-    "@argos-ci/cli": "3.1.0",
-    "@argos-ci/core": "4.2.0",
+    "@argos-ci/cli": "3.1.1-alpha.4+d1894ac",
+    "@argos-ci/core": "4.2.1-alpha.4+d1894ac",
     "@argos-ci/util": "workspace:*",
     "@storybook/addon-docs": "^9.1.10",
     "@storybook/addon-links": "^9.1.10",
@@ -98,5 +98,5 @@
     "check-format": "prettier --check --ignore-unknown --ignore-path=./.gitignore --ignore-path=../../.gitignore --ignore-path=../../.prettierignore .",
     "lint": "eslint ."
   },
-  "gitHead": "9437efd6cefa301411b44bc7955ae8df1a14e9f7"
+  "gitHead": "d1894acc62fb886d4673c88cc7e8cc95e81db78b"
 }