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

@@ -4,7 +4,7 @@ import { UnoptimizedInMemoryRepository } from "../../src/repositories/Unoptimize
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { ServerContext } from "../../src/ServerContext";
import { IEta, IShuttle, IStop, ISystem } from "../../src/entities/entities";
import { addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import { addMockStopToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import { generateMockEtas, generateMockShuttles, generateMockStops } from "../generators";
import assert = require("node:assert");
@@ -30,10 +30,7 @@ describe("EtaResolvers", () => {
mockShuttle.systemId = mockSystem.id;
await repository.addOrUpdateShuttle(mockShuttle);
const mockStops = generateMockStops();
mockStop = mockStops[0];
mockStop.systemId = mockSystem.id;
await repository.addOrUpdateStop(mockStop);
mockStop = await addMockStopToRepository(repository, mockSystem.id);
const etas = generateMockEtas();
expectedEta = etas[0];

View File

@@ -12,7 +12,7 @@ import {
generateMockSystems
} from "../generators";
import assert = require("node:assert");
import { addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import { addMockRouteToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
describe("OrderedStopResolvers", () => {
let testServer: ApolloServer<ServerContext>;
@@ -29,14 +29,9 @@ describe("OrderedStopResolvers", () => {
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
const mockRoutes = generateMockRoutes();
mockRoute = mockRoutes[0];
mockRoute.id = "1";
await repository.addOrUpdateRoute(mockRoute);
mockRoute = await addMockRouteToRepository(repository, mockSystem.id);
mockStops = generateMockStops();
await Promise.all(mockStops.map(async (mockStop) => {
await repository.addOrUpdateStop(mockStop);
}));

View File

@@ -3,7 +3,11 @@ import { ServerContext } from "../../src/ServerContext";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { ApolloServer } from "@apollo/server";
import { addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import {
addMockRouteToRepository,
addMockStopToRepository,
addMockSystemToRepository
} from "../testHelpers/repositorySetupHelpers";
import { generateMockOrderedStops, generateMockRoutes, generateMockShuttles, generateMockStops } from "../generators";
import { IRoute, IStop, ISystem } from "../../src/entities/entities";
import assert = require("node:assert");
@@ -23,16 +27,10 @@ describe("RouteResolvers", () => {
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
const systemId = mockSystem.id;
const mockRoutes = generateMockRoutes();
mockRoute = mockRoutes[0];
mockRoute.systemId = mockSystem.id;
await repository.addOrUpdateRoute(mockRoute);
const mockStops = generateMockStops();
mockStop = mockStops[0];
mockStop.systemId = mockSystem.id;
await repository.addOrUpdateStop(mockStop);
mockRoute = await addMockRouteToRepository(repository, systemId);
mockStop = await addMockStopToRepository(repository, systemId);
});

View File

@@ -6,7 +6,7 @@ import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { generateMockEtas, generateMockOrderedStops, generateMockStops, generateMockSystems } from "../generators";
import { IStop, ISystem } from "../../src/entities/entities";
import assert = require("node:assert");
import { addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import { addMockStopToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
describe("StopResolvers", () => {
let testServer: ApolloServer<ServerContext>;
@@ -22,11 +22,7 @@ describe("StopResolvers", () => {
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
const mockStops = generateMockStops();
mockStop = mockStops[0];
mockStop.id = "1";
await repository.addOrUpdateStop(mockStop);
mockStop = await addMockStopToRepository(repository, mockSystem.id);
})
async function getResponseForQuery(query: string) {