From 0468c16bc2da167a530882bf5f18929c6df7f88a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Richard=20V=C3=A9zina?= <ml.richard.vezina@gmail.com>
Date: Wed, 13 Apr 2016 11:44:16 -0400
Subject: [PATCH] Some more coverage for XML() and DIV()

---
 gluon/tests/test_html.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/gluon/tests/test_html.py b/gluon/tests/test_html.py
index 5c233bc5..f1d32c7d 100644
--- a/gluon/tests/test_html.py
+++ b/gluon/tests/test_html.py
@@ -173,6 +173,9 @@ class TestBareHelpers(unittest.TestCase):
         # bug check for the sanitizer for closing no-close tags
         self.assertEqual(XML('<p>Test</p><br/><p>Test</p><br/>', sanitize=True),
                          XML('<p>Test</p><br /><p>Test</p><br />'))
+        # basic flatten test
+        self.assertEqual(XML('<p>Test</p>').flatten(), '<p>Test</p>')
+        self.assertEqual(XML('<p>Test</p>').flatten(render=lambda text, tag, attr: text), '<p>Test</p>')
 
     def test_XML_pickle_unpickle(self):
         # weird test
@@ -236,6 +239,9 @@ class TestBareHelpers(unittest.TestCase):
         #     DIV(BR('<>')).xml()
         # self.assertEqual(cm.exception[0], '<br/> tags cannot have components')
 
+        # test .get('attrib')
+        self.assertEqual(DIV('<p>Test</p>', _class="class_test").get('_class'), 'class_test')
+
     def test_CAT(self):
         # Empty CAT()
         self.assertEqual(CAT().xml(), '')