import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import {
  generateMockEtas,
  generateMockRoutes,
  generateMockShuttles,
  generateMockStops,
  generateMockSystems
} from "./mockDataGenerators";

export async function addMockSystemToRepository(repository: UnoptimizedInMemoryRepository) {
  const mockSystems = generateMockSystems();
  const mockSystem = mockSystems[0];
  mockSystem.id = "1";
  await repository.addOrUpdateSystem(mockSystem);

  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;
}

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;
}