update eta resolver tests to use new method

This commit is contained in:
2025-01-29 15:08:18 -08:00
parent fbd0c76b70
commit d69197acd8

View File

@@ -1,7 +1,7 @@
import { beforeEach, describe, expect, it } from "@jest/globals"; import { beforeEach, describe, expect, it } from "@jest/globals";
import { ApolloServer } from "@apollo/server"; import { ApolloServer } from "@apollo/server";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository"; import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers"; import { setUpTestServer, setupTestServerContext } 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 { import {
@@ -13,13 +13,7 @@ import { generateMockEtas, generateMockShuttles, generateMockStops } from "../ge
import assert = require("node:assert"); import assert = require("node:assert");
describe("EtaResolvers", () => { describe("EtaResolvers", () => {
let testServer: ApolloServer<ServerContext>; const context = setupTestServerContext();
let repository: UnoptimizedInMemoryRepository;
beforeEach(async () => {
testServer = setUpTestServer();
repository = new UnoptimizedInMemoryRepository();
});
let mockSystem: ISystem; let mockSystem: ISystem;
let mockShuttle: IShuttle; let mockShuttle: IShuttle;
@@ -27,14 +21,14 @@ describe("EtaResolvers", () => {
let expectedEta: IEta; let expectedEta: IEta;
beforeEach(async () => { beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository); mockSystem = await addMockSystemToRepository(context.repository);
mockShuttle = await addMockShuttleToRepository(repository, mockSystem.id); mockShuttle = await addMockShuttleToRepository(context.repository, mockSystem.id);
mockStop = await addMockStopToRepository(repository, mockSystem.id); mockStop = await addMockStopToRepository(context.repository, mockSystem.id);
expectedEta = await addMockEtaToRepository(repository, mockStop.id, mockShuttle.id); expectedEta = await addMockEtaToRepository(context.repository, mockStop.id, mockShuttle.id);
}); });
async function getResponseForEtaQuery(query: string) { async function getResponseForEtaQuery(query: string) {
const response = await testServer.executeOperation({ const response = await context.testServer.executeOperation({
query, query,
variables: { variables: {
systemId: mockSystem.id, systemId: mockSystem.id,
@@ -42,7 +36,7 @@ describe("EtaResolvers", () => {
}, },
}, { }, {
contextValue: { contextValue: {
repository, repository: context.repository,
} }
}); });
return response; return response;