create tests for the account type chip

Author: SaharBatol-HDRUK

src/organisms/NavBar/NavBar.test.tsx

@@ -15,6 +15,9 @@ import {
   waitFor,
 } from "../../utils/testUtils";
 import NavBar from "./NavBar";
+import { mockedOrganisation } from "@/mocks/data/organisation";
+import { mockedCustodian } from "@/mocks/data/custodian";
+import { AccountType } from "@/types/accounts";
 
 jest.mock("js-cookie", () => ({
   get: jest.fn(),
@@ -118,8 +121,8 @@ describe("NavBar Component", () => {
         getUser: () => mockedUser(),
         setUser: jest.fn(),
         config: {
-          organisation: undefined,
-          custodian: undefined,
+          organisation: mockedOrganisation,
+          custodian: mockedCustodian,
         },
       } as unknown as StoreState)
     );
@@ -131,6 +134,40 @@ describe("NavBar Component", () => {
     expect(handleLogout).toHaveBeenCalled();
   });
 
+  it("displays 'Organisation' chip when the user belongs to an organisation", () => {
+    mockUseStore.mockImplementation(selector =>
+      selector({
+        getUser: () => mockedUser(),
+        setUser: jest.fn(),
+        config: {
+          organisation: mockedOrganisation,
+          custodian: undefined,
+        },
+      } as unknown as StoreState)
+    );
+
+    render(<NavBar loggedIn />);
+
+    expect(screen.getByText(AccountType.ORGANISATION)).toBeInTheDocument();
+  });
+
+  it("displays 'Custodian' chip when the user belongs to an custodian", () => {
+    mockUseStore.mockImplementation(selector =>
+      selector({
+        getUser: () => mockedUser(),
+        setUser: jest.fn(),
+        config: {
+          organisation: undefined,
+          custodian: mockedCustodian,
+        },
+      } as unknown as StoreState)
+    );
+
+    render(<NavBar loggedIn />);
+
+    expect(screen.getByText(AccountType.CUSTODIAN)).toBeInTheDocument();
+  });
+
   it("displays 'My Account' and 'Sign Out' if the user is authenticated", () => {
     mockGetMe.mockResolvedValueOnce({
       status: 200,