Refactor so all node types have outerHTML.

Author: ntoll

polyplug.py

@@ -368,12 +368,7 @@ def innerHTML(self):
         """
         result = ""
         for child in self.childNodes:
-            if isinstance(child, ElementNode):
-                result += child.outerHTML
-            elif isinstance(child, TextNode):
-                result += child.nodeValue
-            elif isinstance(child, CommentNode):
-                result += "<!--" + child.nodeValue + "-->"
+            result += child.outerHTML
         return result
 
     @innerHTML.setter
@@ -433,6 +428,13 @@ def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self.nodeValue = kwargs.get("nodeValue")
 
+    @property
+    def outerHTML(self):
+        """
+        Get a string representation of the element's outer HTML.
+        """
+        return self.nodeValue
+
     @property
     def as_dict(self):
         return {
@@ -452,6 +454,13 @@ def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self.nodeValue = kwargs.get("nodeValue")
 
+    @property
+    def outerHTML(self):
+        """
+        Get a string representation of the element's outer HTML.
+        """
+        return "<!--" + self.nodeValue + "-->"
+
     @property
     def as_dict(self):
         return {
@@ -470,6 +479,13 @@ class FragmentNode(Node):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
+    @property
+    def outerHTML(self):
+        """
+        Get a string representation of the element's outer HTML.
+        """
+        return ""
+
     @property
     def as_dict(self):
         return {"nodeType": 11, "childNodes": []}

tests/test_polyplug.py

@@ -378,6 +378,7 @@ def test_text_node():
     )
     assert n.nodeValue == "Test text."
     assert n.parent == "fakeParent"
+    assert n.outerHTML == "Test text."
 
 
 def test_comment_node():
@@ -392,6 +393,7 @@ def test_comment_node():
     )
     assert n.nodeValue == "Test comment."
     assert n.parent == "fakeParent"
+    assert n.outerHTML == "<!--Test comment.-->"
 
 
 def test_fragment_node():
@@ -404,6 +406,7 @@ def test_fragment_node():
         parent="fakeParent",
     )
     assert n.parent == "fakeParent"
+    assert n.outerHTML == ""
 
 
 def test_htmltokenizer_init():