Update DOMElement.ts

KonnorRogers · web-flow · commit ea52ca4add05 · 2024-06-24T03:40:36.000-04:00
diff --git a/packages/pretty-format/src/plugins/DOMElement.ts b/packages/pretty-format/src/plugins/DOMElement.ts
@@ -30,24 +30,27 @@ const testHasAttribute = (val: any) => {
   }
 };
 
+const isCustomElement = (val: any) => {
+  const { tagName } = val
+  return (typeof tagName === 'string' && tagName.includes('-')) ||
+  testHasAttribute(val);
+}
+
 const testNode = (val: any) => {
   const constructorName = val.constructor.name;
-  const {nodeType, tagName} = val;
-  const isCustomElement =
-    (typeof tagName === 'string' && tagName.includes('-')) ||
-    testHasAttribute(val);
+  const {nodeType} = val;
 
   return (
     (nodeType === ELEMENT_NODE &&
-      (ELEMENT_REGEXP.test(constructorName) || isCustomElement)) ||
+      (ELEMENT_REGEXP.test(constructorName) || isCustomElement(val))) ||
     (nodeType === TEXT_NODE && constructorName === 'Text') ||
     (nodeType === COMMENT_NODE && constructorName === 'Comment') ||
     (nodeType === FRAGMENT_NODE && constructorName === 'DocumentFragment')
   );
 };
 
 export const test: NewPlugin['test'] = (val: any) =>
-  val?.constructor && testNode(val);
+  (val?.constructor?.name || isCustomElement(val)) && testNode(val);
 
 type HandledType = Element | Text | Comment | DocumentFragment;
 
