Add starting and ending tags

1 files changed, 10 insertions, 0 deletions

diff --git a/tests/html_generation_tests.py b/tests/html_generation_tests.py
index 68034d9..395af30 100644
--- a/tests/html_generation_tests.py
+++ b/tests/html_generation_tests.py
@@ -14,3 +14,13 @@ def html_escapes_text():
     generator = HtmlGenerator()
     generator.text("<")
     assert_equal("&lt;", generator.html_string())
+
+
+@istest
+def all_elements_are_closed_by_end_all():
+    generator = HtmlGenerator()
+    generator.start("p")
+    generator.start("span")
+    generator.text("Hello!")
+    generator.end_all()
+    assert_equal("<p><span>Hello!</span></p>", generator.html_string())