Tune locator

PostHog · Mar 11, 2024 · 685e72f · 685e72f
1 parent 47b3234
commit 685e72f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.storybook/test-runner.ts b/.storybook/test-runner.ts
@@ -166,7 +166,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(