diff --git a/test/loaders/ApiBasedRepositoryLoaderTests.test.ts b/test/loaders/ApiBasedRepositoryLoaderTests.test.ts index ea34042..ed4b92d 100644 --- a/test/loaders/ApiBasedRepositoryLoaderTests.test.ts +++ b/test/loaders/ApiBasedRepositoryLoaderTests.test.ts @@ -12,6 +12,7 @@ import { IStop } from "../../src/entities/entities"; import { fetchShuttleDataSuccessfulResponse } from "../jsonSnapshots/fetchShuttleData/fetchShuttleDataSuccessfulResponse"; +import { fetchEtaDataSuccessfulResponse } from "../jsonSnapshots/fetchEtaData/fetchEtaDataSuccessfulResponse"; /** * Function to update behavior of the global `fetch` function. @@ -255,7 +256,15 @@ describe("ApiBasedRepositoryLoader", () => { describe("fetchAndUpdateEtaDataForStopId", () => { it("updates ETA data for stop id if response received", async () => { + updateGlobalFetchMockJson(fetchEtaDataSuccessfulResponse); + const stopId = "177666"; + // @ts-ignore + const etasFromResponse = fetchEtaDataSuccessfulResponse.ETAs[stopId] + await loader.fetchAndUpdateEtaDataForStopId(stopId); + + const etas = await loader.repository.getEtasForStopId(stopId); + expect(etas.length).toEqual(etasFromResponse.length); }); it("throws the correct error if the API response contains no data", async () => {