mirror of
https://github.com/brendan-ch/project-inter-server.git
synced 2026-04-17 07:50:31 +00:00
add mock shuttle and mock eta methods
This commit is contained in:
@@ -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;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user