From 841de73928538038496f15467aba9001b47a2d9c Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 14 Dec 2013 15:15:05 +0000
Subject: Add appending of other HTML generators

---
 tests/html_generation_tests.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'tests/html_generation_tests.py')

diff --git a/tests/html_generation_tests.py b/tests/html_generation_tests.py
index fac6f05..f32b713 100644
--- a/tests/html_generation_tests.py
+++ b/tests/html_generation_tests.py
@@ -73,3 +73,21 @@ def opening_tag_can_have_attributes():
     generator.text("Hello!")
     generator.end()
     assert_equal('<p data-blah="42">Hello!</p>', generator.html_string())
+
+
+@istest
+def appending_another_html_generator_does_nothing_if_empty():
+    generator = HtmlGenerator()
+    generator.start("p")
+    generator.append(HtmlGenerator())
+    assert_equal('', generator.html_string())
+
+
+@istest
+def appending_another_html_generator_writes_out_elements_if_other_generator_is_not_empty():
+    generator = HtmlGenerator()
+    generator.start("p")
+    other = HtmlGenerator()
+    other.text("Hello!")
+    generator.append(other)
+    assert_equal('<p>Hello!', generator.html_string())
-- 
cgit v1.2.3