added fix for 14

enzymejs · Dec 10, 2020 · 76e0109 · 76e0109
1 parent 48b8aa7
commit 76e0109
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 0 deletions.
diff --git a/packages/enzyme-adapter-react-14/package.json b/packages/enzyme-adapter-react-14/package.json
@@ -39,6 +39,7 @@
   "license": "MIT",
   "dependencies": {
     "enzyme-adapter-utils": "^1.13.1",
+    "enzyme-shallow-equal": "^1.0.4",
     "object.assign": "^4.1.0",
     "object.values": "^1.1.1",
     "prop-types": "^15.7.2",

diff --git a/packages/enzyme-adapter-react-14/src/ReactFourteenAdapter.js b/packages/enzyme-adapter-react-14/src/ReactFourteenAdapter.js
@@ -24,6 +24,8 @@ import {
   wrapWithWrappingComponent,
   getWrappingComponentMountRenderer,
 } from 'enzyme-adapter-utils';
+import { spyMethod } from 'enzyme/build/Utils';
+import shallowEqual from 'enzyme-shallow-equal';
 
 function typeToNodeType(type) {
   if (typeof type === 'function') {
@@ -172,6 +174,51 @@ class ReactFourteenAdapter extends EnzymeAdapter {
           isDOM = true;
         } else {
           isDOM = false;
+
+          const inst = renderer._instance;
+          if (inst) {
+            const { restore: restoreUpdateComponent } = spyMethod(
+              inst,
+              'updateComponent',
+              (originalUpadateComponentMethod) => function updateComponent(
+                transaction,
+                prevParentElement,
+                nextParentElement,
+                prevUnmaskedContext,
+                nextUnmaskedContext,
+              ) {
+                if (prevParentElement === nextParentElement) {
+                  const { restore: restoreProcessPendingState } = spyMethod(
+                    inst,
+                    '_processPendingState',
+                    (originalProcessPendingStateMethod) => function _processPendingState(nextProps, nextContext) {
+                      if (!shallowEqual(prevUnmaskedContext, nextUnmaskedContext)) {
+                        if (inst._instance.componentWillReceiveProps) {
+                          inst._instance.componentWillReceiveProps(nextProps, nextContext);
+                        }
+                      }
+
+                      const result = originalProcessPendingStateMethod.call(inst, nextProps, nextContext);
+                      restoreProcessPendingState();
+                      return result;
+                    },
+                  );
+                }
+
+                const result = originalUpadateComponentMethod.call(
+                  inst,
+                  transaction,
+                  prevParentElement,
+                  nextParentElement,
+                  prevUnmaskedContext,
+                  nextUnmaskedContext,
+                );
+                restoreUpdateComponent();
+                return result;
+              },
+            );
+          }
+
           return withSetStateAllowed(() => renderer.render(el, context));
         }
       },