From 1511b3c1184f8d80f30a66f79172d4326be4940e Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Fri, 14 Nov 2025 10:23:02 -0800
Subject: [PATCH] Add test case to check that last stop data is removed

---
 .../__tests__/ShuttleRepositorySharedTests.test.ts | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/repositories/shuttle/__tests__/ShuttleRepositorySharedTests.test.ts b/src/repositories/shuttle/__tests__/ShuttleRepositorySharedTests.test.ts
index 038ea56..f875f2e 100644
--- a/src/repositories/shuttle/__tests__/ShuttleRepositorySharedTests.test.ts
+++ b/src/repositories/shuttle/__tests__/ShuttleRepositorySharedTests.test.ts
@@ -386,6 +386,20 @@ describe.each(repositoryImplementations)('$name', (holder) => {
       expect(remainingShuttles).toHaveLength(mockShuttles.length - 1);
     });
 
+    test("removes shuttle last stop if exists", async () => {
+      const { route, stop1, systemId } = await setupRouteAndOrderedStops();
+      const mockShuttle = generateMockShuttles()[0];
+      mockShuttle.routeId = route.id;
+      mockShuttle.coordinates = stop1.coordinates;
+      mockShuttle.systemId = systemId;
+
+      await repository.addOrUpdateShuttle(mockShuttle);
+      await repository.removeShuttleIfExists(mockShuttle.id);
+
+      const lastStopData = await repository.getShuttleLastStopArrival(mockShuttle.id);
+      expect(lastStopData).toBeUndefined();
+    });
+
     test("does nothing if shuttle doesn't exist", async () => {
       const systemId = "1";
       const mockShuttles = generateMockShuttles();