Apply patch from enzymejs/enzyme#2189 (comment): support .contextType…

… on components
pablopalacios · Oct 6, 2022 · 6e63b07 · 6e63b07
1 parent 802c735
commit 6e63b07
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/ReactShallowRenderer.js b/src/ReactShallowRenderer.js
@@ -515,7 +515,9 @@ See https://fb.me/react-invalid-hook-call for tips about how to debug and fix th
 
     this._rendering = true;
     this._element = element;
-    this._context = getMaskedContext(elementType.contextTypes, context);
+    this._context = element.contextType
+      ? context
+      : getMaskedContext(elementType.contextTypes, context);
 
     // Inner memo component props aren't currently validated in createElement.
     let prevGetStack;