Merge branch 'develop' into roster
This commit is contained in:
@@ -12,7 +12,6 @@ class TestStreamExceptions(SleekTest):
|
||||
"""
|
||||
|
||||
def tearDown(self):
|
||||
sys.excepthook = sys.__excepthook__
|
||||
self.stream_close()
|
||||
|
||||
def testExceptionReply(self):
|
||||
@@ -23,8 +22,6 @@ class TestStreamExceptions(SleekTest):
|
||||
msg['body'] = 'Body changed'
|
||||
raise XMPPError(clear=False)
|
||||
|
||||
|
||||
sys.excepthook = lambda *args, **kwargs: None
|
||||
self.stream_start()
|
||||
self.xmpp.add_event_handler('message', message)
|
||||
|
||||
@@ -44,6 +41,49 @@ class TestStreamExceptions(SleekTest):
|
||||
</message>
|
||||
""")
|
||||
|
||||
def testExceptionContinueWorking(self):
|
||||
"""Test that Sleek continues to respond after an XMPPError is raised."""
|
||||
|
||||
def message(msg):
|
||||
msg.reply()
|
||||
msg['body'] = 'Body changed'
|
||||
raise XMPPError(clear=False)
|
||||
|
||||
self.stream_start()
|
||||
self.xmpp.add_event_handler('message', message)
|
||||
|
||||
self.recv("""
|
||||
<message>
|
||||
<body>This is going to cause an error.</body>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.send("""
|
||||
<message type="error">
|
||||
<body>This is going to cause an error.</body>
|
||||
<error type="cancel" code="500">
|
||||
<undefined-condition
|
||||
xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
|
||||
</error>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.recv("""
|
||||
<message>
|
||||
<body>This is going to cause an error.</body>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.send("""
|
||||
<message type="error">
|
||||
<body>This is going to cause an error.</body>
|
||||
<error type="cancel" code="500">
|
||||
<undefined-condition
|
||||
xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
|
||||
</error>
|
||||
</message>
|
||||
""")
|
||||
|
||||
def testXMPPErrorException(self):
|
||||
"""Test raising an XMPPError exception."""
|
||||
|
||||
@@ -153,9 +193,8 @@ class TestStreamExceptions(SleekTest):
|
||||
def catch_error(*args, **kwargs):
|
||||
raised_errors.append(True)
|
||||
|
||||
sys.excepthook = catch_error
|
||||
|
||||
self.stream_start()
|
||||
self.xmpp.exception = catch_error
|
||||
self.xmpp.add_event_handler('message', message)
|
||||
|
||||
self.recv("""
|
||||
@@ -178,6 +217,58 @@ class TestStreamExceptions(SleekTest):
|
||||
|
||||
self.assertEqual(raised_errors, [True], "Exception was not raised: %s" % raised_errors)
|
||||
|
||||
def testUnknownException(self):
|
||||
"""Test Sleek continues to respond after an unknown exception."""
|
||||
|
||||
raised_errors = []
|
||||
|
||||
def message(msg):
|
||||
raise ValueError("Did something wrong")
|
||||
|
||||
def catch_error(*args, **kwargs):
|
||||
raised_errors.append(True)
|
||||
|
||||
self.stream_start()
|
||||
self.xmpp.exception = catch_error
|
||||
self.xmpp.add_event_handler('message', message)
|
||||
|
||||
self.recv("""
|
||||
<message>
|
||||
<body>This is going to cause an error.</body>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.send("""
|
||||
<message type="error">
|
||||
<error type="cancel" code="500">
|
||||
<undefined-condition
|
||||
xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
|
||||
<text xmlns="urn:ietf:params:xml:ns:xmpp-stanzas">
|
||||
SleekXMPP got into trouble.
|
||||
</text>
|
||||
</error>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.recv("""
|
||||
<message>
|
||||
<body>This is going to cause an error.</body>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.send("""
|
||||
<message type="error">
|
||||
<error type="cancel" code="500">
|
||||
<undefined-condition
|
||||
xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" />
|
||||
<text xmlns="urn:ietf:params:xml:ns:xmpp-stanzas">
|
||||
SleekXMPP got into trouble.
|
||||
</text>
|
||||
</error>
|
||||
</message>
|
||||
""")
|
||||
|
||||
self.assertEqual(raised_errors, [True, True], "Exceptions were not raised: %s" % raised_errors)
|
||||
|
||||
|
||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamExceptions)
|
||||
|
@@ -12,7 +12,6 @@ class TestStreamDisco(SleekTest):
|
||||
"""
|
||||
|
||||
def tearDown(self):
|
||||
sys.excepthook = sys.__excepthook__
|
||||
self.stream_close()
|
||||
|
||||
def testInfoEmptyDefaultNode(self):
|
||||
@@ -531,11 +530,6 @@ class TestStreamDisco(SleekTest):
|
||||
|
||||
raised_exceptions = []
|
||||
|
||||
def catch_exception(*args, **kwargs):
|
||||
raised_exceptions.append(True)
|
||||
|
||||
sys.excepthook = catch_exception
|
||||
|
||||
self.stream_start(mode='client',
|
||||
plugins=['xep_0030', 'xep_0059'])
|
||||
|
||||
@@ -544,8 +538,14 @@ class TestStreamDisco(SleekTest):
|
||||
iterator=True)
|
||||
results.amount = 10
|
||||
|
||||
def run_test():
|
||||
try:
|
||||
results.next()
|
||||
except StopIteration:
|
||||
raised_exceptions.append(True)
|
||||
|
||||
t = threading.Thread(name="get_items_iterator",
|
||||
target=results.next)
|
||||
target=run_test)
|
||||
t.start()
|
||||
|
||||
self.send("""
|
||||
|
@@ -13,7 +13,6 @@ class TestStreamExtendedDisco(SleekTest):
|
||||
"""
|
||||
|
||||
def tearDown(self):
|
||||
sys.excepthook = sys.__excepthook__
|
||||
self.stream_close()
|
||||
|
||||
def testUsingExtendedInfo(self):
|
||||
|
@@ -13,7 +13,6 @@ class TestStreamDirectInvite(SleekTest):
|
||||
"""
|
||||
|
||||
def tearDown(self):
|
||||
sys.excepthook = sys.__excepthook__
|
||||
self.stream_close()
|
||||
|
||||
def testReceiveInvite(self):
|
||||
|
Reference in New Issue
Block a user