add/update tests

Enterprise-CMCS · Dec 12, 2024 · f9aa787 · f9aa787
1 parent a4c3d8a
commit f9aa787
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 11 deletions.
diff --git a/services/ui-src/src/components/layout/Header.test.tsx b/services/ui-src/src/components/layout/Header.test.tsx
@@ -16,25 +16,19 @@ const mockedUseStore = useStore as jest.MockedFunction<typeof useStore>;
 const headerComponent = (
   <RouterWrappedComponent>
     <ReportContext.Provider value={mockReportMethods}>
-      <Header handleLogout={() => {}} />
+      <Header />
     </ReportContext.Provider>
   </RouterWrappedComponent>
 );
 
 const reportComponent = (
   <RouterWrappedComponent>
     <ReportContext.Provider value={mockReportMethods}>
-      <Header handleLogout={() => {}} />
+      <Header />
     </ReportContext.Provider>
   </RouterWrappedComponent>
 );
 
-/*
- * jest.mock("utils/reports/routing", () => ({
- * isReportFormPage: jest.fn(() => true),
- * }));
- */
-
 describe("<Header />", () => {
   describe("Test Header", () => {
     beforeEach(() => {

diff --git a/services/ui-src/src/components/menus/Menu.test.tsx b/services/ui-src/src/components/menus/Menu.test.tsx
@@ -1,13 +1,27 @@
 import { render, screen } from "@testing-library/react";
-// utils
-import { RouterWrappedComponent } from "utils/testing/setupJest";
+import userEvent from "@testing-library/user-event";
 //components
 import { Menu } from "components";
+// utils
+import {
+  mockUserContext,
+  RouterWrappedComponent,
+} from "utils/testing/setupJest";
+import { UserContext } from "utils";
 import { testA11y } from "utils/testing/commonTests";
 
+const mockLogout = jest.fn();
+
+const mockContext = {
+  ...mockUserContext,
+  logout: mockLogout,
+};
+
 const menuComponent = (
   <RouterWrappedComponent>
-    <Menu handleLogout={() => {}} />
+    <UserContext.Provider value={mockContext}>
+      <Menu />
+    </UserContext.Provider>
   </RouterWrappedComponent>
 );
 
@@ -17,5 +31,12 @@ describe("<Menu />", () => {
     expect(screen.getByTestId("header-menu-dropdown-button")).toBeVisible();
   });
 
+  test("Menu button logout fires logout function", async () => {
+    render(menuComponent);
+    const logoutButton = screen.getByText("Log Out");
+    await userEvent.click(logoutButton);
+    expect(mockLogout).toHaveBeenCalledTimes(1);
+  });
+
   testA11y(menuComponent);
 });