feat: add React17 support (#74)

Co-authored-by: Baruch Velez <baruch.velez@wooga.net>
baruchvlz · Nov 17, 2020 · 465ab5e · 465ab5e
1 parent f256c70
commit 465ab5e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/utils.js b/src/utils.js
@@ -330,7 +330,9 @@ export function findReactInstance(element) {
         return element._reactRootContainer._internalRoot.current
     }
 
-    const instanceId = Object.keys(element).find(key => key.startsWith('__reactInternalInstance'))
+    const instanceId = Object.keys(element).find(
+        key => key.startsWith('__reactInternalInstance') || key.startsWith('__reactFiber')
+    )
 
     if (instanceId) {
         return element[instanceId]

diff --git a/tests/utils.test.js b/tests/utils.test.js
@@ -635,6 +635,14 @@ describe('utils', () => {
             expect(findReactInstance(element)).toBeTruthy()
         })
 
+        it('should work with React 17', () => {
+            const element = {
+                __reactFiber$test1234: true,
+            }
+
+            expect(findReactInstance(element)).toBeTruthy()
+        })
+
         it('should return undefined if no instance is found', () => {
             expect(findReactInstance(document.createElement('div'))).toBeFalsy()
         })