VIA-630 Show or hide pregnancy hub content based on age group

Author: donna-belsey-nhs

src/app/check-and-book-vaccinations/page.test.tsx

@@ -31,58 +31,71 @@ jest.mock("@src/app/_components/hub/PregnancyHubContent", () => ({
     .mockImplementation(() => <p data-testid={"pregnancy-hub-content"}>Pregnancy hub content test</p>),
 }));
 
-const mockAgeGroup = AgeGroup.AGE_25_to_64;
-
-const mockSessionValue: Partial<Session> = {
-  expires: new Date(Date.now() + 60000).toISOString(),
-  user: {
-    nhs_number: "" as NhsNumber,
-    age_group: mockAgeGroup,
-  },
+const mockSessionDataForAgeGroup = (ageGroup: AgeGroup): Partial<Session> => {
+  return {
+    expires: new Date(Date.now() + 60000).toISOString(),
+    user: {
+      nhs_number: "" as NhsNumber,
+      age_group: ageGroup,
+    },
+  };
 };
 
-const mockSession = { data: mockSessionValue, status: "authenticated" };
+describe("Vaccination Hub Page", () => {
+  describe("for all ages", () => {
+    const mockAgeGroup = AgeGroup.AGE_25_to_64;
 
-jest.mock("next-auth/react", () => ({
-  useSession: () => mockSession,
-}));
+    beforeEach(async () => {
+      (auth as jest.Mock).mockResolvedValue(mockSessionDataForAgeGroup(mockAgeGroup));
 
-describe("Vaccination Hub Page", () => {
-  beforeEach(async () => {
-    (auth as jest.Mock).mockResolvedValue(mockSessionValue);
+      render(await VaccinationsHub());
+    });
 
-    render(await VaccinationsHub());
-  });
+    it("renders main heading", async () => {
+      expectHeadingToBeRendered();
+    });
 
-  it("renders main heading", async () => {
-    expectHeadingToBeRendered();
-  });
+    it("renders age based cards for user", () => {
+      const ageBasedHubCards: HTMLElement = screen.getByTestId("age-based-hub-cards");
 
-  it("renders age based cards for user", () => {
-    const ageBasedHubCards: HTMLElement = screen.getByTestId("age-based-hub-cards");
+      expect(ageBasedHubCards).toBeVisible();
+      expect(AgeBasedHubCards).toHaveBeenCalledWith(
+        {
+          ageGroup: mockAgeGroup,
+        },
+        undefined,
+      );
+    });
 
-    expect(ageBasedHubCards).toBeVisible();
-    expect(AgeBasedHubCards).toHaveBeenCalledWith(
-      {
-        ageGroup: mockAgeGroup,
-      },
-      undefined,
-    );
-  });
+    it("should show at risk expander ", () => {
+      const atRiskHubExpander: HTMLElement = screen.getByTestId("at-risk-hub-expander");
+      expect(atRiskHubExpander).toBeVisible();
+    });
 
-  it("should show at risk expander ", () => {
-    const atRiskHubExpander: HTMLElement = screen.getByTestId("at-risk-hub-expander");
-    expect(atRiskHubExpander).toBeVisible();
-  });
+    it("should show pregnancy hub content ", () => {
+      const pregnancyHubContent: HTMLElement = screen.getByTestId("pregnancy-hub-content");
 
-  it("should show pregnancy hub content ", () => {
-    const pregnancyHubContent: HTMLElement = screen.getByTestId("pregnancy-hub-content");
+      expect(pregnancyHubContent).toBeVisible();
+    });
 
-    expect(pregnancyHubContent).toBeVisible();
+    it("renders vaccines for all ages button", async () => {
+      expectLinkToBeRendered("View vaccines for all ages", "/vaccines-for-all-ages");
+    });
   });
 
-  it("renders vaccines for all ages button", async () => {
-    expectLinkToBeRendered("View vaccines for all ages", "/vaccines-for-all-ages");
+  describe("pregnancy hub content", () => {
+    it.each([
+      { description: "hide", ageGroup: AgeGroup.AGE_65_to_74, shouldShowPregnancyContent: false },
+      { description: "show", ageGroup: AgeGroup.UNKNOWN_AGE_GROUP, shouldShowPregnancyContent: true },
+    ])(`$ageGroup should $description pregnancy content`, async ({ ageGroup, shouldShowPregnancyContent }) => {
+      (auth as jest.Mock).mockResolvedValue(mockSessionDataForAgeGroup(ageGroup));
+
+      render(await VaccinationsHub());
+
+      const pregnancyHubContent: HTMLElement | null = screen.queryByTestId("pregnancy-hub-content");
+
+      shouldShowPregnancyContent ? expect(pregnancyHubContent).toBeVisible() : expect(pregnancyHubContent).toBeNull();
+    });
   });
 });
 

src/app/check-and-book-vaccinations/page.tsx

@@ -7,7 +7,7 @@ import { PregnancyHubContent } from "@src/app/_components/hub/PregnancyHubConten
 import BackToNHSAppLink from "@src/app/_components/nhs-app/BackToNHSAppLink";
 import MainContent from "@src/app/_components/nhs-frontend/MainContent";
 import { NHS_TITLE_SUFFIX, SERVICE_HEADING } from "@src/app/constants";
-import { AgeGroup } from "@src/models/ageBasedHub";
+import { AgeBasedHubDetails, AgeBasedHubInfo, AgeGroup } from "@src/models/ageBasedHub";
 import { Session } from "next-auth";
 import Link from "next/link";
 import React from "react";
@@ -17,6 +17,7 @@ const VaccinationsHub = async () => {
 
   const session: Session | null = await auth();
   const ageGroup: AgeGroup = session?.user.age_group as AgeGroup;
+  const hubInfoForAgeGroup: AgeBasedHubDetails | undefined = AgeBasedHubInfo[ageGroup];
 
   return (
     <>
@@ -25,9 +26,8 @@ const VaccinationsHub = async () => {
       <MainContent>
         <h1 className={"nhsuk-heading-xl nhsuk-u-margin-bottom-3"}>{SERVICE_HEADING}</h1>
         <AgeBasedHubCards ageGroup={ageGroup} />
-        {/*TODO VIA-630 control visibility of the following components based on age*/}
         <AtRiskHubExpander />
-        <PregnancyHubContent />
+        {(hubInfoForAgeGroup?.showPregnancyHubContent || hubInfoForAgeGroup === undefined) && <PregnancyHubContent />}
         <Link href={"/vaccines-for-all-ages"} className={"nhsuk-button nhsuk-button--secondary"}>
           View vaccines for all ages
         </Link>

src/models/ageBasedHub.ts

@@ -18,6 +18,7 @@ type AgeSpecificVaccineCardDetails = {
 export type AgeBasedHubDetails = {
   heading: string;
   vaccines: AgeSpecificVaccineCardDetails[];
+  showPregnancyHubContent: boolean;
 };
 
 // TODO: VIA-630 remove 'undefined' from the type definition below after all hubs are implemented?
@@ -32,6 +33,7 @@ const AgeBasedHubInfo: Record<AgeGroup, AgeBasedHubDetails | undefined> = {
       { vaccineName: VaccineType.FLU_FOR_ADULTS, cardLinkDescription: "65 years and over" },
       { vaccineName: VaccineType.SHINGLES, cardLinkDescription: "65 to 67 and 70 to 79 years" },
     ],
+    showPregnancyHubContent: false,
   },
   AGE_75_to_80: undefined,
   AGE_81_PLUS: undefined,