From d423cdb95b25119606d96a88dd074464e9c1bf96 Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Tue, 28 Jan 2025 14:52:10 -0800
Subject: [PATCH] add test case for if next stop object does not exist

---
 .../OrderedStopResolverTests.test.ts          | 20 ++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

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();
     });
   });