fix: fix disableCssAnimation bug when using multiple viewports

reg-viz · Jun 20, 2022 · 2bdab36 · 2bdab36
1 parent ada0d97
commit 2bdab36
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/storycap/src/node/capturing-browser.ts b/packages/storycap/src/node/capturing-browser.ts
@@ -109,6 +109,12 @@ export class CapturingBrowser extends StoryPreviewBrowser {
     Object.entries(exposed).forEach(([k, f]) => this.page.exposeFunction(k, f));
   }
 
+  private async reload() {
+    await this.page.reload();
+    await sleep(this.opt.viewportDelay);
+    await this.addStyles();
+  }
+
   private async waitIfTouched() {
     if (!this.touched) return;
     await sleep(this.opt.stateChangeDelay);
@@ -255,7 +261,7 @@ export class CapturingBrowser extends StoryPreviewBrowser {
       this.viewport = nextViewport;
       if (willBeReloaded || this.opt.reloadAfterChangeViewport) {
         this.processedStories.delete(this.currentRequestId);
-        await Promise.all([this.page.reload(), this.waitForOptionsFromBrowser()]);
+        await Promise.all([this.reload(), this.waitForOptionsFromBrowser()]);
       } else {
         await sleep(this.opt.viewportDelay);
       }