diff options
Diffstat (limited to 'tests/test_handlers.py')
-rw-r--r-- | tests/test_handlers.py | 112 |
1 files changed, 0 insertions, 112 deletions
diff --git a/tests/test_handlers.py b/tests/test_handlers.py deleted file mode 100644 index 70eb0248..00000000 --- a/tests/test_handlers.py +++ /dev/null @@ -1,112 +0,0 @@ -from . sleektest import * -import sleekxmpp -from sleekxmpp.xmlstream.handler import * -from sleekxmpp.xmlstream.matcher import * - -class TestHandlers(SleekTest): - """ - Test using handlers and waiters. - """ - - def setUp(self): - self.stream_start() - - def tearDown(self): - self.stream_close() - - def testCallback(self): - """Test using stream callback handlers.""" - - def callback_handler(stanza): - self.xmpp.sendRaw(""" - <message> - <body>Success!</body> - </message> - """) - - callback = Callback('Test Callback', - MatchXPath('{test}tester'), - callback_handler) - - self.xmpp.registerHandler(callback) - - self.stream_recv("""<tester xmlns="test" />""") - - msg = self.Message() - msg['body'] = 'Success!' - self.stream_send_message(msg) - - def testWaiter(self): - """Test using stream waiter handler.""" - - def waiter_handler(stanza): - iq = self.xmpp.Iq() - iq['id'] = 'test' - iq['type'] = 'set' - iq['query'] = 'test' - reply = iq.send(block=True) - if reply: - self.xmpp.sendRaw(""" - <message> - <body>Successful: %s</body> - </message> - """ % reply['query']) - - self.xmpp.add_event_handler('message', waiter_handler, threaded=True) - - # Send message to trigger waiter_handler - self.stream_recv(""" - <message> - <body>Testing</body> - </message> - """) - - # Check that Iq was sent by waiter_handler - iq = self.Iq() - iq['id'] = 'test' - iq['type'] = 'set' - iq['query'] = 'test' - self.stream_send_iq(iq) - - # Send the reply Iq - self.stream_recv(""" - <iq id="test" type="result"> - <query xmlns="test" /> - </iq> - """) - - # Check that waiter_handler received the reply - msg = self.Message() - msg['body'] = 'Successful: test' - self.stream_send_message(msg) - - def testWaiterTimeout(self): - """Test that waiter handler is removed after timeout.""" - - def waiter_handler(stanza): - iq = self.xmpp.Iq() - iq['id'] = 'test2' - iq['type'] = 'set' - iq['query'] = 'test2' - reply = iq.send(block=True, timeout=0) - - self.xmpp.add_event_handler('message', waiter_handler, threaded=True) - - # Start test by triggerig waiter_handler - self.stream_recv("""<message><body>Start Test</body></message>""") - - # Check that Iq was sent to trigger start of timeout period - iq = self.Iq() - iq['id'] = 'test2' - iq['type'] = 'set' - iq['query'] = 'test2' - self.stream_send_iq(iq) - - # Check that the waiter is no longer registered - waiter_exists = self.xmpp.removeHandler('IqWait_test2') - - self.failUnless(waiter_exists == False, - "Waiter handler was not removed.") - - -suite = unittest.TestLoader().loadTestsFromTestCase(TestHandlers) |