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,29 +1,21 @@
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 { generateMockEtas, generateMockRoutes } from "../generators";
import { IShuttle, ISystem } from "../../src/entities/entities";
import { setUpTestServer } from "../testHelpers/apolloSetupHelpers";
import { setupTestServerContext } from "../testHelpers/apolloSetupHelpers";
import { addMockShuttleToRepository, addMockSystemToRepository } from "../testHelpers/repositorySetupHelpers";
import assert = require("node:assert");
describe("ShuttleResolvers", () => {
let testServer: ApolloServer<ServerContext>
let repository: UnoptimizedInMemoryRepository;
beforeEach(async () => {
testServer = setUpTestServer();
repository = new UnoptimizedInMemoryRepository();
});
const context = setupTestServerContext();
let mockSystem: ISystem;
let mockShuttle: IShuttle;
beforeEach(async () => {
mockSystem = await addMockSystemToRepository(repository);
mockShuttle = await addMockShuttleToRepository(repository, mockSystem.id);
mockSystem = await addMockSystemToRepository(context.repository);
mockShuttle = await addMockShuttleToRepository(context.repository,
mockSystem.id);
});
@@ -31,7 +23,7 @@ describe("ShuttleResolvers", () => {
const etas = generateMockEtas();
await Promise.all(etas.map(async (eta) => {
eta.shuttleId = shuttleId;
await repository.addOrUpdateEta(eta);
await context.repository.addOrUpdateEta(eta);
}));
return etas;
}
@@ -55,7 +47,7 @@ describe("ShuttleResolvers", () => {
const mockEta = etas[1];
// Act
const response = await testServer.executeOperation({
const response = await context.testServer.executeOperation({
query,
variables: {
systemId: mockSystem.id,
@@ -64,18 +56,19 @@ describe("ShuttleResolvers", () => {
},
}, {
contextValue: {
repository,
repository: context.repository,
},
});
// Assert
assert(response.body.kind === "single");
expect(response.body.singleResult.errors).toBeUndefined();
expect((response.body.singleResult.data as any).system.shuttle.eta.secondsRemaining).toEqual(mockEta.secondsRemaining);
expect((response.body.singleResult.data as
any).system.shuttle.eta.secondsRemaining).toEqual(mockEta.secondsRemaining);
});
it("returns null if it doesn't exist", async () => {
const response = await testServer.executeOperation({
const response = await context.testServer.executeOperation({
query,
variables: {
systemId: mockSystem.id,
@@ -84,14 +77,15 @@ describe("ShuttleResolvers", () => {
}
}, {
contextValue: {
repository,
repository: context.repository,
}
});
// Assert
assert(response.body.kind === "single");
expect(response.body.singleResult.errors).toBeUndefined();
expect((response.body.singleResult.data as any).system.shuttle.eta).toBeNull();
expect((response.body.singleResult.data as
any).system.shuttle.eta).toBeNull();
});
});
@@ -112,7 +106,7 @@ describe("ShuttleResolvers", () => {
it("returns associated ETAs if they exist for the shuttle", async () => {
const etas = await addMockEtas(mockShuttle.id);
const response = await testServer.executeOperation({
const response = await context.testServer.executeOperation({
query,
variables: {
systemId: mockSystem.id,
@@ -120,17 +114,18 @@ describe("ShuttleResolvers", () => {
},
}, {
contextValue: {
repository,
repository: context.repository,
}
});
assert(response.body.kind === "single");
expect(response.body.singleResult.errors).toBeUndefined();
expect((response.body.singleResult.data as any).system.shuttle.etas).toHaveLength(etas.length);
expect((response.body.singleResult.data as
any).system.shuttle.etas).toHaveLength(etas.length);
});
it("returns empty array if no ETAs exist", async () => {
const response = await testServer.executeOperation({
const response = await context.testServer.executeOperation({
query,
variables: {
systemId: mockSystem.id,
@@ -138,17 +133,17 @@ describe("ShuttleResolvers", () => {
},
}, {
contextValue: {
repository,
repository: context.repository,
}
});
assert(response.body.kind === "single");
expect(response.body.singleResult.errors).toBeUndefined();
expect((response.body.singleResult.data as any).system.shuttle.etas).toHaveLength(0);
expect((response.body.singleResult.data as
any).system.shuttle.etas).toHaveLength(0);
});
});
describe("route", () => {
const query = `
query GetShuttleRoute($systemId: ID!, $shuttleId: ID!) {
@@ -169,7 +164,7 @@ describe("ShuttleResolvers", () => {
`
async function getResponseForQuery() {
return await testServer.executeOperation({
return await context.testServer.executeOperation({
query,
variables: {
systemId: mockSystem.id,
@@ -177,14 +172,14 @@ describe("ShuttleResolvers", () => {
}
}, {
contextValue: {
repository,
repository: context.repository,
}
});
}
it("returns the route if it exists", async () => {
const mockRoute = generateMockRoutes()[0];
await repository.addOrUpdateRoute(mockRoute);
await context.repository.addOrUpdateRoute(mockRoute);
const response = await getResponseForQuery();
@@ -200,5 +195,6 @@ describe("ShuttleResolvers", () => {
expect(response.body.singleResult.errors).toBeUndefined();
expect((response.body.singleResult.data as any).system.shuttle.route).toBeNull();
});
});
});