From 96846cbd88937b08666990dfa1d7a1c90dc18e63 Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Tue, 28 Jan 2025 15:25:09 -0800
Subject: [PATCH] add remaining test cases for ordered stop resolvers

---
 .../OrderedStopResolverTests.test.ts          | 43 +++++++++++++++----
 1 file changed, 34 insertions(+), 9 deletions(-)

diff --git a/test/resolvers/OrderedStopResolverTests.test.ts b/test/resolvers/OrderedStopResolverTests.test.ts
index e562b83..b9602f0 100644
--- a/test/resolvers/OrderedStopResolverTests.test.ts
+++ b/test/resolvers/OrderedStopResolverTests.test.ts
@@ -261,23 +261,48 @@ describe("OrderedStopResolvers", () => {
   });
 
   describe("stop", () => {
-    const query = `
-    query GetNextStop($systemId: ID!, $routeId: ID!, $stopId: ID!) {
-      system(id: $systemId) {
-        route(id: $routeId) {
-          orderedStop(forStopId: $stopId) {
-            stop {
-              name
-              id
+    async function getResponseForStopQuery(stopId: string) {
+      const query = `
+      query GetNextStop($systemId: ID!, $routeId: ID!, $stopId: ID!) {
+        system(id: $systemId) {
+          route(id: $routeId) {
+            orderedStop(forStopId: $stopId) {
+              stop {
+                name
+                id
+              }
             }
           }
         }
       }
+      `;
+
+      return await testServer.executeOperation({
+        query,
+        variables: {
+          systemId: mockSystem.id,
+          routeId: mockRoute.id,
+          stopId,
+        }
+      }, {
+        contextValue: {
+          repository,
+        }
+      });
     }
-    `;
 
     it("returns the associated stop if it exists", async () => {
+      const orderedStops = await setUpOrderedStopsInRepository();
+      orderedStops[0].stopId = mockStops[0].id;
+      await repository.addOrUpdateOrderedStop(orderedStops[0]);
 
+      const response = await getResponseForStopQuery(orderedStops[0].stopId);
+
+      assert(response.body.kind === "single");
+      expect(response.body.singleResult.errors).toBeUndefined();
+      const stop = (response.body.singleResult.data?.system as any).route.orderedStop.stop;
+      expect(stop.name).toEqual(mockStops[0].name);
+      expect(stop.id).toEqual(mockStops[0].id);
     });
   });
 });