project-inter-server/testHelpers/repositorySetupHelpers.ts

import {
  generateMockEtas,
  generateMockRoutes,
  generateMockShuttles,
  generateMockStops,
} from "./mockDataGenerators";
import { ShuttleGetterSetterRepository } from "../src/repositories/shuttle/ShuttleGetterSetterRepository";
import { ExternalSourceETARepository } from "../src/repositories/shuttle/eta/ExternalSourceETARepository";

export async function addMockRouteToRepository(repository: ShuttleGetterSetterRepository, systemId: string) {
  const mockRoutes = generateMockRoutes();
  const mockRoute = mockRoutes[0];
  mockRoute.systemId = systemId;
  await repository.addOrUpdateRoute(mockRoute);

  return mockRoute;
}

export async function addMockStopToRepository(repository: ShuttleGetterSetterRepository, systemId: string) {
  const mockStops = generateMockStops();
  const mockStop = mockStops[0];
  mockStop.systemId = systemId;
  await repository.addOrUpdateStop(mockStop);

  return mockStop;
}

export async function addMockShuttleToRepository(repository: ShuttleGetterSetterRepository, systemId: string) {
  const mockShuttles = generateMockShuttles();
  const mockShuttle = mockShuttles[0];
  mockShuttle.systemId = systemId;
  await repository.addOrUpdateShuttle(mockShuttle);
  return mockShuttle;
}

export async function addMockEtaToRepository(repository: ExternalSourceETARepository, stopId: string, shuttleId: string) {
  const etas = generateMockEtas();
  const expectedEta = etas[0];
  expectedEta.stopId = stopId;
  expectedEta.shuttleId = shuttleId;
  await repository.addOrUpdateEtaFromExternalSource(expectedEta);

  return expectedEta;
}