Implement node Timer api when running in node environment.

packages/jest-environment-jsdom/src/__tests__/jsdom_environment.test.js

@@ -0,0 +1,24 @@
+/**
+ * Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+'use strict';
+
+const JSDomEnvironment = require.requireActual('../');
+
+describe('JSDomEnvironment', () => {
+  it('should configure setTimeout/setInterval to use the browser api', () => {
+    const env1 = new JSDomEnvironment({});
+
+    env1.fakeTimers.useFakeTimers();
+
+    const timer1 = env1.global.setTimeout(() => {}, 0);
+    const timer2 = env1.global.setInterval(() => {}, 0);
+
+    [timer1, timer2].forEach(timer => {
+      expect(typeof timer).toBe('number');
+    });
+  });
+});

packages/jest-environment-jsdom/src/index.js

@@ -17,7 +17,7 @@ import JSDom from 'jsdom';
 
 class JSDOMEnvironment {
   document: ?Object;
-  fakeTimers: ?FakeTimers;
+  fakeTimers: ?FakeTimers<number>;
   global: ?Global;
   moduleMocker: ?ModuleMocker;
 
@@ -44,7 +44,18 @@ class JSDOMEnvironment {
     }
 
     this.moduleMocker = new mock.ModuleMocker(global);
-    this.fakeTimers = new FakeTimers(global, this.moduleMocker, config);
+
+    const timerConfig = {
+      idToRef: (id: number) => id,
+      refToId: (ref: number) => ref,
+    };
+
+    this.fakeTimers = new FakeTimers({
+      config,
+      global,
+      moduleMocker: this.moduleMocker,
+      timerConfig,
+    });
   }
 
   dispose(): void {

packages/jest-environment-node/src/__tests__/node_environment.test.js

@@ -27,4 +27,19 @@ describe('NodeEnvironment', () => {
 
     expect(env1.global.global).toBe(env1.global);
   });
+
+  it('should configure setTimeout/setInterval to use the node api', () => {
+    const env1 = new NodeEnvironment({});
+
+    env1.fakeTimers.useFakeTimers();
+
+    const timer1 = env1.global.setTimeout(() => {}, 0);
+    const timer2 = env1.global.setInterval(() => {}, 0);
+
+    [timer1, timer2].forEach(timer => {
+      expect(timer.id).not.toBeUndefined();
+      expect(typeof timer.ref).toBe('function');
+      expect(typeof timer.unref).toBe('function');
+    });
+  });
 });

packages/jest-environment-node/src/index.js

@@ -16,9 +16,15 @@ import vm from 'vm';
 import {FakeTimers, installCommonGlobals} from 'jest-util';
 import mock from 'jest-mock';
 
+type Timer = {|
+  id: number,
+  ref: () => Timer,
+  unref: () => Timer,
+|};
+
 class NodeEnvironment {
   context: ?vm$Context;
-  fakeTimers: ?FakeTimers;
+  fakeTimers: ?FakeTimers<Timer>;
   global: ?Global;
   moduleMocker: ?ModuleMocker;
 
@@ -33,7 +39,30 @@ class NodeEnvironment {
     global.setTimeout = setTimeout;
     installCommonGlobals(global, config.globals);
     this.moduleMocker = new mock.ModuleMocker(global);
-    this.fakeTimers = new FakeTimers(global, this.moduleMocker, config);
+
+    const timerIdToRef = (id: number) => ({
+      id,
+      ref() {
+        return this;
+      },
+      unref() {
+        return this;
+      },
+    });
+
+    const timerRefToId = (timer: Timer) => timer.id;
+
+    const timerConfig = {
+      idToRef: timerIdToRef,
+      refToId: timerRefToId,
+    };
+
+    this.fakeTimers = new FakeTimers({
+      config,
+      global,
+      moduleMocker: this.moduleMocker,
+      timerConfig,
+    });
   }
 
   dispose() {