From 9c35b03a5b960a8f06062ce0d285b7013d02f7d7 Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Tue, 28 Jan 2025 15:47:06 -0800
Subject: [PATCH] update test data for correctness

---
 test/resolvers/EtaResolverTests.test.ts | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/test/resolvers/EtaResolverTests.test.ts b/test/resolvers/EtaResolverTests.test.ts
index 7986a87..daf5fef 100644
--- a/test/resolvers/EtaResolverTests.test.ts
+++ b/test/resolvers/EtaResolverTests.test.ts
@@ -48,15 +48,13 @@ describe("EtaResolvers", () => {
       const stops = generateMockStops();
       const stop = stops[0];
       stop.systemId = mockSystem.id;
+      await repository.addOrUpdateStop(stop);
 
       const etas = generateMockEtas();
-      await Promise.all(etas.map(async (eta) => {
-        eta.stopId = stop.id;
-        await repository.addOrUpdateEta(eta);
-      }));
       const expectedEta = etas[0];
-
-      await repository.addOrUpdateStop(stop);
+      expectedEta.stopId = stop.id;
+      expectedEta.shuttleId = mockShuttle.id;
+      await repository.addOrUpdateEta(expectedEta);
 
       const response = await testServer.executeOperation({
         query,