82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
import unittest
|
|
from sleekxmpp.test import SleekTest
|
|
|
|
|
|
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)
|