diff --git a/src/repository.ts b/src/repository.ts index 657c43b..b8dfb77 100644 --- a/src/repository.ts +++ b/src/repository.ts @@ -41,8 +41,8 @@ export interface IEta { } export interface IOrderedStop { - nextStopId?: string; - previousStopId?: string; + nextStop?: IOrderedStop, + previousStop?: IOrderedStop, routeId: string; stopId: string; } diff --git a/src/testData.ts b/src/testData.ts index eb14cd1..ff6626d 100644 --- a/src/testData.ts +++ b/src/testData.ts @@ -46,37 +46,32 @@ const stops: IStop[] = [ } ]; -// const orderedStopsForRedRoute: IOrderedStop[] = [ -// { -// routeId: routes[0].id, -// stopId: stops[0].id -// }, -// { -// route: routes[0], -// stop: stops[1], -// }, -// ]; -// -// const orderedStopsForTealRoute: OrderedStop[] = [ -// { -// route: routes[1], -// stop: stops[1], -// }, -// { -// route: routes[1], -// stop: stops[0], -// } -// ] +const orderedStopsForRedRoute: IOrderedStop[] = [ + { + routeId: routes[0].id, + stopId: stops[0].id, + }, + { + routeId: routes[0].id, + stopId: stops[1].id, + }, +]; -// orderedStopsForRedRoute[0].nextStop = orderedStopsForRedRoute[1]; -// orderedStopsForRedRoute[1].previousStop = orderedStopsForRedRoute[0]; -// orderedStopsForTealRoute[0].nextStop = orderedStopsForTealRoute[1]; -// orderedStopsForTealRoute[1].previousStop = orderedStopsForTealRoute[0]; +const orderedStopsForTealRoute: IOrderedStop[] = [ + { + routeId: routes[1].id, + stopId: stops[1].id, + }, + { + routeId: routes[1].id, + stopId: stops[0].id, + }, +] -// stops[0].orderedStops = [orderedStopsForRedRoute[0], orderedStopsForTealRoute[1]]; -// stops[1].orderedStops = [orderedStopsForRedRoute[1], orderedStopsForTealRoute[0]]; -// routes[0].orderedStops = orderedStopsForRedRoute; -// routes[1].orderedStops = orderedStopsForTealRoute; +orderedStopsForRedRoute[0].nextStop = orderedStopsForRedRoute[1]; +orderedStopsForRedRoute[1].previousStop = orderedStopsForRedRoute[0]; +orderedStopsForTealRoute[0].nextStop = orderedStopsForTealRoute[1]; +orderedStopsForTealRoute[1].previousStop = orderedStopsForTealRoute[0]; const shuttles: IShuttle[] = [ { @@ -129,4 +124,10 @@ export async function loadTestData(repository: Repository) { await Promise.all(stops.map(async (stop) => { await repository.addOrUpdateStop(stop); })); + await Promise.all(orderedStopsForRedRoute.map(async (orderedStop) => { + await repository.addOrUpdateOrderedStop(orderedStop); + })); + await Promise.all(orderedStopsForTealRoute.map(async (orderedStop) => { + await repository.addOrUpdateOrderedStop(orderedStop); + })); }