diff --git a/src/loaders/shuttle/__tests__/ApiBasedShuttleRepositoryLoaderTests.test.ts b/src/loaders/shuttle/__tests__/ApiBasedShuttleRepositoryLoaderTests.test.ts index f215b4c..ff98c40 100644 --- a/src/loaders/shuttle/__tests__/ApiBasedShuttleRepositoryLoaderTests.test.ts +++ b/src/loaders/shuttle/__tests__/ApiBasedShuttleRepositoryLoaderTests.test.ts @@ -9,6 +9,7 @@ import { generateMockRoutes, generateMockShuttles, generateMockStops } from "../ import { fetchShuttleDataSuccessfulResponse } from "../../../../testHelpers/jsonSnapshots/fetchShuttleData/fetchShuttleDataSuccessfulResponse"; +import { InMemoryExternalSourceETARepository } from "../../../repositories/shuttle/eta/InMemoryExternalSourceETARepository"; import { resetGlobalFetchMockJson, updateGlobalFetchMockJson, @@ -58,7 +59,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { const routesToPrune = generateMockRoutes(); await Promise.all(routesToPrune.map(async (route) => { route.systemId = systemId; - await loader.repository.addOrUpdateRoute(route); + await loader.shuttleRepository.addOrUpdateRoute(route); })); updateGlobalFetchMockJson(fetchRouteDataSuccessfulResponse); @@ -67,7 +68,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { await loader.updateRouteDataForSystem(); // Assert - const routes = await loader.repository.getRoutes(); + const routes = await loader.shuttleRepository.getRoutes(); expect(routes.length).toEqual(fetchRouteDataSuccessfulResponse.all.length) }); @@ -91,7 +92,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { const stopsToPrune = generateMockStops(); await Promise.all(stopsToPrune.map(async (stop) => { stop.systemId = systemId; - await loader.repository.addOrUpdateStop(stop); + await loader.shuttleRepository.addOrUpdateStop(stop); })); updateGlobalFetchMockJson(fetchStopAndPolylineDataSuccessfulResponse); @@ -100,15 +101,15 @@ describe("ApiBasedShuttleRepositoryLoader", () => { await loader.updateStopAndPolylineDataForRoutesInSystem(); - const stops = await loader.repository.getStops(); + const stops = await loader.shuttleRepository.getStops(); expect(stops.length).toEqual(stopsArray.length); await Promise.all(stops.map(async (stop) => { - const orderedStops = await loader.repository.getOrderedStopsByStopId(stop.id) + const orderedStops = await loader.shuttleRepository.getOrderedStopsByStopId(stop.id) expect(orderedStops.length).toBeGreaterThan(0); })); - const routes = await loader.repository.getRoutes(); + const routes = await loader.shuttleRepository.getRoutes(); routes.forEach((route) => { expect(route.polylineCoordinates.length).toBeGreaterThan(0); }); @@ -150,7 +151,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { async function addMockRoutes(routes: IRoute[]) { await Promise.all(routes.map(async (route) => { - await loader.repository.addOrUpdateRoute(route); + await loader.shuttleRepository.addOrUpdateRoute(route); })); } @@ -160,6 +161,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { "263", "1", new UnoptimizedInMemoryShuttleRepository(), + new InMemoryExternalSourceETARepository(), distanceMiles, ); @@ -175,7 +177,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { await loader.updateShuttleDataForSystemBasedOnProximityToRoutes(); - const shuttles = await loader.repository.getShuttles(); + const shuttles = await loader.shuttleRepository.getShuttles(); expect(shuttles.length).toEqual(busesInResponse.length); }); @@ -185,6 +187,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { "263", "1", new UnoptimizedInMemoryShuttleRepository(), + new InMemoryExternalSourceETARepository(), distanceMiles, ); @@ -200,7 +203,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { await loader.updateShuttleDataForSystemBasedOnProximityToRoutes(); - const shuttles = await loader.repository.getShuttles(); + const shuttles = await loader.shuttleRepository.getShuttles(); expect(shuttles.length).toEqual(0); }); @@ -210,6 +213,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { "263", "1", new UnoptimizedInMemoryShuttleRepository(), + new InMemoryExternalSourceETARepository(), distanceMiles, ); @@ -217,7 +221,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { const shuttlesToPrune = generateMockShuttles(); await Promise.all(shuttlesToPrune.map(async (shuttle) => { shuttle.systemId = systemId; - await loader.repository.addOrUpdateShuttle(shuttle); + await loader.shuttleRepository.addOrUpdateShuttle(shuttle); })); const routes = generateMockRoutesWithPolylineCoordinates(); @@ -235,7 +239,7 @@ describe("ApiBasedShuttleRepositoryLoader", () => { await loader.updateShuttleDataForSystemBasedOnProximityToRoutes(); // Old shuttles should be pruned, only API shuttles should remain - const shuttles = await loader.repository.getShuttles(); + const shuttles = await loader.shuttleRepository.getShuttles(); const busesInResponse = Object.values(modifiedSuccessfulResponse.buses); expect(shuttles.length).toEqual(busesInResponse.length);