Now using overrideOnce instead of a custom context provider

kiesraad · Jul 25, 2024 · 4c0aded · 4c0aded
1 parent f07cc09
commit 4c0aded
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 9 deletions.
diff --git a/frontend/app/component/form/polling_station_choice/PollingStationChoiceForm.test.tsx b/frontend/app/component/form/polling_station_choice/PollingStationChoiceForm.test.tsx
@@ -2,12 +2,14 @@ import { userEvent } from "@testing-library/user-event";
 import { describe, expect, test } from "vitest";
 
 import { PollingStationChoiceForm } from "app/component/form/polling_station_choice/PollingStationChoiceForm.tsx";
-import { render, screen, within } from "app/test/unit";
+import { overrideOnce, render, screen, within } from "app/test/unit";
 
-import { PollingStationProvider, PollingStationsContext } from "@kiesraad/api";
+import { PollingStationProvider } from "@kiesraad/api";
+import { pollingStationMock } from "@kiesraad/api-mocks";
 
 describe("Test PollingStationChoiceForm", () => {
   test("Form field entry", async () => {
+    overrideOnce("get", "/api/polling_stations/1", 200, pollingStationMock);
     const user = userEvent.setup();
 
     render(
@@ -35,6 +37,7 @@ describe("Test PollingStationChoiceForm", () => {
   });
 
   test("Selecting a valid polling station", async () => {
+    overrideOnce("get", "/api/polling_stations/1", 200, pollingStationMock);
     const user = userEvent.setup();
     render(
       <PollingStationProvider electionId={1}>
@@ -50,6 +53,7 @@ describe("Test PollingStationChoiceForm", () => {
   });
 
   test("Selecting a non-existing polling station", async () => {
+    overrideOnce("get", "/api/polling_stations/1", 200, pollingStationMock);
     const user = userEvent.setup();
     render(
       <PollingStationProvider electionId={1}>
@@ -67,6 +71,7 @@ describe("Test PollingStationChoiceForm", () => {
   });
 
   test("Polling station list", async () => {
+    overrideOnce("get", "/api/polling_stations/1", 200, pollingStationMock);
     const user = userEvent.setup();
 
     render(
@@ -90,17 +95,15 @@ describe("Test PollingStationChoiceForm", () => {
   });
 
   test("Polling station list no stations", async () => {
+    overrideOnce("get", "/api/polling_stations/1", 200, {
+      polling_stations: [],
+    });
     const user = userEvent.setup();
 
     render(
-      <PollingStationsContext.Provider
-        value={{
-          pollingStationsLoading: false,
-          pollingStations: [],
-        }}
-      >
+      <PollingStationProvider electionId={1}>
         <PollingStationChoiceForm />
-      </PollingStationsContext.Provider>,
+      </PollingStationProvider>,
     );
 
     const openPollingStationList = screen.getByTestId("openPollingStationList");

diff --git a/frontend/lib/api-mocks/index.ts b/frontend/lib/api-mocks/index.ts
@@ -7,6 +7,7 @@ import {
   PoliticalGroup,
   POLLING_STATION_DATA_ENTRY_REQUEST_BODY,
   POLLING_STATION_DATA_ENTRY_REQUEST_PARAMS,
+  PollingStationListResponse,
   VotersCounts,
   VotesCounts,
 } from "@kiesraad/api";
@@ -16,6 +17,7 @@ import { pollingStationMockData } from "./PollingStationMockData.ts";
 
 export const electionMock = electionMockData as Required<Election>;
 export const politicalGroupMock = politicalGroupMockData as Required<PoliticalGroup>;
+export const pollingStationMock = pollingStationMockData as Required<PollingStationListResponse>;
 
 type ParamsToString<T> = {
   [P in keyof T]: string;