From b89005feb3593f8cbf36f0e8dbb83df57c90d48b Mon Sep 17 00:00:00 2001
From: Cesar Gonzalez <cgonzalez@bitwarden.com>
Date: Fri, 7 Jun 2024 06:45:18 -0500
Subject: [PATCH] [PM-6426] Adding jest tests to the
 ForegroundTaskSchedulerService and the BackgroundTaskSchedulerService

---
 .../task-scheduler/background-task-scheduler.service.spec.ts    | 2 ++
 .../task-scheduler/background-task-scheduler.service.ts         | 1 +
 2 files changed, 3 insertions(+)

diff --git a/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.spec.ts b/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.spec.ts
index 110326e3d7..c9eeffe83a 100644
--- a/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.spec.ts
+++ b/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.spec.ts
@@ -68,6 +68,8 @@ describe("BackgroundTaskSchedulerService", () => {
 
       triggerPortOnDisconnectEvent(portMock);
       expect(backgroundTaskSchedulerService["ports"].size).toBe(0);
+      expect(portMock.onMessage.removeListener).toHaveBeenCalled();
+      expect(portMock.onDisconnect.removeListener).toHaveBeenCalled();
     });
   });
 
diff --git a/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.ts b/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.ts
index 6e5a857c39..23b580988f 100644
--- a/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.ts
+++ b/apps/browser/src/platform/services/task-scheduler/background-task-scheduler.service.ts
@@ -41,6 +41,7 @@ export class BackgroundTaskSchedulerService extends BrowserTaskSchedulerServiceI
    */
   private handlePortOnDisconnect = (port: chrome.runtime.Port) => {
     port.onMessage.removeListener(this.handlePortMessage);
+    port.onDisconnect.removeListener(this.handlePortOnDisconnect);
     this.ports.delete(port);
   };