From cf3f26e255036baddf1a8c69d4eb880c705e2a93 Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Tue, 28 Jan 2025 14:54:35 -0800
Subject: [PATCH] refactor querying call into a separate method

---
 .../OrderedStopResolverTests.test.ts          | 47 ++++++-------------
 1 file changed, 15 insertions(+), 32 deletions(-)

diff --git a/test/resolvers/OrderedStopResolverTests.test.ts b/test/resolvers/OrderedStopResolverTests.test.ts
index a75ac68..9041895 100644
--- a/test/resolvers/OrderedStopResolverTests.test.ts
+++ b/test/resolvers/OrderedStopResolverTests.test.ts
@@ -79,23 +79,28 @@ describe("OrderedStopResolvers", () => {
     }
     `;
 
-    it("returns the next stop if it exists", async () => {
-      // Arrange
-      const orderedStops = await setUpOrderedStopsInRepository();
-
-      // Act
-      const response = await testServer.executeOperation({
+    async function getResponseForQuery(stopId: string) {
+      return await testServer.executeOperation({
         query,
         variables: {
           systemId: mockSystem.id,
           routeId: mockRoute.id,
-          stopId: orderedStops[0].stopId,
-        }
+          stopId,
+        },
       }, {
         contextValue: {
           repository,
         }
       });
+    }
+
+
+    it("returns the next stop if it exists", async () => {
+      // Arrange
+      const orderedStops = await setUpOrderedStopsInRepository();
+
+      // Act
+      const response = await getResponseForQuery(orderedStops[0].stopId);
 
       // Assert
       assert(response.body.kind === "single");
@@ -111,18 +116,7 @@ describe("OrderedStopResolvers", () => {
       orderedStops[0].nextStop = undefined;
       await repository.addOrUpdateOrderedStop(orderedStops[0]);
 
-      const response = await testServer.executeOperation({
-        query,
-        variables: {
-          systemId: mockSystem.id,
-          routeId: mockRoute.id,
-          stopId: orderedStops[0].stopId,
-        },
-      }, {
-        contextValue: {
-          repository,
-        }
-      });
+      const response = await getResponseForQuery(orderedStops[0].stopId);
 
       assert(response.body.kind === "single");
       const nonexistentNextStop = (response.body.singleResult.data?.system as any).route.orderedStop.nextStop;
@@ -133,18 +127,7 @@ describe("OrderedStopResolvers", () => {
       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,
-        }
-      });
+      const response = await getResponseForQuery(orderedStops[0].stopId);
 
       assert(response.body.kind === "single");
       const nonexistentNextStop = (response.body.singleResult.data?.system as any).route.orderedStop.nextStop;