Add additional type checks to adapters

enzymejs · Jul 2, 2018 · 12dc650 · 12dc650
1 parent aeef090
commit 12dc650
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/enzyme/src/Utils.js b/packages/enzyme/src/Utils.js
@@ -61,6 +61,13 @@ 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 (
@@ -235,6 +242,7 @@ export function AND(fns) {
   return x => fnsReversed.every(fn => fn(x));
 }
 
+// TODO: semver-major: remove
 export function displayNameOfNode(node) {
   if (!node) return null;