From 89456e9f01ec927d4b53bf5b9bb8fa43e917d977 Mon Sep 17 00:00:00 2001
From: Herman Wikner <wiknerherman@gmail.com>
Date: Fri, 15 Dec 2023 11:54:54 +0100
Subject: [PATCH] test(core): fix navbar tests

---
 .../sanity/src/core/studio/components/navbar/StudioNavbar.tsx   | 2 +-
 .../studio/components/navbar/new-document/NewDocumentButton.tsx | 1 +
 test/e2e/tests/navbar/createNewDocumentNav.spec.ts              | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/sanity/src/core/studio/components/navbar/StudioNavbar.tsx b/packages/sanity/src/core/studio/components/navbar/StudioNavbar.tsx
index 89f1ac1430e4..a70cfb6a541e 100644
--- a/packages/sanity/src/core/studio/components/navbar/StudioNavbar.tsx
+++ b/packages/sanity/src/core/studio/components/navbar/StudioNavbar.tsx
@@ -144,7 +144,7 @@ export function StudioNavbar() {
     <FreeTrialProvider>
       <RootLayer zOffset={100} data-search-open={searchFullscreenOpen}>
         <RootCard
-          data-testid="navbar"
+          data-testid="studio-navbar"
           data-ui="Navbar"
           padding={2}
           scheme="dark"
diff --git a/packages/sanity/src/core/studio/components/navbar/new-document/NewDocumentButton.tsx b/packages/sanity/src/core/studio/components/navbar/new-document/NewDocumentButton.tsx
index 64820ce4b83b..8130b228e28b 100644
--- a/packages/sanity/src/core/studio/components/navbar/new-document/NewDocumentButton.tsx
+++ b/packages/sanity/src/core/studio/components/navbar/new-document/NewDocumentButton.tsx
@@ -151,6 +151,7 @@ export function NewDocumentButton(props: NewDocumentButtonProps) {
   const sharedOpenButtonProps: ButtonProps = useMemo(
     () => ({
       'aria-label': openDialogAriaLabel,
+      'data-testid': 'new-document-button',
       disabled: disabled || loading,
       icon: AddIcon,
       mode: 'bleed',
diff --git a/test/e2e/tests/navbar/createNewDocumentNav.spec.ts b/test/e2e/tests/navbar/createNewDocumentNav.spec.ts
index 0bf704d54c0b..99765819889f 100644
--- a/test/e2e/tests/navbar/createNewDocumentNav.spec.ts
+++ b/test/e2e/tests/navbar/createNewDocumentNav.spec.ts
@@ -3,7 +3,7 @@ import {test} from '@sanity/test'
 
 test('create new document from menu button', async ({page, baseURL}) => {
   await page.goto(baseURL ?? '/test/content')
-  await page.getByLabel('Create new document').click()
+  await page.getByTestId('new-document-button').click()
   await page.getByTestId('new-document-button-search-input').fill('Author')
   const authorLink = await page.getByRole('link', {name: 'Author', exact: true})