Merge branch 'develop' into exceptions
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("""
|
||||
|
||||
72
tests/test_stream_xep_0066.py
Normal file
72
tests/test_stream_xep_0066.py
Normal file
@@ -0,0 +1,72 @@
|
||||
import time
|
||||
import threading
|
||||
|
||||
from sleekxmpp.test import *
|
||||
|
||||
|
||||
class TestOOB(SleekTest):
|
||||
|
||||
def tearDown(self):
|
||||
self.stream_close()
|
||||
|
||||
def testSendOOB(self):
|
||||
"""Test sending an OOB transfer request."""
|
||||
self.stream_start(plugins=['xep_0066', 'xep_0030'])
|
||||
|
||||
url = 'http://github.com/fritzy/SleekXMPP/blob/master/README'
|
||||
|
||||
t = threading.Thread(
|
||||
name='send_oob',
|
||||
target=self.xmpp['xep_0066'].send_oob,
|
||||
args=('user@example.com', url),
|
||||
kwargs={'desc': 'SleekXMPP README'})
|
||||
|
||||
t.start()
|
||||
|
||||
self.send("""
|
||||
<iq to="user@example.com" type="set" id="1">
|
||||
<query xmlns="jabber:iq:oob">
|
||||
<url>http://github.com/fritzy/SleekXMPP/blob/master/README</url>
|
||||
<desc>SleekXMPP README</desc>
|
||||
</query>
|
||||
</iq>
|
||||
""")
|
||||
|
||||
self.recv("""
|
||||
<iq id="1" type="result"
|
||||
to="tester@localhost"
|
||||
from="user@example.com" />
|
||||
""")
|
||||
|
||||
t.join()
|
||||
|
||||
def testReceiveOOB(self):
|
||||
"""Test receiving an OOB request."""
|
||||
self.stream_start(plugins=['xep_0066', 'xep_0030'])
|
||||
|
||||
events = []
|
||||
|
||||
def receive_oob(iq):
|
||||
events.append(iq['oob_transfer']['url'])
|
||||
|
||||
self.xmpp.add_event_handler('oob_transfer', receive_oob)
|
||||
|
||||
self.recv("""
|
||||
<iq to="tester@localhost"
|
||||
from="user@example.com"
|
||||
type="set" id="1">
|
||||
<query xmlns="jabber:iq:oob">
|
||||
<url>http://github.com/fritzy/SleekXMPP/blob/master/README</url>
|
||||
<desc>SleekXMPP README</desc>
|
||||
</query>
|
||||
</iq>
|
||||
""")
|
||||
|
||||
time.sleep(0.1)
|
||||
|
||||
self.assertEqual(events,
|
||||
['http://github.com/fritzy/SleekXMPP/blob/master/README'],
|
||||
'URL was not received: %s' % events)
|
||||
|
||||
|
||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestOOB)
|
||||
@@ -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