Merge branch 'main' into feat/notification-service

test/repositories/UnoptimizedInMemoryRepositoryTests.test.ts

@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, test } from "@jest/globals";
+import { beforeEach, describe, expect, jest, test } from "@jest/globals";
 import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
 import {
   generateMockEtas,
@@ -225,6 +225,51 @@ describe("UnoptimizedInMemoryRepository", () => {
     });
   });
 
+  describe("subscribeToEtaChanges", () => {
+    test("notifies listeners if etas have been added or changed", async () => {
+      const mockCallback = jest.fn(); // Jest mock function to simulate a listener
+      repository.subscribeToEtaUpdates(mockCallback);
+
+      const mockEtas = generateMockEtas();
+      for (const eta of mockEtas) {
+        await repository.addOrUpdateEta(eta); // Trigger changes in ETAs
+      }
+
+      expect(mockCallback).toHaveBeenCalledTimes(mockEtas.length);
+      expect(mockCallback).toHaveBeenCalledWith(mockEtas[0]); // First notification
+      expect(mockCallback).toHaveBeenCalledWith(mockEtas[mockEtas.length - 1]); // Last notification
+    });
+  });
+
+  describe("unsubscribeFromEtaChanges", () => {
+    test("stops notifying listeners after etas have stopped changing", async () => {
+      const mockCallback = jest.fn(); // Jest mock function to simulate a listener
+      repository.subscribeToEtaUpdates(mockCallback);
+
+      const mockEtas = generateMockEtas();
+      await repository.addOrUpdateEta(mockEtas[0]);
+
+      repository.unsubscribeFromEtaUpdates(mockCallback);
+
+      await repository.addOrUpdateEta(mockEtas[mockEtas.length - 1]);
+
+      expect(mockCallback).toHaveBeenCalledTimes(1);
+      expect(mockCallback).toHaveBeenCalledWith(mockEtas[0]); // First notification
+      expect(mockCallback).not.toHaveBeenCalledWith(mockEtas[mockEtas.length - 1]); // Last notification
+    });
+
+    test("does nothing if the listener doesn't exist", async () => {
+      const mockCallback = jest.fn();
+      repository.subscribeToEtaUpdates(mockCallback);
+
+      const mockEtas = generateMockEtas();
+
+      repository.unsubscribeFromEtaUpdates(() => {});
+      await repository.addOrUpdateEta(mockEtas[0]);
+      expect(mockCallback).toHaveBeenCalledTimes(1);
+    });
+  });
+
   describe("getOrderedStopByRouteAndStopId", () => {
     test("gets an ordered stop by route ID and stop ID", async () => {
       const mockOrderedStops = generateMockOrderedStops();
@@ -711,4 +756,4 @@ describe("UnoptimizedInMemoryRepository", () => {
       expect(result).toEqual([]);
     });
   });
-});
+});