github/SleekXMPP
1
0
Fork 0
mirror of https://github.com/correl/SleekXMPP.git synced 2024-11-23 19:19:53 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add tests for XEP-0085, fix some bugs.

Browse source
This commit is contained in:
Lance Stout 2011-02-24 14:15:02 -05:00
parent 77251452c1
commit 1a81b2f464
2 changed files with 67 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
sleekxmpp/plugins/xep_0085/chat_states.py
View file

@ -14,7 +14,7 @@ from sleekxmpp.xmlstream.handler import Callback
from sleekxmpp.xmlstream.matcher import StanzaPath from sleekxmpp.xmlstream.matcher import StanzaPath
from sleekxmpp.xmlstream import register_stanza_plugin, ElementBase, ET from sleekxmpp.xmlstream import register_stanza_plugin, ElementBase, ET
from sleekxmpp.plugins.base import base_plugin from sleekxmpp.plugins.base import base_plugin
from sleekxmpp.plugins.xep_0085 import stanza from sleekxmpp.plugins.xep_0085 import stanza, ChatState
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -31,16 +31,17 @@ class xep_0085(base_plugin):
self.description = 'Chat State Notifications' self.description = 'Chat State Notifications'
self.stanza = stanza self.stanza = stanza
self.xmpp.registerHandler( for state in ChatState.states:
Callback('Chat States', self.xmpp.register_handler(
StanzaPath('message/chat_state'), Callback('Chat State: %s' % state,
StanzaPath('message@chat_state=%s' % state),
self._handle_chat_state)) self._handle_chat_state))
register_stanza_plugin(Message, stanza.ChatState) register_stanza_plugin(Message, ChatState)
def post_init(self): def post_init(self):
base_plugin.post_init(self) base_plugin.post_init(self)
self.xmpp.plugin['xep_0030'].add_feature(stanza.ChatState.namepsace) self.xmpp.plugin['xep_0030'].add_feature(ChatState.namespace)
def _handle_chat_state(self, msg): def _handle_chat_state(self, msg):
state = msg['chat_state'] state = msg['chat_state']

59
tests/test_stream_xep_0085.py Normal file
View file

@ -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)