tests/test_stanza_xep_0085.py

   1 import unittest
   2 from slixmpp import Message
   3 from slixmpp.test import SlixTest
   4 import slixmpp.plugins.xep_0085 as xep_0085
   5 from slixmpp.xmlstream import register_stanza_plugin
   6
   7
   8 class TestChatStates(SlixTest):
   9
  10     def setUp(self):
  11         register_stanza_plugin(Message, xep_0085.stanza.Active)
  12         register_stanza_plugin(Message, xep_0085.stanza.Composing)
  13         register_stanza_plugin(Message, xep_0085.stanza.Gone)
  14         register_stanza_plugin(Message, xep_0085.stanza.Inactive)
  15         register_stanza_plugin(Message, xep_0085.stanza.Paused)
  16
  17     def testCreateChatState(self):
  18         """Testing creating chat states."""
  19
  20         xmlstring = """
  21           <message>
  22             <%s xmlns="http://jabber.org/protocol/chatstates" />
  23           </message>
  24         """
  25
  26         msg = self.Message()
  27
  28         self.assertEqual(msg['chat_state'], '')
  29         self.check(msg, "<message />", use_values=False)
  30
  31         msg['chat_state'] = 'active'
  32         self.check(msg, xmlstring % 'active', use_values=False)
  33
  34         msg['chat_state'] = 'composing'
  35         self.check(msg, xmlstring % 'composing', use_values=False)
  36
  37         msg['chat_state'] = 'gone'
  38         self.check(msg, xmlstring % 'gone', use_values=False)
  39
  40         msg['chat_state'] = 'inactive'
  41         self.check(msg, xmlstring % 'inactive', use_values=False)
  42
  43         msg['chat_state'] = 'paused'
  44         self.check(msg, xmlstring % 'paused', use_values=False)
  45
  46         del msg['chat_state']
  47         self.check(msg, "<message />")
  48
  49 suite = unittest.TestLoader().loadTestsFromTestCase(TestChatStates)