From ad7d62e1ac9ee06edfdd8eac7e0ab3fac868f336 Mon Sep 17 00:00:00 2001
From: Paul Slaughter <pslaughter@gitlab.com>
Date: Wed, 8 May 2019 09:36:38 -0500
Subject: [PATCH] Use process timer for jest timeout testing

---
 spec/frontend/helpers/timeout.js | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/spec/frontend/helpers/timeout.js b/spec/frontend/helpers/timeout.js
index 318593a48a4..b30b7f1ce1e 100644
--- a/spec/frontend/helpers/timeout.js
+++ b/spec/frontend/helpers/timeout.js
@@ -1,24 +1,31 @@
-let testTimeoutInMs;
+const NS_PER_SEC = 1e9;
+const NS_PER_MS = 1e6;
 
-export const setTestTimeout = newTimeoutInMs => {
-  testTimeoutInMs = newTimeoutInMs;
-  jest.setTimeout(newTimeoutInMs);
+let testTimeoutNS;
+
+export const setTestTimeout = newTimeoutMS => {
+  testTimeoutNS = newTimeoutMS * NS_PER_MS;
+  jest.setTimeout(newTimeoutMS);
 };
 
-export const initializeTestTimeout = defaultTimeoutInMs => {
-  setTestTimeout(defaultTimeoutInMs);
+export const initializeTestTimeout = defaultTimeoutMS => {
+  setTestTimeout(defaultTimeoutMS);
 
   let testStartTime;
 
   // https://github.com/facebook/jest/issues/6947
   beforeEach(() => {
-    testStartTime = Date.now();
+    testStartTime = process.hrtime();
   });
 
   afterEach(() => {
-    const elapsedTimeInMs = Date.now() - testStartTime;
-    if (elapsedTimeInMs > testTimeoutInMs) {
-      throw new Error(`Test took too long (${elapsedTimeInMs}ms > ${testTimeoutInMs}ms)!`);
+    const [seconds, remainingNs] = process.hrtime(testStartTime);
+    const elapsedNS = seconds * NS_PER_SEC + remainingNs;
+
+    if (elapsedNS > testTimeoutNS) {
+      throw new Error(
+        `Test took too long (${elapsedNS / NS_PER_MS}ms > ${testTimeoutNS / NS_PER_MS}ms)!`,
+      );
     }
   });
 };