Add setTestTimeout for Jest tests

Allows contributors to set the timeout
for individual jest tests.

1 files changed, 24 insertions, 0 deletions

diff --git a/spec/frontend/helpers/timeout.js b/spec/frontend/helpers/timeout.js
new file mode 100644
index 00000000000..318593a48a4
--- /dev/null
+++ b/spec/frontend/helpers/timeout.js
@@ -0,0 +1,24 @@
+let testTimeoutInMs;
+
+export const setTestTimeout = newTimeoutInMs => {
+  testTimeoutInMs = newTimeoutInMs;
+  jest.setTimeout(newTimeoutInMs);
+};
+
+export const initializeTestTimeout = defaultTimeoutInMs => {
+  setTestTimeout(defaultTimeoutInMs);
+
+  let testStartTime;
+
+  // https://github.com/facebook/jest/issues/6947
+  beforeEach(() => {
+    testStartTime = Date.now();
+  });
+
+  afterEach(() => {
+    const elapsedTimeInMs = Date.now() - testStartTime;
+    if (elapsedTimeInMs > testTimeoutInMs) {
+      throw new Error(`Test took too long (${elapsedTimeInMs}ms > ${testTimeoutInMs}ms)!`);
+    }
+  });
+};