[enzyme-adapter-react-16] [new] add shouldUpdateComponent

sstern6 · Jan 15, 2019 · e7106c4 · e7106c4
1 parent 0a17404
commit e7106c4
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js b/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
@@ -31,7 +31,7 @@ import {
   StrictMode,
 } from 'react-is';
 import { EnzymeAdapter } from 'enzyme';
-import { typeOfNode } from 'enzyme/build/Utils';
+import { typeOfNode, shallowEqual } from 'enzyme/build/Utils';
 import {
   displayNameOfNode,
   elementToTree as utilElementToTree,
@@ -701,6 +701,18 @@ class ReactSixteenAdapter extends EnzymeAdapter {
     return isElement(element);
   }
 
+  shouldUpdateComponent(prevProps, root) {
+    const instance = root.instance();
+    if (instance !== null) {
+      const { updater, props } = instance;
+      return (
+        updater._renderer._newState !== null || !shallowEqual(prevProps, props)
+      );
+    }
+
+    return true;
+  }
+
   isValidElementType(object) {
     return !!object && isValidElementType(object);
   }