49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import unittest
|
|
from sleekxmpp import Message
|
|
from sleekxmpp.test import SleekTest
|
|
import sleekxmpp.plugins.xep_0085 as xep_0085
|
|
from sleekxmpp.xmlstream import register_stanza_plugin
|
|
|
|
|
|
class TestChatStates(SleekTest):
|
|
|
|
def setUp(self):
|
|
register_stanza_plugin(Message, xep_0085.stanza.Active)
|
|
register_stanza_plugin(Message, xep_0085.stanza.Composing)
|
|
register_stanza_plugin(Message, xep_0085.stanza.Gone)
|
|
register_stanza_plugin(Message, xep_0085.stanza.Inactive)
|
|
register_stanza_plugin(Message, xep_0085.stanza.Paused)
|
|
|
|
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)
|