add mock shuttle and mock eta methods

This commit is contained in:
2025-01-28 20:35:18 -08:00
parent 3fd8efdd8a
commit fe705ed8e3
3 changed files with 34 additions and 21 deletions

View File

@@ -4,7 +4,11 @@ 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 { addMockStopToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import {
addMockEtaToRepository, addMockShuttleToRepository,
addMockStopToRepository,
addMockSystemToRepository
} from "../testHelpers/repositorySetupHelpers";
import { generateMockEtas, generateMockShuttles, generateMockStops } from "../generators";
import assert = require("node:assert");
@@ -24,19 +28,9 @@ describe("EtaResolvers", () => {
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
const mockShuttles = generateMockShuttles();
mockShuttle = mockShuttles[0];
mockShuttle.systemId = mockSystem.id;
await repository.addOrUpdateShuttle(mockShuttle);
mockShuttle = await addMockShuttleToRepository(repository, mockSystem.id);
mockStop = await addMockStopToRepository(repository, mockSystem.id);
const etas = generateMockEtas();
expectedEta = etas[0];
expectedEta.stopId = mockStop.id;
expectedEta.shuttleId = mockShuttle.id;
await repository.addOrUpdateEta(expectedEta);
expectedEta = await addMockEtaToRepository(repository, mockStop.id, mockShuttle.id);
});
async function getResponseForEtaQuery(query: string) {