run postVisit on failures as well

storybookjs · Jul 16, 2024 · 7268a7d · 7268a7d
1 parent 5d012d8
commit 7268a7d
Show file tree

Hide file tree

Showing 5 changed files with 566 additions and 177 deletions.
diff --git a/src/csf/__snapshots__/transformCsf.test.ts.snap b/src/csf/__snapshots__/transformCsf.test.ts.snap
@@ -29,12 +29,25 @@ if (!require.main) {
           if (globalThis.__sbPreVisit) {
             await globalThis.__sbPreVisit(page, context);
           }
-          const result = await page.evaluate(({
-            id,
-            hasPlayFn
-          }) => __test(id, hasPlayFn), {
-            id: "button--primary"
-          });
+          let result;
+          try {
+            result = await page.evaluate(({
+              id,
+              hasPlayFn
+            }) => __test(id, hasPlayFn), {
+              id: "button--primary"
+            });
+          } catch (err) {
+            if (err.toString().includes('Execution context was destroyed')) {
+              throw err;
+            } else {
+              await globalThis.__sbPostVisit(page, {
+                ...context,
+                hasFailure: true
+              });
+              throw err;
+            }
+          }
           if (globalThis.__sbPostVisit) {
             await globalThis.__sbPostVisit(page, context);
           }
@@ -96,12 +109,25 @@ if (!require.main) {
           if (globalThis.__sbPreVisit) {
             await globalThis.__sbPreVisit(page, context);
           }
-          const result = await page.evaluate(({
-            id,
-            hasPlayFn
-          }) => __test(id, hasPlayFn), {
-            id: "button--primary"
-          });
+          let result;
+          try {
+            result = await page.evaluate(({
+              id,
+              hasPlayFn
+            }) => __test(id, hasPlayFn), {
+              id: "button--primary"
+            });
+          } catch (err) {
+            if (err.toString().includes('Execution context was destroyed')) {
+              throw err;
+            } else {
+              await globalThis.__sbPostVisit(page, {
+                ...context,
+                hasFailure: true
+              });
+              throw err;
+            }
+          }
           if (globalThis.__sbPostVisit) {
             await globalThis.__sbPostVisit(page, context);
           }
@@ -149,12 +175,25 @@ if (!require.main) {
           if (globalThis.__sbPreVisit) {
             await globalThis.__sbPreVisit(page, context);
           }
-          const result = await page.evaluate(({
-            id,
-            hasPlayFn
-          }) => __test(id, hasPlayFn), {
-            id: "button--primary"
-          });
+          let result;
+          try {
+            result = await page.evaluate(({
+              id,
+              hasPlayFn
+            }) => __test(id, hasPlayFn), {
+              id: "button--primary"
+            });
+          } catch (err) {
+            if (err.toString().includes('Execution context was destroyed')) {
+              throw err;
+            } else {
+              await globalThis.__sbPostVisit(page, {
+                ...context,
+                hasFailure: true
+              });
+              throw err;
+            }
+          }
           if (globalThis.__sbPostVisit) {
             await globalThis.__sbPostVisit(page, context);
           }

diff --git a/src/playwright/hooks.ts b/src/playwright/hooks.ts
@@ -5,6 +5,7 @@ export type TestContext = {
   id: string;
   title: string;
   name: string;
+  hasFailure?: boolean;
 };
 
 export type PrepareContext = {