fix: use inner displayName as default

Author: jquense

package.json

@@ -1,12 +1,12 @@
 {
-  "name": "react-context-toolbox",
+  "name": "@restart/context",
   "version": "2.0.2",
   "main": "lib/index.js",
   "module": "lib/es/index.js",
   "types": "index.d.ts",
   "repository": {
     "type": "git",
-    "url": "https://github.com/4Catalyzer/react-context-toolbox.git"
+    "url": "https://github.com/react-restart/context"
   },
   "author": "4Catalyzer",
   "license": "MIT",
@@ -42,9 +42,6 @@
     ]
   },
   "jest": {
-    "roots": [
-      "<rootDir>/test"
-    ],
     "testEnvironment": "jsdom",
     "setupFiles": [
       "<rootDir>/test/index.js"

src/forwardRef.js

@@ -2,11 +2,15 @@ import React from 'react';
 
 export default function forwardRef(
   renderFn,
-  { displayName, propTypes, defaultProps, allowFallback = false },
+  {
+    propTypes,
+    defaultProps,
+    allowFallback = false,
+    displayName = renderFn.name || renderFn.displayName,
+  } = {},
 ) {
   const render = (props, ref) => renderFn(props, ref);
-
-  Object.assign(render, { displayName });
+  render.displayName = displayName;
 
   if (React.forwardRef || !allowFallback)
     return Object.assign(React.forwardRef(render), {

test/forwardRef.test.js

@@ -0,0 +1,15 @@
+import React from 'react';
+import { mount } from 'enzyme';
+
+import forwardRef from '../src/forwardRef';
+
+describe('forwardRef', () => {
+  it('should use existing displayName', () => {
+    // eslint-disable-next-line prefer-arrow-callback
+    const Foo = forwardRef(function Foo() {
+      return null;
+    });
+
+    expect(mount(<Foo />).find('ForwardRef(Foo)')).toHaveLength(1);
+  });
+});