Tune locator

PostHog · Jan 9, 2024 · e688f7b · e688f7b
1 parent c82cacd
commit e688f7b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.storybook/test-runner.ts b/.storybook/test-runner.ts
@@ -167,7 +167,12 @@ async function expectStoryToMatchSceneSnapshot(
     theme: SnapshotTheme
 ): Promise<void> {
     // Using .or() because in some cases there's no <main> (primarily logged-out screens)
-    await expectLocatorToMatchStorySnapshot(page.locator('main').or(page), context, browser, theme)
+    await expectLocatorToMatchStorySnapshot(
+        page.locator('main').or(page.locator('body')).first(),
+        context,
+        browser,
+        theme
+    )
 }
 
 async function expectStoryToMatchComponentSnapshot(