1 files changed, 41 insertions, 0 deletions
diff --git a/tests/test_stanza_xep_0085.py b/tests/test_stanza_xep_0085.py
new file mode 100644
index 00000000..b08404e2
--- /dev/null
+++ b/tests/test_stanza_xep_0085.py
@@ -0,0 +1,41 @@
+from sleekxmpp.test import *
+import sleekxmpp.plugins.xep_0085 as xep_0085
+
+class TestChatStates(SleekTest):
+
+    def setUp(self):
+        register_stanza_plugin(Message, xep_0085.ChatState)
+
+    def testCreateChatState(self):
+        """Testing creating chat states."""
+
+        xmlstring = """
+          <message>
+            <%s xmlns="http://jabber.org/protocol/chatstates" />
+          </message>
+        """
+
+        msg = self.Message()
+
+        self.assertEqual(msg['chat_state'], '')
+        self.check(msg, "<message />", use_values=False)
+
+        msg['chat_state'] = 'active'
+        self.check(msg, xmlstring % 'active', use_values=False)
+
+        msg['chat_state'] = 'composing'
+        self.check(msg, xmlstring % 'composing', use_values=False)
+
+        msg['chat_state'] = 'gone'
+        self.check(msg, xmlstring % 'gone', use_values=False)
+
+        msg['chat_state'] = 'inactive'
+        self.check(msg, xmlstring % 'inactive', use_values=False)
+
+        msg['chat_state'] = 'paused'
+        self.check(msg, xmlstring % 'paused', use_values=False)
+
+        del msg['chat_state']
+        self.check(msg, "<message />")
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestChatStates)