from sleekxmpp.test import * class TestErrorStanzas(SleekTest): def setUp(self): # Ensure that the XEP-0086 plugin has been loaded. self.stream_start() self.stream_close() def testSetup(self): """Test setting initial values in error stanza.""" msg = self.Message() msg.enable('error') self.check(msg, """ <message type="error"> <error type="cancel" code="501"> <feature-not-implemented xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" /> </error> </message> """) def testCondition(self): """Test modifying the error condition.""" msg = self.Message() msg['error']['condition'] = 'item-not-found' self.check(msg, """ <message type="error"> <error type="cancel" code="404"> <item-not-found xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" /> </error> </message> """) self.failUnless(msg['error']['condition'] == 'item-not-found', "Error condition doesn't match.") msg['error']['condition'] = 'resource-constraint' self.check(msg, """ <message type="error"> <error type="wait" code="500"> <resource-constraint xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" /> </error> </message> """) def testDelCondition(self): """Test that deleting error conditions doesn't remove extra elements.""" msg = self.Message() msg['error']['text'] = 'Error!' msg['error']['condition'] = 'internal-server-error' del msg['error']['condition'] self.check(msg, """ <message type="error"> <error type="wait" code="500"> <text xmlns="urn:ietf:params:xml:ns:xmpp-stanzas">Error!</text> </error> </message> """, use_values=False) def testDelText(self): """Test deleting the text of an error.""" msg = self.Message() msg['error']['test'] = 'Error!' msg['error']['condition'] = 'internal-server-error' del msg['error']['text'] self.check(msg, """ <message type="error"> <error type="wait" code="500"> <internal-server-error xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" /> </error> </message> """) suite = unittest.TestLoader().loadTestsFromTestCase(TestErrorStanzas)