refactor: Expose popper testing helper

cozy · Apr 27, 2020 · 2c32652 · 2c32652
1 parent f30f6e3
commit 2c32652
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 17 deletions.
diff --git a/react/ActionMenu/index.spec.jsx b/react/ActionMenu/index.spec.jsx
@@ -2,25 +2,10 @@ import React from 'react'
 import { mount } from 'enzyme'
 
 import ActionMenu, { ActionMenuItem } from './'
+import fixPopperTesting from '../Popper/testing'
 
 describe('ActionMenu', () => {
-  let createRangeBackup
-
-  beforeAll(() => {
-    createRangeBackup = global.document.createRange
-    global.document.createRange = jest.fn(() => ({
-      setStart: () => {},
-      setEnd: () => {},
-      commonAncestorContainer: {
-        nodeName: 'BODY',
-        ownerDocument: document
-      }
-    }))
-  })
-
-  afterAll(() => {
-    global.document.createRange = createRangeBackup
-  })
+  fixPopperTesting()
 
   it('should support null children', () => {
     const comp = mount(

diff --git a/react/Popper/testing.js b/react/Popper/testing.js
@@ -0,0 +1,23 @@
+// Fixes the error "TypeError: document.createRange is not a function" in jest,
+// see https://github.com/mui-org/material-ui/issues/15726#issuecomment-493124813
+const fixPopperTesting = () => {
+  let createRangeBackup
+
+  beforeAll(() => {
+    createRangeBackup = global.document.createRange
+    global.document.createRange = jest.fn(() => ({
+      setStart: () => {},
+      setEnd: () => {},
+      commonAncestorContainer: {
+        nodeName: 'BODY',
+        ownerDocument: document
+      }
+    }))
+  })
+
+  afterAll(() => {
+    global.document.createRange = createRangeBackup
+  })
+}
+
+export default fixPopperTesting