diff --git a/test/resolvers/OrderedStopResolverTests.test.ts b/test/resolvers/OrderedStopResolverTests.test.ts index f0541ad..a75ac68 100644 --- a/test/resolvers/OrderedStopResolverTests.test.ts +++ b/test/resolvers/OrderedStopResolverTests.test.ts @@ -129,8 +129,26 @@ describe("OrderedStopResolvers", () => { expect(nonexistentNextStop).toBeNull(); }); - it("returns null if the current stop no longer exists", async () => { + it("returns null if the next stop object no longer exists", async () => { + const orderedStops = await setUpOrderedStopsInRepository(); + await repository.removeStopIfExists(orderedStops[1].stopId); + const response = await testServer.executeOperation({ + query, + variables: { + systemId: mockSystem.id, + routeId: mockRoute.id, + stopId: orderedStops[0].stopId, + }, + }, { + contextValue: { + repository, + } + }); + + assert(response.body.kind === "single"); + const nonexistentNextStop = (response.body.singleResult.data?.system as any).route.orderedStop.nextStop; + expect(nonexistentNextStop).toBeNull(); }); });