fix: add async and await explicitly

reg-viz · Mar 31, 2024 · a89f409 · a89f409
1 parent ee3e1c9
commit a89f409
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/packages/storycrawler/src/browser/stories-browser.ts b/packages/storycrawler/src/browser/stories-browser.ts
@@ -77,18 +77,17 @@ export class StoriesBrowser extends BaseBrowser {
         timeout: 60_000,
       },
     );
-    await this.page.waitForTimeout(500);
-    await this.page.evaluate(() => {
+    await this.page.evaluate(async () => {
       const api = (window as ExposedWindow).__STORYBOOK_CLIENT_API__ || (window as ExposedWindow).__STORYBOOK_PREVIEW__;
       function isPreviewApi(api: API | PreviewAPI): api is PreviewAPI {
         return (api as PreviewAPI).storyStoreValue !== undefined;
       }
       if (api === undefined) return;
 
       if (isPreviewApi(api)) {
-        return api.storyStoreValue && api.storyStoreValue.cacheAllCSFFiles();
+        return api.storyStoreValue && (await api.storyStoreValue.cacheAllCSFFiles());
       }
-      return api.storyStore?.cacheAllCSFFiles && api.storyStore.cacheAllCSFFiles();
+      return api.storyStore?.cacheAllCSFFiles && (await api.storyStore.cacheAllCSFFiles());
     });
     const result = await this.page.evaluate(() => {
       function isPreviewApi(api: API | PreviewAPI): api is PreviewAPI {