Addon-docs: Fix exotic React components in Source block

addons/docs/src/frameworks/react/jsxDecorator.test.tsx

@@ -106,6 +106,31 @@ describe('renderJsx', () => {
        />
     `);
   });
+
+  it('forwardRef component', () => {
+    const MyExoticComponent = React.forwardRef(function MyExoticComponent(props: any, _ref: any) {
+      return <div>{props.children}</div>;
+    });
+
+    expect(renderJsx(<MyExoticComponent>I'm forwardRef!</MyExoticComponent>, {}))
+      .toMatchInlineSnapshot(`
+        <MyExoticComponent>
+          I'm forwardRef!
+        </MyExoticComponent>
+      `);
+  });
+
+  it('memo component', () => {
+    const MyMemoComponent = React.memo(function MyMemoComponent(props: any) {
+      return <div>{props.children}</div>;
+    });
+
+    expect(renderJsx(<MyMemoComponent>I'm memo!</MyMemoComponent>, {})).toMatchInlineSnapshot(`
+      <MyMemoComponent>
+        I'm memo!
+      </MyMemoComponent>
+    `);
+  });
 });
 
 // @ts-ignore

addons/docs/src/frameworks/react/jsxDecorator.tsx

@@ -87,7 +87,16 @@ export const renderJsx = (code: React.ReactElement, options: JSXOptions) => {
   const displayNameDefaults =
     typeof options.displayName === 'string'
       ? { showFunctions: true, displayName: () => options.displayName }
-      : {};
+      : {
+          // To get exotic component names resolving properly
+          displayName: (el: any): string =>
+            el.type.displayName ||
+            (el.type.name !== '_default' ? el.type.name : null) ||
+            (typeof el.type === 'function' ? 'No Display Name' : null) ||
+            (el.type.$$typeof === Symbol.for('react.forward_ref') ? el.type.render.name : null) ||
+            (el.type.$$typeof === Symbol.for('react.memo') ? el.type.type.name : null) ||
+            el.type,
+        };
 
   const filterDefaults = {
     filterProps: (value: any, key: string): boolean => value !== undefined,

lib/cli/versions.json

@@ -52,4 +52,4 @@
   "@storybook/ui": "6.1.0-alpha.18",
   "@storybook/vue": "6.1.0-alpha.18",
   "@storybook/web-components": "6.1.0-alpha.18"
-}
+}