fix: Support clearing the timeouts with numeric IDs

Fixes #461
angular · Oct 4, 2016 · 9991b5c · 9991b5c
1 parent 63bca67
commit 9991b5c
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/lib/common/timers.ts b/lib/common/timers.ts
@@ -19,14 +19,21 @@ export function patchTimer(window: any, setName: string, cancelName: string, nam
   setName += nameSuffix;
   cancelName += nameSuffix;
 
+  var tasksByHandleId: Object = {};
+
   function scheduleTask(task: Task) {
     const data = <TimerOptions>task.data;
-    data.args[0] = task.invoke;
+    data.args[0] = function() {
+      task.invoke.apply(this, arguments);
+      delete tasksByHandleId[data.handleId];
+    };
     data.handleId = setNative.apply(window, data.args);
+    tasksByHandleId[data.handleId] = task;
     return task;
   }
 
   function clearTask(task: Task) {
+    delete tasksByHandleId[(<TimerOptions>task.data).handleId];
     return clearNative((<TimerOptions>task.data).handleId);
   }
 
@@ -60,6 +67,9 @@ export function patchTimer(window: any, setName: string, cancelName: string, nam
   clearNative =
       patchMethod(window, cancelName, (delegate: Function) => function(self: any, args: any[]) {
         var task: Task = args[0];
+        if (task && typeof task === 'number') {
+          task = tasksByHandleId[<number>task];
+        }
         if (task && typeof task.type === 'string') {
           if (task.cancelFn && task.data.isPeriodic || task.runCount === 0) {
             // Do not cancel already canceled functions

diff --git a/test/common/setTimeout.spec.ts b/test/common/setTimeout.spec.ts
@@ -102,6 +102,27 @@ describe('setTimeout', function() {
     expect(typeof(cancelId.toString())).toBe('number');
   });
 
+  it('should allow cancelation by numeric timeout Id', function (done) {
+    // Node returns complex object from setTimeout, ignore this test.
+    if (isNode) {
+      done();
+      return;
+    }
+
+    var testZone = Zone.current.fork(Zone['wtfZoneSpec']).fork({ name: 'TestZone' });
+    testZone.run(() => {
+      var spy = jasmine.createSpy('spy');
+      var task: Task = <any>setTimeout(spy, 0);
+      var cancelId: number = <any>task;
+      clearTimeout(0 + cancelId);
+      setTimeout(function () {
+        expect(spy).not.toHaveBeenCalled();
+        expect(task.runCount).toEqual(-1);
+        done();
+      });
+    });
+  });
+
   it('should pass invalid values through', function() {
     clearTimeout(null);
     clearTimeout(<any>{});