extract mock system add to repository setup helpers file

This commit is contained in:
2025-01-28 15:33:38 -08:00
parent d03a99ee06
commit 4b84527fa4
3 changed files with 22 additions and 8 deletions

View File

@@ -3,6 +3,8 @@ import { ApolloServer } from "@apollo/server";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository"; import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers"; import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { ServerContext } from "../../src/ServerContext"; import { ServerContext } from "../../src/ServerContext";
import { ISystem } from "../../src/entities/entities";
import { addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
describe("EtaResolvers", () => { describe("EtaResolvers", () => {
let testServer: ApolloServer<ServerContext>; let testServer: ApolloServer<ServerContext>;
@@ -13,6 +15,12 @@ describe("EtaResolvers", () => {
repository = new UnoptimizedInMemoryRepository(); repository = new UnoptimizedInMemoryRepository();
}); });
let mockSystem: ISystem;
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
})
describe("stop", () => { describe("stop", () => {
const query = ` const query = `
query GetETAStop($systemId: ID!, $shuttleId: ID!) { query GetETAStop($systemId: ID!, $shuttleId: ID!) {

View File

@@ -6,14 +6,8 @@ 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";
async function addMockSystemToRepository(mockSystem: ISystem, repository: UnoptimizedInMemoryRepository) {
const mockSystems = generateMockSystems();
mockSystem = mockSystems[0];
mockSystem.id = "1";
await repository.addOrUpdateSystem(mockSystem);
return mockSystem;
}
describe("ShuttleResolvers", () => { describe("ShuttleResolvers", () => {
let testServer: ApolloServer<ServerContext> let testServer: ApolloServer<ServerContext>
@@ -28,7 +22,7 @@ describe("ShuttleResolvers", () => {
let mockShuttle: IShuttle; let mockShuttle: IShuttle;
beforeEach(async () => { beforeEach(async () => {
mockSystem = await addMockSystemToRepository(mockSystem, repository); mockSystem = await addMockSystemToRepository(repository);
const mockShuttles = generateMockShuttles(); const mockShuttles = generateMockShuttles();
mockShuttle = mockShuttles[0]; mockShuttle = mockShuttles[0];

View File

@@ -0,0 +1,12 @@
import { ISystem } from "../../src/entities/entities";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { 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;
}