From 77251452c106618ab1cfdad546eb224bc9693dea Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Thu, 24 Feb 2011 12:10:29 -0500
Subject: Updated the XEP-0085 plugin.

Can now be used as so:

>>> msg['chat_state']
''
>>> msg
<message />

>>> msg['chat_state'] = 'paused'
>>> msg
<message>
  <paused xmlns="http://jabber.org/protocol/chatstates" />
</message>

>>> msg['chat_state']
'paused'

>>> del msg['chat_state']
>>> msg
<message />
---
 tests/test_stanza_xep_0085.py | 37 +++++++++++++++++--------------------
 1 file changed, 17 insertions(+), 20 deletions(-)

(limited to 'tests')

diff --git a/tests/test_stanza_xep_0085.py b/tests/test_stanza_xep_0085.py
index 5db7139a..b08404e2 100644
--- a/tests/test_stanza_xep_0085.py
+++ b/tests/test_stanza_xep_0085.py
@@ -4,11 +4,7 @@ import sleekxmpp.plugins.xep_0085 as xep_0085
 class TestChatStates(SleekTest):
 
     def setUp(self):
-        register_stanza_plugin(Message, xep_0085.Active)
-        register_stanza_plugin(Message, xep_0085.Composing)
-        register_stanza_plugin(Message, xep_0085.Gone)
-        register_stanza_plugin(Message, xep_0085.Inactive)
-        register_stanza_plugin(Message, xep_0085.Paused)
+        register_stanza_plugin(Message, xep_0085.ChatState)
 
     def testCreateChatState(self):
         """Testing creating chat states."""
@@ -20,25 +16,26 @@ class TestChatStates(SleekTest):
         """
 
         msg = self.Message()
-        msg['chat_state'].active()
-        self.check(msg, xmlstring % 'active',
-                          use_values=False)
 
-        msg['chat_state'].composing()
-        self.check(msg, xmlstring % 'composing',
-                          use_values=False)
+        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'].gone()
-        self.check(msg, xmlstring % 'gone',
-                          use_values=False)
+        msg['chat_state'] = 'composing'
+        self.check(msg, xmlstring % 'composing', use_values=False)
 
-        msg['chat_state'].inactive()
-        self.check(msg, xmlstring % 'inactive',
-                          use_values=False)
+        msg['chat_state'] = 'gone'
+        self.check(msg, xmlstring % 'gone', use_values=False)
 
-        msg['chat_state'].paused()
-        self.check(msg, xmlstring % 'paused',
-                          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)
-- 
cgit v1.2.3


From 1a81b2f464a3f35c99ff1659b7debb6006004d4f Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Thu, 24 Feb 2011 14:15:02 -0500
Subject: Add tests for XEP-0085, fix some bugs.

---
 tests/test_stream_xep_0085.py | 59 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 tests/test_stream_xep_0085.py

(limited to 'tests')

diff --git a/tests/test_stream_xep_0085.py b/tests/test_stream_xep_0085.py
new file mode 100644
index 00000000..2a814805
--- /dev/null
+++ b/tests/test_stream_xep_0085.py
@@ -0,0 +1,59 @@
+import threading
+import time
+
+from sleekxmpp.test import *
+
+
+class TestStreamChatStates(SleekTest):
+
+    def tearDown(self):
+        self.stream_close()
+
+    def testChatStates(self):
+        self.stream_start(mode='client', plugins=['xep_0030', 'xep_0085'])
+
+        results = []
+
+        def handle_state(msg):
+            results.append(msg['chat_state'])
+
+        self.xmpp.add_event_handler('chatstate_active', handle_state)
+        self.xmpp.add_event_handler('chatstate_inactive', handle_state)
+        self.xmpp.add_event_handler('chatstate_paused', handle_state)
+        self.xmpp.add_event_handler('chatstate_gone', handle_state)
+        self.xmpp.add_event_handler('chatstate_composing', handle_state)
+
+        self.recv("""
+          <message>
+            <active xmlns="http://jabber.org/protocol/chatstates" />
+          </message>
+        """)
+        self.recv("""
+          <message>
+            <inactive xmlns="http://jabber.org/protocol/chatstates" />
+          </message>
+        """)
+        self.recv("""
+          <message>
+            <paused xmlns="http://jabber.org/protocol/chatstates" />
+          </message>
+        """)
+        self.recv("""
+          <message>
+            <composing xmlns="http://jabber.org/protocol/chatstates" />
+          </message>
+        """)
+        self.recv("""
+          <message>
+            <gone xmlns="http://jabber.org/protocol/chatstates" />
+          </message>
+        """)
+
+        # Give event queue time to process
+        time.sleep(0.3)
+        expected = ['active', 'inactive', 'paused', 'composing', 'gone']
+        self.failUnless(results == expected,
+                "Chat state event not handled: %s" % results)
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamChatStates)
-- 
cgit v1.2.3