brendan/project-inter-server
1
0
Fork 0
mirror of https://github.com/brendan-ch/project-inter-server.git synced 2026-04-17 07:50:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

add mock shuttle and mock eta methods

Browse Source
This commit is contained in:
Brendan Chen
2025-01-28 20:35:18 -08:00
parent 3fd8efdd8a
commit fe705ed8e3
3 changed files with 34 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
test/resolvers/EtaResolverTests.test.ts
View File

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

9
test/resolvers/ShuttleResolverTests.test.ts
View File

@@ -6,7 +6,7 @@ import { generateMockEtas, generateMockRoutes, generateMockShuttles, generateMoc
import { IShuttle, ISystem } from "../../src/entities/entities"; import { IShuttle, ISystem } from "../../src/entities/entities";
import assert = require("node:assert"); import assert = require("node:assert");
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers"; import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers"; import { addMockShuttleToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
describe("ShuttleResolvers", () => { describe("ShuttleResolvers", () => {
@@ -23,12 +23,7 @@ describe("ShuttleResolvers", () => {
beforeEach(async () => { beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository); mockSystem = await addMockSystemToRepository(repository);
mockShuttle = await addMockShuttleToRepository(repository, mockSystem.id);
const mockShuttles = generateMockShuttles();
mockShuttle = mockShuttles[0];
mockShuttle.systemId = mockSystem.id;
mockShuttle.id = "1";
await repository.addOrUpdateShuttle(mockShuttle);
}); });

26
test/testHelpers/repositorySetupHelpers.ts
View File

@@ -1,5 +1,11 @@
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository"; import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { generateMockRoutes, generateMockStops, generateMockSystems } from "../generators"; import {
generateMockEtas,
generateMockRoutes,
generateMockShuttles,
generateMockStops,
generateMockSystems
} from "../generators";
export async function addMockSystemToRepository(repository: UnoptimizedInMemoryRepository) { export async function addMockSystemToRepository(repository: UnoptimizedInMemoryRepository) {
const mockSystems = generateMockSystems(); const mockSystems = generateMockSystems();
@@ -28,3 +34,21 @@ export async function addMockStopToRepository(repository: UnoptimizedInMemoryRep
return 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;
}