Show functions and not <Profiler/> with react 16.3

enzymejs · Aug 27, 2018 · 4c24a76 · 4c24a76
1 parent a7b6e78
commit 4c24a76
Showing 1 changed file with 20 additions and 17 deletions.
diff --git a/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js b/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js
@@ -417,27 +417,30 @@ class ReactSixteenThreeAdapter extends EnzymeAdapter {
   displayNameOfNode(node) {
     if (!node) return null;
     const { type, $$typeof } = node;
+
+    const nodeType = type || $$typeof;
 
-    switch (type || $$typeof) {
-      case AsyncMode: return 'AsyncMode';
-      case Fragment: return 'Fragment';
-      case StrictMode: return 'StrictMode';
-      case Profiler: return 'Profiler';
-      case Portal: return 'Portal';
+    // newer node types may be undefined, so only test if the nodeType exists
+    if (nodeType) {
+      switch (nodeType) {
+        case AsyncMode: return 'AsyncMode';
+        case Fragment: return 'Fragment';
+        case StrictMode: return 'StrictMode';
+        case Profiler: return 'Profiler';
+        case Portal: return 'Portal';
+      }
+    }
 
-      default: {
-        const $$typeofType = type && type.$$typeof;
+    const $$typeofType = type && type.$$typeof;
 
-        switch ($$typeofType) {
-          case ContextConsumer: return 'ContextConsumer';
-          case ContextProvider: return 'ContextProvider';
-          case ForwardRef: {
-            const name = type.render.displayName || functionName(type.render);
-            return name ? `ForwardRef(${name})` : 'ForwardRef';
-          }
-          default: return displayNameOfNode(node);
-        }
+    switch ($$typeofType) {
+      case ContextConsumer: return 'ContextConsumer';
+      case ContextProvider: return 'ContextProvider';
+      case ForwardRef: {
+        const name = type.render.displayName || functionName(type.render);
+        return name ? `ForwardRef(${name})` : 'ForwardRef';
       }
+      default: return displayNameOfNode(node);
     }
   }