Fix TimeoutScheduler disposing

src/core/timeoutScheduler.ts

@@ -2,7 +2,7 @@ import type { Disposable } from '../common';
 
 export class TimeoutScheduler implements Disposable {
   private counterExpirationWatcherId: ReturnType<typeof setTimeout> | undefined;
-  private actionWatcherId: ReturnType<typeof setTimeout> | undefined;
+  private actionWatchers = new Set<ReturnType<typeof setTimeout>>();
   private _counter = 0;
 
   constructor(
@@ -22,8 +22,7 @@ export class TimeoutScheduler implements Disposable {
     if (this.counterExpirationWatcherId)
       clearTimeout(this.counterExpirationWatcherId);
 
-    if (this.actionWatcherId)
-      clearTimeout(this.actionWatcherId);
+    this.actionWatchers.forEach(watcher => clearTimeout(watcher));
   }
 
   setTimeout(action: () => void) {
@@ -32,7 +31,13 @@ export class TimeoutScheduler implements Disposable {
 
     const timeoutIndex = Math.min(this.counter, this.timeouts.length - 1);
     const timeout = this.timeouts[timeoutIndex];
-    this.actionWatcherId = setTimeout(action, timeout);
+
+    const watcherId = setTimeout(() => {
+      this.actionWatchers.delete(watcherId);
+      clearTimeout(watcherId);
+      action();
+    }, timeout);
+    this.actionWatchers.add(watcherId);
 
     this.counter++;
   }

tests/core/timeoutScheduler.test.ts

@@ -4,6 +4,10 @@ import { TimeoutScheduler } from '../../src/core';
 describe('TimeoutScheduler', () => {
   let scheduler: TimeoutScheduler;
 
+  beforeEach(() => {
+    jest.useRealTimers();
+  });
+
   afterEach(() => {
     scheduler.dispose();
   });
@@ -76,4 +80,17 @@ describe('TimeoutScheduler', () => {
     expect(counter).toEqual(4);
     expect(scheduler.counter).toEqual(1);
   });
+
+  test('stops all scheduled tasks on dispose', () => {
+    jest.useFakeTimers();
+    const action = jest.fn();
+    scheduler = new TimeoutScheduler([10000, 10000]);
+
+    scheduler.setTimeout(action);
+    scheduler.setTimeout(action);
+    scheduler.setTimeout(action);
+    expect(jest.getTimerCount()).toBe(3);
+    scheduler.dispose();
+    expect(jest.getTimerCount()).toBe(0);
+  });
 });