fix: safer reads from context consumer

gaearon · Feb 25, 2019 · 7942d26 · 7942d26
1 parent b1a9bd7
commit 7942d26
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/reactHotLoader.js b/src/reactHotLoader.js
@@ -76,10 +76,18 @@ const reactHotLoader = {
       registerComponent(type)
     }
     if (isContextType({ type })) {
+      // possible options - Context, Consumer, Provider.
+      ;['Provider', 'Consumer'].forEach(prop => {
+        const descriptor = Object.getOwnPropertyDescriptor(type, prop)
+        if (descriptor && descriptor.value) {
+          updateFunctionProxyById(
+            `${id}:${prop}`,
+            descriptor.value,
+            updateContext,
+          )
+        }
+      })
       updateFunctionProxyById(id, type, updateContext)
-      if (type.Provider) {
-        updateFunctionProxyById(`${id}:provider`, type.Provider, updateContext)
-      }
       incrementGeneration()
     }
     if (isLazyType({ type })) {