use server context method for other resolver tests

This commit is contained in:
2025-01-29 15:26:23 -08:00
parent 7586ebba04
commit 0344e7b99b
5 changed files with 91 additions and 127 deletions

View File

@@ -1,32 +1,23 @@
import { beforeEach, describe, expect, it } from "@jest/globals";
import { ApolloServer } from "@apollo/server";
import { ServerContext } from "../../src/ServerContext";
import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { setupTestServerContext } from "../testHelpers/apolloSetupHelpers";
import { generateMockEtas, generateMockOrderedStops } from "../generators";
import { IStop, ISystem } from "../../src/entities/entities";
import { addMockStopToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import assert = require("node:assert");
describe("StopResolvers", () => {
let testServer: ApolloServer<ServerContext>;
let repository: UnoptimizedInMemoryRepository;
beforeEach(() => {
testServer = setUpTestServer();
repository = new UnoptimizedInMemoryRepository();
});
const context = setupTestServerContext();
let mockStop: IStop;
let mockSystem: ISystem;
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
mockStop = await addMockStopToRepository(repository, mockSystem.id);
mockSystem = await addMockSystemToRepository(context.repository);
mockStop = await addMockStopToRepository(context.repository, mockSystem.id);
})
async function getResponseForQuery(query: string) {
return await testServer.executeOperation({
return await context.testServer.executeOperation({
query,
variables: {
systemId: mockSystem.id,
@@ -34,7 +25,7 @@ describe("StopResolvers", () => {
},
}, {
contextValue: {
repository,
repository: context.repository,
}
});
}
@@ -59,7 +50,7 @@ describe("StopResolvers", () => {
mockOrderedStops = mockOrderedStops.filter((orderedStop) => orderedStop.stopId === mockOrderedStops[0].stopId);
await Promise.all(mockOrderedStops.map(async orderedStop => {
orderedStop.stopId = mockStop.id;
await repository.addOrUpdateOrderedStop(orderedStop);
await context.repository.addOrUpdateOrderedStop(orderedStop);
}));
const response = await getResponseForQuery(query);
@@ -96,7 +87,7 @@ describe("StopResolvers", () => {
mockEtas = mockEtas.filter((eta) => eta.stopId === mockEtas[0].stopId);
await Promise.all(mockEtas.map(async eta => {
eta.stopId = mockStop.id;
await repository.addOrUpdateEta(eta);
await context.repository.addOrUpdateEta(eta);
}));
const response = await getResponseForQuery(query);