add addMockRouteToRepository and addMockStopToRepository methods

This commit is contained in:
2025-01-28 20:31:06 -08:00
parent 2d4f866308
commit 3fd8efdd8a
5 changed files with 34 additions and 30 deletions

View File

@@ -1,6 +1,5 @@
import { ISystem } from "../../src/entities/entities";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { generateMockSystems } from "../generators";
import { generateMockRoutes, generateMockStops, generateMockSystems } from "../generators";
export async function addMockSystemToRepository(repository: UnoptimizedInMemoryRepository) {
const mockSystems = generateMockSystems();
@@ -10,3 +9,22 @@ export async function addMockSystemToRepository(repository: UnoptimizedInMemoryR
return mockSystem;
}
export async function addMockRouteToRepository(repository: UnoptimizedInMemoryRepository, systemId: string) {
const mockRoutes = generateMockRoutes();
const mockRoute = mockRoutes[0];
mockRoute.systemId = systemId;
await repository.addOrUpdateRoute(mockRoute);
return mockRoute;
}
export async function addMockStopToRepository(repository: UnoptimizedInMemoryRepository, systemId: string) {
const mockStops = generateMockStops();
const mockStop = mockStops[0];
mockStop.systemId = systemId;
await repository.addOrUpdateStop(mockStop);
return mockStop;
}