[New] RSTTraversal: add getHTMLFromHostNodes

enzymejs · Mar 15, 2019 · 8d4262b · 8d4262b
1 parent ebdd4e7
commit 8d4262b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/packages/enzyme/src/RSTTraversal.js b/packages/enzyme/src/RSTTraversal.js
@@ -181,3 +181,18 @@ export function getTextFromHostNodes(node, adapter) {
     },
   });
 }
+
+export function getHTMLFromHostNodes(node, adapter) {
+  if (node === null) return null;
+
+  const hostNode = adapter.nodeToHostNode(node, true);
+  if (hostNode === null) return null;
+
+  const nodeArray = Array.isArray(hostNode) ? hostNode : [hostNode];
+  const nodesHTML = nodeArray.map(item => (item === null
+    ? null
+    : item.outerHTML.replace(/\sdata-(reactid|reactroot)+="([^"]*)+"/g, '')
+  ));
+
+  return nodesHTML.join('');
+}