diff --git a/test/repositories/ApiBasedRepositoryTests.test.ts b/test/repositories/ApiBasedRepositoryTests.test.ts index 28cbdfd..3248494 100644 --- a/test/repositories/ApiBasedRepositoryTests.test.ts +++ b/test/repositories/ApiBasedRepositoryTests.test.ts @@ -46,6 +46,12 @@ describe("getEtaForShuttleAndStopId", () => { shuttleId: "5577", stopId: "177666", millisecondsSinceEpoch: Date.now(), + }, + { + secondsRemaining: 226, + shuttleId: "9909", + stopId: "177666", + millisecondsSinceEpoch: Date.now(), } ], }, @@ -74,6 +80,8 @@ describe("getEtaForShuttleAndStopId", () => { expect(result?.secondsRemaining).toEqual(587); expect(result?.millisecondsSinceEpoch).toBeDefined(); + expect(result?.shuttleId).toEqual("5577"); + expect(result?.stopId).toEqual("177666"); }); test("getEtaForShuttleAndStopId returns null if API call is invalid and cache is empty", async () => {