adding wait for stories to be ready

pixeleye-io · May 29, 2024 · d1da2c3 · d1da2c3
1 parent 821e076
commit d1da2c3
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/.changeset/violet-ways-run.md b/.changeset/violet-ways-run.md
@@ -0,0 +1,5 @@
+---
+"@pixeleye/storybook": patch
+---
+
+Adding wait for stories to be ready
diff --git a/integrations/storybook/src/browser/stories.ts b/integrations/storybook/src/browser/stories.ts
@@ -30,6 +30,7 @@ interface Channel {
 export type SBWindow = typeof window & {
   __STORYBOOK_CLIENT_API__: API;
   __STORYBOOK_PREVIEW__: {
+    ready: () => boolean;
     channel: Channel;
     storyStoreValue?: {
       cacheAllCSFFiles: () => Promise<void>;

diff --git a/integrations/storybook/src/capture.ts b/integrations/storybook/src/capture.ts
@@ -44,14 +44,21 @@ export async function captureStories({
     }
   );
 
+  await page.waitForFunction(
+    () => (window as SBWindow).__STORYBOOK_PREVIEW__?.ready?.(),
+    {
+      timeout: 60_000,
+    }
+  );
+
   await Promise.race([
     page
       .waitForFunction(
         () =>
           (window as SBWindow).__STORYBOOK_CLIENT_API__._storyStore!
             .cacheAllCSFFiles,
         {
-          timeout: 60_000,
+          timeout: 30_000,
         }
       )
       .catch(() => {}),
@@ -61,7 +68,7 @@ export async function captureStories({
           (window as SBWindow).__STORYBOOK_PREVIEW__.storyStoreValue!
             .cacheAllCSFFiles,
         {
-          timeout: 60_000,
+          timeout: 30_000,
         }
       )
       .catch(() => {}),