From 3aeaca68440337ee3d567d43003225a2b7eb49bc Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Tue, 28 Jan 2025 15:07:54 -0800
Subject: [PATCH] add test for associated route

---
 .../OrderedStopResolverTests.test.ts          | 59 ++++++++++++++-----
 1 file changed, 45 insertions(+), 14 deletions(-)

diff --git a/test/resolvers/OrderedStopResolverTests.test.ts b/test/resolvers/OrderedStopResolverTests.test.ts
index d724170..eb7140a 100644
--- a/test/resolvers/OrderedStopResolverTests.test.ts
+++ b/test/resolvers/OrderedStopResolverTests.test.ts
@@ -208,28 +208,59 @@ describe("OrderedStopResolvers", () => {
 
   describe("route", () => {
     // Note that there is no `orderedStop(forRouteId)` resolver,
-    // so fetch all ordered stops for a stop instead
-    const query = `
-    query GetNextStop($systemId: ID!, $stopId: ID!) {
-      system(id: $systemId) {
-        stop(id: $stopId) {
-          orderedStops {
-            route {
-              id
-              name
+    // so fetch all ordered stops for a stop instead.
+    // If we went through the route ID, it would've
+    // relied on the parent data within the route.orderedStop resolver
+    async function getResponseForRouteQuery(stopId: string) {
+      const query = `
+      query GetNextStop($systemId: ID!, $stopId: ID!) {
+        system(id: $systemId) {
+          stop(id: $stopId) {
+            orderedStops {
+              route {
+                id
+                name
+              }
             }
           }
         }
       }
+      `;
+
+      return await testServer.executeOperation({
+        query,
+        variables: {
+          systemId: mockSystem.id,
+          stopId,
+        }
+      }, {
+        contextValue: {
+          repository,
+        }
+      });
     }
-    `;
-  });
 
-  it("returns the associated route if it exists", async () => {
+    it("returns the associated route if it exists", async () => {
+      const orderedStops = generateMockOrderedStops();
+      orderedStops[0].routeId = mockRoute.id;
+      orderedStops[0].stopId = mockStops[0].id;
 
-  });
+      // Add one stop only
+      await repository.addOrUpdateOrderedStop(orderedStops[0]);
 
-  it("returns null if the route doesn't exist", async () => {
+      const response = await getResponseForRouteQuery(orderedStops[1].stopId);
+
+      assert(response.body.kind === "single");
+      expect(response.body.singleResult.errors).toBeUndefined();
+      const route = (response.body.singleResult.data?.system as any).stop.orderedStops[0].route
+
+      expect(route.id).toEqual(mockRoute.id);
+      expect(route.name).toEqual(mockRoute.name);
+    });
+
+    it("returns null if the route doesn't exist", async () => {
+
+    });
 
   });