Update ETA mock helper to take an ExternalSourceETARepository

This commit is contained in:
2025-11-11 15:37:07 -08:00
parent 1d366bd63d
commit 2854912e07
2 changed files with 5 additions and 3 deletions

View File

@@ -6,7 +6,8 @@ import {
addMockShuttleToRepository, addMockShuttleToRepository,
addMockStopToRepository, addMockStopToRepository,
} from "../../../testHelpers/repositorySetupHelpers"; } from "../../../testHelpers/repositorySetupHelpers";
import assert = require("node:assert"); import { ExternalSourceETARepository } from "../../repositories/shuttle/eta/ExternalSourceETARepository";
import assert from "node:assert";
describe("EtaResolvers", () => { describe("EtaResolvers", () => {
const holder = setupTestServerHolder(); const holder = setupTestServerHolder();
@@ -19,7 +20,7 @@ describe("EtaResolvers", () => {
beforeEach(async () => { beforeEach(async () => {
mockShuttle = await addMockShuttleToRepository(context.systems[0].shuttleRepository, context.systems[0].id); mockShuttle = await addMockShuttleToRepository(context.systems[0].shuttleRepository, context.systems[0].id);
mockStop = await addMockStopToRepository(context.systems[0].shuttleRepository, context.systems[0].id); mockStop = await addMockStopToRepository(context.systems[0].shuttleRepository, context.systems[0].id);
expectedEta = await addMockEtaToRepository(context.systems[0].shuttleRepository, mockStop.id, mockShuttle.id); expectedEta = await addMockEtaToRepository(context.systems[0].etaRepository as ExternalSourceETARepository, mockStop.id, mockShuttle.id);
}); });
async function getResponseForEtaQuery(query: string) { async function getResponseForEtaQuery(query: string) {

View File

@@ -5,6 +5,7 @@ import {
generateMockStops, generateMockStops,
} from "./mockDataGenerators"; } from "./mockDataGenerators";
import { ShuttleGetterSetterRepository } from "../src/repositories/shuttle/ShuttleGetterSetterRepository"; import { ShuttleGetterSetterRepository } from "../src/repositories/shuttle/ShuttleGetterSetterRepository";
import { ExternalSourceETARepository } from "../src/repositories/shuttle/eta/ExternalSourceETARepository";
export async function addMockRouteToRepository(repository: ShuttleGetterSetterRepository, systemId: string) { export async function addMockRouteToRepository(repository: ShuttleGetterSetterRepository, systemId: string) {
const mockRoutes = generateMockRoutes(); const mockRoutes = generateMockRoutes();
@@ -32,7 +33,7 @@ export async function addMockShuttleToRepository(repository: ShuttleGetterSetter
return mockShuttle; return mockShuttle;
} }
export async function addMockEtaToRepository(repository: ShuttleGetterSetterRepository, stopId: string, shuttleId: string) { export async function addMockEtaToRepository(repository: ExternalSourceETARepository, stopId: string, shuttleId: string) {
const etas = generateMockEtas(); const etas = generateMockEtas();
const expectedEta = etas[0]; const expectedEta = etas[0];
expectedEta.stopId = stopId; expectedEta.stopId = stopId;