From f1fde07eb9538fdce00ccf16cd34144feda69a58 Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Tue, 27 Mar 2012 21:16:53 -0700
Subject: Add tests for bool_interfaces.

---
 tests/test_stanza_element.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'tests')

diff --git a/tests/test_stanza_element.py b/tests/test_stanza_element.py
index f7ec59c0..60ccb7c9 100644
--- a/tests/test_stanza_element.py
+++ b/tests/test_stanza_element.py
@@ -742,5 +742,37 @@ class TestElementBase(SleekTest):
           <foo xmlns="foo" bar="override-foo" />
         """)
 
+    def testBoolInterfaces(self):
+        """Test using boolean interfaces."""
+
+        class TestStanza(ElementBase):
+            name = "foo"
+            namespace = "foo"
+            interfaces = set(['bar'])
+            bool_interfaces = interfaces
+
+        stanza = TestStanza()
+        self.check(stanza, """
+          <foo xmlns="foo" />
+        """)
+
+        self.assertFalse(stanza['bar'], 
+                "Returned True for missing bool interface element.")
+
+        stanza['bar'] = True
+        self.check(stanza, """
+          <foo xmlns="foo">
+            <bar />
+          </foo>
+        """)
+
+        self.assertTrue(stanza['bar'],
+                "Returned False for present bool interface element.")
+
+        stanza['bar'] = False
+        self.check(stanza, """
+          <foo xmlns="foo" />
+        """)
+
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestElementBase)
-- 
cgit v1.2.3