tests extensions

webapp/src/components/MainPage.test.js

@@ -18,13 +18,41 @@ describe('MainPage', () => {
     useAuthUser.mockReturnValue({ username: 'testUser' });
     const { getByText } = render(<MainPage />);
     expect(getByText('Welcome back, testUser!')).toBeInTheDocument();
+    expect(getByText('Start Playing')).toBeInTheDocument();
+    expect(getByText('Squads')).toBeInTheDocument();
+    expect(getByText('My stats')).toBeInTheDocument();
   });
 
   it('renders welcome message for unauthenticated user', () => {
     useIsAuthenticated.mockReturnValue(() => false);
     const { getByText } = render(<MainPage />);
     expect(getByText('Welcome to WIQ, Please log in to play!')).toBeInTheDocument();
   });
-
-
+  it('navigates to /play when Start Playing button is clicked', () => {
+    useIsAuthenticated.mockReturnValue(() => true);
+    useAuthUser.mockReturnValue({ username: 'testUser' });
+    const { getByText } = render(<MainPage />);
+    fireEvent.click(getByText('Start Playing'));
+    expect(mock).toHaveBeenCalledWith('/play');
+  });
+  it('navigates to /squads when Squads button is clicked', () => {
+    useIsAuthenticated.mockReturnValue(() => true);
+    useAuthUser.mockReturnValue({ username: 'testUser' });
+    const { getByText } = render(<MainPage />);
+    fireEvent.click(getByText('Squads'));
+    expect(mock).toHaveBeenCalledWith('/squads');
+  });
+  it('navigates to /userProfile when My stats button is clicked', () => {
+    useIsAuthenticated.mockReturnValue(() => true);
+    useAuthUser.mockReturnValue({ username: 'testUser' });
+    const { getByText } = render(<MainPage />);
+    fireEvent.click(getByText('My stats'));
+    expect(mock).toHaveBeenCalledWith('/userProfile');
+  });
+  it('navigates to /register when Create account button is clicked', () => {
+    useIsAuthenticated.mockReturnValue(() => false);
+    const { getByText } = render(<MainPage />);
+    fireEvent.click(getByText('Create account'));
+    expect(mock).toHaveBeenCalledWith('/register');
+  });
 });

webapp/src/components/Navbar.test.js

@@ -43,5 +43,17 @@ describe('Navbar', () => {
     expect(getByText('Rankings')).toBeInTheDocument();
     expect(getByText('Log In')).toBeInTheDocument();
   });
+  it('calls signOut when Log out button is clicked', () => {
+    useIsAuthenticated.mockReturnValue(() => true);
+    useSignOut.mockReturnValue(() => {});
+    const { getByText } = render(
+      <MemoryRouter>
+        <Navbar />
+      </MemoryRouter>
+    );
+
+    fireEvent.click(getByText('Log out'));
+    expect(useSignOut).toHaveBeenCalled();
+  });
 
 });