Allow sinon.useFakeTimers() to modify setTimeout() and clearTimeout()…

… after module loads (#45)
sindresorhus · Jun 12, 2019 · 1316fd7 · 1316fd7
1 parent e9b049e
commit 1316fd7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -6,14 +6,15 @@ const createAbortError = () => {
 	return error;
 };
 
-const createDelay = ({clearTimeout: clear = clearTimeout, setTimeout: set = setTimeout, willResolve}) => (ms, {value, signal} = {}) => {
+const createDelay = ({clearTimeout: defaultClear, setTimeout: set, willResolve}) => (ms, {value, signal} = {}) => {
 	if (signal && signal.aborted) {
 		return Promise.reject(createAbortError());
 	}
 
 	let timeoutId;
 	let settle;
 	let rejectFn;
+	const clear = defaultClear || clearTimeout;
 
 	const signalListener = () => {
 		clear(timeoutId);
@@ -37,7 +38,7 @@ const createDelay = ({clearTimeout: clear = clearTimeout, setTimeout: set = setT
 		};
 
 		rejectFn = reject;
-		timeoutId = set(settle, ms);
+		timeoutId = (set || setTimeout)(settle, ms);
 	});
 
 	if (signal) {