Test fix

Amsterdam · Jan 5, 2024 · 7476521 · 7476521
1 parent 89fbc3b
commit 7476521
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/packages/react/src/Header/Header.test.tsx b/packages/react/src/Header/Header.test.tsx
@@ -70,9 +70,9 @@ describe('Header', () => {
   })
 
   it('renders with menu button', () => {
-    render(<Header {...defaultProps} menu={<button>Menu Button</button>} />)
+    render(<Header {...defaultProps} menu={<nav>Navigation</nav>} />)
 
-    const menu = screen.getByRole('button')
+    const menu = screen.getByRole('button', { name: 'Open Menu' })
 
     expect(menu).toBeInTheDocument()
   })

diff --git a/packages/react/src/Header/Header.tsx b/packages/react/src/Header/Header.tsx
@@ -91,10 +91,11 @@ export const Header = forwardRef(
         </Grid>
         {menu && (
           <MegaMenu isOpen={isMegaMenuOpen} className="amsterdam-header__mega-menu">
-            {cloneElement(links as ReactElement, {
-              alignEnd: false,
-              className: 'amsterdam-page-menu--links',
-            })}
+            {links &&
+              cloneElement(links as ReactElement, {
+                alignEnd: false,
+                className: 'amsterdam-page-menu--links',
+              })}
             {menu}
           </MegaMenu>
         )}