add Wrapped.displayName

Author: vsravan707

src/ErrorBoundary.js

@@ -59,18 +59,22 @@ class ErrorBoundary extends Component<Props, State> {
       );
     }
 
-    return children;
+    return children || null;
   }
 }
 
 export const withErrorBoundary = (
   Component: ComponentType<any>,
   FallbackComponent: ComponentType<any>,
   onError: Function,
-): Function => props => (
-  <ErrorBoundary FallbackComponent={FallbackComponent} onError={onError}>
-    <Component {...props} />
-  </ErrorBoundary>
-);
+): Function => {
+  const Wrapped = props => (
+    <ErrorBoundary FallbackComponent={FallbackComponent} onError={onError}>
+      <Component {...props} />
+    </ErrorBoundary>
+  );
+  Wrapped.displayName = `WithErrorBoundary(${getDisplayName(Component)})`;
+  return Wrapped;
+};
 
 export default ErrorBoundary;