Fix check of node type

package.json

@@ -43,7 +43,7 @@
     "lodash": "^4.17.12"
   },
   "peerDependencies": {
-    "enzyme": "^3.4.0"
+    "enzyme": "^3.10.0"
   },
   "devDependencies": {
     "babel-cli": "^6.26.0",
@@ -54,8 +54,8 @@
     "babel-plugin-transform-react-jsx": "^6.24.1",
     "babel-preset-env": "^1.6.0",
     "codecov": "^2.3.0",
-    "enzyme": "^3.7.0",
-    "enzyme-adapter-react-16": "^1.7.0",
+    "enzyme": "^3.10.0",
+    "enzyme-adapter-react-16": "^1.14.0",
     "eslint": "^4.8.0",
     "eslint-config-prettier": "^2.6.0",
     "eslint-config-standard": "^10.2.1",

src/utils.js

@@ -36,19 +36,19 @@ export const applyMap = (json, options) => {
 };
 
 export const extractTypeName = node => {
-  const name = typeName(node);
+  const type = node.type.$$typeof;
 
-  if (name.$$typeof === Symbol.for('react.lazy')) {
+  if (type === Symbol.for('react.lazy')) {
     return 'React.Lazy';
   }
 
-  if (name.$$typeof === Symbol.for('react.memo')) {
+  if (type === Symbol.for('react.memo')) {
     return 'React.Memo';
   }
 
-  if (name === Symbol.for('react.suspense')) {
+  if (node.type === Symbol.for('react.suspense')) {
     return 'React.Suspense';
   }
 
-  return name;
+  return typeName(node);
 };

tests/utils.test.js

@@ -1,12 +1,16 @@
 /* eslint-env jest */
 
-import React from 'react';
+// import React from 'react';
+import React, {memo, lazy, Suspense} from 'react';
 import Enzyme, {shallow, mount, render} from 'enzyme';
 import Adapter from 'enzyme-adapter-react-16';
-import {isEnzymeWrapper} from '../src/utils';
+import {isEnzymeWrapper, extractTypeName} from '../src/utils';
 
 Enzyme.configure({adapter: new Adapter()});
 
+const Lazy = lazy(() => Promise.resolve(<div>lazy div</div>));
+const Memo = memo(() => <div>memoized div</div>);
+
 it('returns true whenever it is an enzyme wrapper', () => {
   expect(isEnzymeWrapper(shallow(<div />))).toBe(true);
   expect(isEnzymeWrapper(mount(<div />))).toBe(true);
@@ -18,3 +22,9 @@ it('returns false whenever it is an enzyme wrapper', () => {
   expect(isEnzymeWrapper('test')).toBe(false);
   expect(isEnzymeWrapper(<div />)).toBe(false);
 });
+
+it('returns correct node type', () => {
+  expect(extractTypeName(<Lazy />)).toBe('React.Lazy');
+  expect(extractTypeName(<Memo />)).toBe('React.Memo');
+  expect(extractTypeName(<Suspense />)).toBe('React.Suspense');
+})