add mock shuttle and mock eta methods

This commit is contained in:
2025-01-28 20:35:18 -08:00
parent 3fd8efdd8a
commit fe705ed8e3
3 changed files with 34 additions and 21 deletions

View File

@@ -1,5 +1,11 @@
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { generateMockRoutes, generateMockStops, generateMockSystems } from "../generators";
import {
generateMockEtas,
generateMockRoutes,
generateMockShuttles,
generateMockStops,
generateMockSystems
} from "../generators";
export async function addMockSystemToRepository(repository: UnoptimizedInMemoryRepository) {
const mockSystems = generateMockSystems();
@@ -28,3 +34,21 @@ export async function addMockStopToRepository(repository: UnoptimizedInMemoryRep
return mockStop;
}
export async function addMockShuttleToRepository(repository: UnoptimizedInMemoryRepository, systemId: string) {
const mockShuttles = generateMockShuttles();
const mockShuttle = mockShuttles[0];
mockShuttle.systemId = systemId;
await repository.addOrUpdateShuttle(mockShuttle);
return mockShuttle;
}
export async function addMockEtaToRepository(repository: UnoptimizedInMemoryRepository, stopId: string, shuttleId: string) {
const etas = generateMockEtas();
const expectedEta = etas[0];
expectedEta.stopId = stopId;
expectedEta.shuttleId = shuttleId;
await repository.addOrUpdateEta(expectedEta);
return expectedEta;
}