Add additional type checks to adapters

enzymejs · Jul 2, 2018 · 9b354d1 · 9b354d1
1 parent 0914749
commit 9b354d1
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/enzyme/src/Utils.js b/packages/enzyme/src/Utils.js
@@ -43,10 +43,18 @@ export function typeOfNode(node) {
 
 export function nodeHasType(node, type) {
   if (!type || !node) return false;
+
+  const adapter = getAdapter();
+  if (adapter.displayNameOfNode) {
+    const displayName = adapter.displayNameOfNode(node);
+    if (displayName === type) return true;
+  }
+
   if (!node.type) return false;
   if (typeof node.type === 'string') return node.type === type;
-  return (typeof node.type === 'function' ?
-    functionName(node.type) === type : node.type.name === type) || node.type.displayName === type;
+  return (
+    typeof node.type === 'function' ? functionName(node.type) === type : node.type.name === type
+  ) || node.type.displayName === type;
 }
 
 function internalChildrenCompare(a, b, lenComp, isLoose) {
@@ -215,6 +223,7 @@ export function AND(fns) {
   return x => fnsReversed.every(fn => fn(x));
 }
 
+// TODO: semver-major: remove
 export function displayNameOfNode(node) {
   if (!node) return null;