tests/test_stanza_error.py

   1 from sleekxmpp.test import *
   2
   3
   4 class TestErrorStanzas(SleekTest):
   5
   6     def testSetup(self):
   7         """Test setting initial values in error stanza."""
   8         msg = self.Message()
   9         msg.enable('error')
  10         self.check(msg, """
  11           <message type="error">
  12             <error type="cancel">
  13               <feature-not-implemented xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
  14             </error>
  15           </message>
  16         """)
  17
  18     def testCondition(self):
  19         """Test modifying the error condition."""
  20         msg = self.Message()
  21         msg['error']['condition'] = 'item-not-found'
  22
  23         self.check(msg, """
  24           <message type="error">
  25             <error type="cancel">
  26               <item-not-found xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
  27             </error>
  28           </message>
  29         """)
  30
  31         self.failUnless(msg['error']['condition'] == 'item-not-found', "Error condition doesn't match.")
  32
  33         msg['error']['condition'] = 'resource-constraint'
  34
  35         self.check(msg, """
  36           <message type="error">
  37             <error type="cancel">
  38               <resource-constraint xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
  39             </error>
  40           </message>
  41          """)
  42
  43     def testDelCondition(self):
  44         """Test that deleting error conditions doesn't remove extra elements."""
  45         msg = self.Message()
  46         msg['error']['text'] = 'Error!'
  47         msg['error']['condition'] = 'internal-server-error'
  48
  49         del msg['error']['condition']
  50
  51         self.check(msg, """
  52           <message type="error">
  53             <error type="cancel">
  54               <text xmlns="urn:ietf:params:xml:ns:xmpp-stanzas">Error!</text>
  55             </error>
  56           </message>
  57         """, use_values=False)
  58
  59     def testDelText(self):
  60         """Test deleting the text of an error."""
  61         msg = self.Message()
  62         msg['error']['test'] = 'Error!'
  63         msg['error']['condition'] = 'internal-server-error'
  64
  65         del msg['error']['text']
  66
  67         self.check(msg, """
  68           <message type="error">
  69             <error type="cancel">
  70               <internal-server-error xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
  71             </error>
  72           </message>
  73         """)
  74
  75
  76 suite = unittest.TestLoader().loadTestsFromTestCase(TestErrorStanzas)