[enzyme, enzyme-adapter-react-16] [new] add matchesElementType

enzymejs · Jun 1, 2019 · 65881b2 · 65881b2
1 parent 9cc5f44
commit 65881b2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js b/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
@@ -767,6 +767,18 @@ class ReactSixteenAdapter extends EnzymeAdapter {
     return React.createElement(isMemo(type) ? type.type : type, propsWithKeysAndRef(node));
   }
 
+  matchesElementType(node, matchingType) {
+    if (!node) {
+      return node;
+    }
+    const { type } = node;
+
+    const nodeType = isMemo(type) ? type.type : type;
+    const matchingTypeType = isMemo(matchingType) ? matchingType.type : matchingType;
+    // console.log('**', isMemo(type), type === matchingType, type.type === matchingType, type === matchingType.type, type.type === matchingType.type);
+    return nodeType === matchingTypeType;
+  }
+
   elementToNode(element) {
     return elementToTree(element);
   }

diff --git a/packages/enzyme/src/EnzymeAdapter.js b/packages/enzyme/src/EnzymeAdapter.js
@@ -22,6 +22,14 @@ class EnzymeAdapter {
     throw unimplementedError('nodeToElement', 'EnzymeAdapter');
   }
 
+  matchesElementType(node, matchingType) {
+    if (!node) {
+      return node;
+    }
+    const { type } = node;
+    return type === matchingType;
+  }
+
   // eslint-disable-next-line class-methods-use-this, no-unused-vars
   isValidElement(element) {
     throw unimplementedError('isValidElement', 'EnzymeAdapter');