fix: update ClientRoutes.test.js and Menu.test.js

Greenstand · Jul 19, 2023 · af3bbeb · af3bbeb
1 parent ab9b2de
commit af3bbeb
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 10 deletions.
diff --git a/src/components/Routes/ClientRoutes.test.js b/src/components/Routes/ClientRoutes.test.js
@@ -4,12 +4,19 @@ import ClientRoutes from './ClientRoutes';
 import { render, screen } from '@testing-library/react';
 import { ThemeProvider } from '@mui/material';
 import theme from '../UI/theme';
+import AuthProvider from '../../store/AuthProvider';
+import AuthContext from '../../store/auth-context';
+
 
 describe('ClientRoutes component', () => {
 
   const TestWrapper = (props) => {
     return <ThemeProvider theme={theme}>
-      <Router>{props.children}</Router>
+      <Router>
+        <AuthContext.Provider value={{ isLoggedIn: true }}>
+          {props.children}
+        </AuthContext.Provider>
+      </Router>
     </ThemeProvider>
       ;
   };
@@ -27,9 +34,11 @@ describe('ClientRoutes component', () => {
     await screen.findAllByRole('link');
     await screen.findByAltText(/Greenstand logo/);
 
-    //Logo, Home and Send Tokens for now
-    expect(screen.getAllByRole('link')).toHaveLength(4);
-    expect(screen.getAllByRole('button')).toHaveLength(3);
+    // screen.getByRole('');
+
+    //Logo, Home, Send Tokens, and My Trasnfers for now
+    expect(await screen.findAllByRole('link')).toHaveLength(4);
+    expect(screen.getAllByRole('button')).toHaveLength(4);
 
     expect(screen.getByText(/Home/)).toBeInTheDocument();
     expect(screen.getByText(/Send Tokens/)).toBeInTheDocument();

diff --git a/src/components/layout/Menu/Menu.test.js b/src/components/layout/Menu/Menu.test.js
@@ -24,15 +24,15 @@ describe('Menu component', () => {
     await screen.findAllByRole('link');
     await screen.findByAltText(/Greenstand logo/);
 
-    //Logo, Home and Send Tokens for now
-    expect(screen.getAllByRole('link')).toHaveLength(3);
-    expect(screen.getAllByRole('button')).toHaveLength(4);
-
-    expect(screen.getByText(/Home/)).toBeInTheDocument();
-    expect(screen.getByText(/Send Tokens/)).toBeInTheDocument();
+    //Logo, Home,  Send Tokens, My Transfers for now
+    expect(screen.getAllByRole('link')).toHaveLength(4);
+    expect(screen.getAllByRole('button')).toHaveLength(5);
     expect(screen.getByRole('button', {
       name: /open drawer/,
     })).toBeInTheDocument();
 
+    expect(screen.getByText(/Home/)).toBeInTheDocument();
+    expect(screen.getByText(/Send Tokens/)).toBeInTheDocument();
+    expect(screen.getByText(/My Transfers/)).toBeInTheDocument();
   });
 });