Improve code coverage

COMP1010UNSW · Apr 13, 2024 · 79e65cd · 79e65cd
1 parent 76c30c1
commit 79e65cd
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 20 deletions.
diff --git a/tests/basic_rendering_test.py b/tests/basic_rendering_test.py
@@ -17,7 +17,6 @@
     br,
     input,
     a,
-    DangerousRawHtml,
 )
 
 
@@ -249,21 +248,3 @@ def test_boolean_tag_attributes_false():
     Attributes with value `False` are skipped
     """
     assert str(input(readonly=False)) == "<input/>"
-
-
-def test_dangerous_raw_html():
-    """
-    Is raw HTML rendered correctly?
-    """
-    assert str(DangerousRawHtml("<script>alert(1)</script>")) \
-        == "<script>alert(1)</script>"
-
-    assert str(
-        html(
-            DangerousRawHtml("<script>alert(1)</script>")
-        )
-    ) == "\n".join([
-        "<html>",
-        "  <script>alert(1)</script>",
-        "</html>",
-    ])
diff --git a/tests/end_to_end/flask_test.py b/tests/end_to_end/flask_test.py
@@ -76,5 +76,6 @@ def test_failed_to_stringify(client: FlaskClient):
     assert "**HINT:** if you're using Flask" in str(exception_info.value)
 
 
-if __name__ == '__main__':
+# Ignore coverage since this won't be used when running tests automatically
+if __name__ == '__main__':  # pragma: no cover
     app.run(debug=True)
diff --git a/tests/raw_html_test.py b/tests/raw_html_test.py
@@ -0,0 +1,33 @@
+"""
+# Tests / Raw HTML Test
+
+Test cases for `DangerousRawHtml` tag
+"""
+import pytest
+from pyhtml import DangerousRawHtml, html
+
+
+def test_dangerous_raw_html():
+    """
+    Is raw HTML rendered correctly?
+    """
+    assert str(DangerousRawHtml("<script>alert(1)</script>")) \
+        == "<script>alert(1)</script>"
+
+    assert str(
+        html(
+            DangerousRawHtml("<script>alert(1)</script>")
+        )
+    ) == "\n".join([
+        "<html>",
+        "  <script>alert(1)</script>",
+        "</html>",
+    ])
+
+
+def test_raw_html_not_callable():
+    """
+    `DangerousRawHtml` elements should not be callable.
+    """
+    with pytest.raises(TypeError):
+        DangerousRawHtml("")()