import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { generateMockRoutes, generateMockStops, generateMockSystems } from "../generators";

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