feat: consider displayName of context types

Author: eps1lon

packages/react/src/ReactContext.js

@@ -120,6 +120,11 @@ export function createContext<T>(
           return context.Consumer;
         },
       },
+      displayName: {
+        get() {
+          return context.displayName;
+        },
+      },
     });
     // $FlowFixMe: Flow complains about missing properties because it doesn't understand defineProperty
     context.Consumer = Consumer;

packages/react/src/__tests__/ReactContext-test.js

@@ -21,7 +21,7 @@ describe('ReactContext', () => {
     ReactDOMServer = require('react-dom/server');
   });
 
-  it('ignores displayName on the context type', () => {
+  it('should honor a displayName if set on the context type', () => {
     const Context = React.createContext(null);
     Context.displayName = 'MyContextType';
     function Validator() {
@@ -38,8 +38,8 @@ describe('ReactContext', () => {
     }).toErrorDev(
       'Warning: Failed prop type: The prop `dontPassToSeeErrorStack` is marked as required in `Validator`, but its value is `undefined`.\n' +
         '    in Validator (at **)\n' +
-        '    in Context.Consumer (at **)\n' +
-        '    in Context.Provider (at **)',
+        '    in MyContextType.Consumer (at **)\n' +
+        '    in MyContextType.Provider (at **)',
     );
   });
 });

packages/shared/getComponentName.js

@@ -24,6 +24,7 @@ import {
   REACT_BLOCK_TYPE,
 } from 'shared/ReactSymbols';
 import {refineResolvedLazyComponent} from 'shared/ReactLazyComponent';
+import type {ReactContext, ReactProviderType} from 'shared/ReactTypes';
 
 function getWrappedName(
   outerType: mixed,
@@ -37,6 +38,10 @@ function getWrappedName(
   );
 }
 
+function getContextName(type: ReactContext<any>) {
+  return type.displayName || 'Context';
+}
+
 function getComponentName(type: mixed): string | null {
   if (type == null) {
     // Host root, text node or just invalid type.
@@ -73,9 +78,11 @@ function getComponentName(type: mixed): string | null {
   if (typeof type === 'object') {
     switch (type.$$typeof) {
       case REACT_CONTEXT_TYPE:
-        return 'Context.Consumer';
+        const context: ReactContext<any> = (type: any);
+        return getContextName(context) + '.Consumer';
       case REACT_PROVIDER_TYPE:
-        return 'Context.Provider';
+        const provider: ReactProviderType<any> = (type: any);
+        return getContextName(provider._context) + '.Provider';
       case REACT_FORWARD_REF_TYPE:
         return getWrappedName(type, type.render, 'ForwardRef');
       case REACT_MEMO_TYPE: