From e98960f007512dca976d3700f5187638d015566c Mon Sep 17 00:00:00 2001
From: Tyler Jones <tylerjdev@github.com>
Date: Tue, 29 Oct 2024 09:25:57 -0400
Subject: [PATCH] Fix tests

---
 packages/react/src/Overlay/Overlay.test.tsx | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/react/src/Overlay/Overlay.test.tsx b/packages/react/src/Overlay/Overlay.test.tsx
index d80db8c755a..5a4fc97516a 100644
--- a/packages/react/src/Overlay/Overlay.test.tsx
+++ b/packages/react/src/Overlay/Overlay.test.tsx
@@ -154,7 +154,7 @@ describe('Overlay', () => {
     const user = userEvent.setup()
     const container = render(
       <ThemeProvider>
-        <PositionedOverlays right />
+        <PositionedOverlays role="dialog" right />
       </ThemeProvider>,
     )
 
@@ -162,8 +162,10 @@ describe('Overlay', () => {
     await user.click(container.getByText('Open right overlay'))
     expect(container.getByText('Look! right aligned')).toBeInTheDocument()
 
-    const overlay = container.getByText('Look! right aligned').parentElement?.parentElement
+    const innerOverlay = container.getByText('Look! right aligned')
+    const overlay = container.getByRole('dialog')
 
+    expect(innerOverlay).toBeInTheDocument()
     expect(overlay).toHaveStyle({position: 'fixed', right: 0})
     expect(overlay).not.toHaveStyle({left: 0})
 
@@ -182,7 +184,7 @@ describe('Overlay', () => {
     const user = userEvent.setup()
     const container = render(
       <ThemeProvider>
-        <PositionedOverlays />
+        <PositionedOverlays role="dialog" />
       </ThemeProvider>,
     )
 
@@ -190,7 +192,10 @@ describe('Overlay', () => {
     await user.click(container.getByText('Open left overlay'))
     expect(container.getByText('Look! left aligned')).toBeInTheDocument()
 
-    const overlay = container.getByText('Look! left aligned').parentElement?.parentElement
+    const innerOverlay = container.getByText('Look! left aligned')
+    const overlay = container.getByRole('dialog')
+
+    expect(innerOverlay).toBeInTheDocument()
     expect(overlay).toHaveStyle({left: 0, position: 'absolute'})
 
     spy.mockRestore()