Make addTravelTimeDataPoint public, add tests, and fix the query interval

src/repositories/shuttle/__tests__/RedisShuttleRepository.test.ts

@@ -253,4 +253,113 @@ describe("RedisShuttleRepository", () => {
       expect(result?.timestamp.getTime()).toBe(secondArrival.timestamp.getTime());
     });
   });
+
+  describe("getAverageTravelTimeSeconds", () => {
+    it("returns the average travel time when historical data exists", async () => {
+      const { route, stop2, stop1 } = await setupRouteAndOrderedStops();
+
+      // Add a single data point: 15 minutes travel time
+      const timestamp = new Date(2025, 0, 1, 12, 15, 0);
+      await repository.addTravelTimeDataPoint(
+        {
+          routeId: route.id,
+          fromStopId: stop1.id,
+          toStopId: stop2.id,
+        },
+        15 * 60, // 15 minutes in seconds
+        timestamp.getTime()
+      );
+
+      const travelTime = await repository.getAverageTravelTimeSeconds({
+        routeId: route.id,
+        fromStopId: stop1.id,
+        toStopId: stop2.id,
+      }, {
+        from: new Date(2025, 0, 1, 11, 0, 0),
+        to: new Date(2025, 0, 1, 13, 0, 0),
+      });
+
+      expect(travelTime).toEqual(15 * 60);
+    });
+
+    it("returns average of multiple data points", async () => {
+      const { route, stop2, stop1 } = await setupRouteAndOrderedStops();
+
+      // First trip: 10 minutes travel time
+      await repository.addTravelTimeDataPoint(
+        {
+          routeId: route.id,
+          fromStopId: stop1.id,
+          toStopId: stop2.id,
+        },
+        10 * 60, // 10 minutes
+        new Date(2025, 0, 1, 12, 0, 0).getTime()
+      );
+
+      // Second trip: 20 minutes travel time
+      await repository.addTravelTimeDataPoint(
+        {
+          routeId: route.id,
+          fromStopId: stop1.id,
+          toStopId: stop2.id,
+        },
+        20 * 60, // 20 minutes
+        new Date(2025, 0, 1, 13, 0, 0).getTime()
+      );
+
+      const averageTravelTime = await repository.getAverageTravelTimeSeconds({
+        routeId: route.id,
+        fromStopId: stop1.id,
+        toStopId: stop2.id,
+      }, {
+        from: new Date(2025, 0, 1, 11, 0, 0),
+        to: new Date(2025, 0, 1, 14, 0, 0),
+      });
+
+      // Average of 10 minutes and 20 minutes = 15 minutes = 900 seconds
+      expect(averageTravelTime).toEqual(15 * 60);
+    });
+
+    it("returns undefined when no data exists", async () => {
+      const { route, stop1, stop2 } = await setupRouteAndOrderedStops();
+
+      // Don't add any data points, just query for data that doesn't exist
+      const averageTravelTime = await repository.getAverageTravelTimeSeconds({
+        routeId: route.id,
+        fromStopId: stop1.id,
+        toStopId: stop2.id,
+      }, {
+        from: new Date(2025, 0, 1, 11, 0, 0),
+        to: new Date(2025, 0, 1, 14, 0, 0),
+      });
+
+      expect(averageTravelTime).toBeUndefined();
+    });
+
+    it("returns undefined when querying outside the time range of data", async () => {
+      const { route, stop2, stop1 } = await setupRouteAndOrderedStops();
+
+      // Add data on Jan 1
+      await repository.addTravelTimeDataPoint(
+        {
+          routeId: route.id,
+          fromStopId: stop1.id,
+          toStopId: stop2.id,
+        },
+        15 * 60, // 15 minutes
+        new Date(2025, 0, 1, 12, 15, 0).getTime()
+      );
+
+      // Query for Jan 2 (no data should exist in this range)
+      const averageTravelTime = await repository.getAverageTravelTimeSeconds({
+        routeId: route.id,
+        fromStopId: stop1.id,
+        toStopId: stop2.id,
+      }, {
+        from: new Date(2025, 0, 2, 11, 0, 0),
+        to: new Date(2025, 0, 2, 13, 0, 0),
+      });
+      expect(averageTravelTime).toBeUndefined();
+    });
+  });
 });