diff --git a/test/loaders/ApiBasedRepositoryLoaderTests.test.ts b/test/loaders/ApiBasedRepositoryLoaderTests.test.ts index 90cb41f..ea34042 100644 --- a/test/loaders/ApiBasedRepositoryLoaderTests.test.ts +++ b/test/loaders/ApiBasedRepositoryLoaderTests.test.ts @@ -9,6 +9,9 @@ import { } from "../jsonSnapshots/fetchStopAndPolylineData/fetchStopAndPolylineDataSuccessfulResponse"; import { generateMockStops, generateMockSystems } from "../generators"; import { IStop } from "../../src/entities/entities"; +import { + fetchShuttleDataSuccessfulResponse +} from "../jsonSnapshots/fetchShuttleData/fetchShuttleDataSuccessfulResponse"; /** * Function to update behavior of the global `fetch` function. @@ -197,7 +200,14 @@ describe("ApiBasedRepositoryLoader", () => { describe("fetchAndUpdateShuttleDataForSystemId", () => { it("updates shuttle data in repository if response received", async () => { + updateGlobalFetchMockJson(fetchShuttleDataSuccessfulResponse); + const busesInResponse = Object.values(fetchShuttleDataSuccessfulResponse.buses); + await loader.fetchAndUpdateShuttleDataForSystemId("263"); + + const shuttles = await loader.repository.getShuttlesBySystemId("263"); + + expect(shuttles.length).toEqual(busesInResponse.length); }); it("throws the correct error if the API response contains no data", async () => {