.

Author: mdjastrzebski

src/helpers/format-element.ts

@@ -1,4 +1,3 @@
-import type { ElementType } from 'react';
 import type { ReactTestInstance, ReactTestRendererJSON } from 'react-test-renderer';
 import type { NewPlugin } from 'pretty-format';
 import prettyFormat, { plugins } from 'pretty-format';
@@ -37,7 +36,7 @@ export function formatElement(
       // This prop is needed persuade the prettyFormat that the element is
       // a ReactTestRendererJSON instance, so it is formatted as JSX.
       $$typeof: Symbol.for('react.test.json'),
-      type: formatElementType(element.type),
+      type: `${element.type}`,
       props: mapProps ? mapProps(props) : props,
       children: childrenToDisplay,
     },
@@ -52,30 +51,6 @@ export function formatElement(
   );
 }
 
-export function formatElementType(type: ElementType): string {
-  if (typeof type === 'function') {
-    return type.displayName ?? type.name;
-  }
-
-  if (typeof type === 'object' && 'type' in type) {
-    // @ts-expect-error: despite typing this can happen for class components, e.g. HOCs
-    const nestedType = formatElementType(type.type);
-    if (nestedType) {
-      return nestedType;
-    }
-  }
-
-  if (typeof type === 'object' && 'render' in type) {
-    // @ts-expect-error: despite typing this can happen for class components, e.g. HOCs
-    const nestedType = formatElementType(type.render);
-    if (nestedType) {
-      return nestedType;
-    }
-  }
-
-  return `${type}`;
-}
-
 export function formatElementList(elements: ReactTestInstance[], options?: FormatElementOptions) {
   if (elements.length === 0) {
     return '(no elements)';