Indent contents of comment tag

COMP1010UNSW · Sep 29, 2023 · 2b413e0 · 2b413e0
1 parent 7b1a07e
commit 2b413e0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/pyhtml/__tag_base.py b/pyhtml/__tag_base.py
@@ -111,7 +111,11 @@ def _render(self) -> list[str]:
 
         return [
             '<!--',
-            *util.escape_string(self.comment_data).splitlines(),
+            *util.increase_indent(
+                util.escape_string(self.comment_data).splitlines(),
+                # FIXME: Yucky magic number
+                2,
+            ),
             '-->'
         ]
 

diff --git a/tests/basic_rendering_test.py b/tests/basic_rendering_test.py
@@ -53,7 +53,7 @@ def test_renders_properties():
 def test_comment_renders():
     assert str(Comment("Hello world")) == '\n'.join([
         '<!--',
-        'Hello world',
+        '  Hello world',
         '-->',
     ])
 
@@ -115,3 +115,10 @@ def test_larger_page():
         '  </body>',
         '</html>',
     ])
+
+
+def test_format_through_repr():
+    """Is HTML rendered through repr?"""
+    doc = html()
+
+    assert repr(doc) == "<html></html>"