OpenSource/python-telegram-bot
OpenSource/python-telegram-bot
1
0
Fork 0
mirror of https://github.com/python-telegram-bot/python-telegram-bot.git synced 2024-12-23 06:50:29 +01:00
Code Issues Projects Releases Packages Wiki Activity

test error in handler, other webhook server functionalites

Browse source
This commit is contained in:
Jannes Höke 2015-11-24 19:35:02 +01:00
parent 8090658ab4
commit f21b6046c5
1 changed files with 59 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

85
tests/test_boteventhandler.py
View file

@ -87,8 +87,11 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
elif args[0] == 'noresend': elif args[0] == 'noresend':
pass pass
def errorRaisingHandlerTest(self, bot, update):
raise TelegramError(update)
def errorHandlerTest(self, bot, update, error): def errorHandlerTest(self, bot, update, error):
self.received_message = error self.received_message = error.message
self.message_count += 1 self.message_count += 1
def test_addTelegramMessageHandler(self): def test_addTelegramMessageHandler(self):
@ -96,7 +99,7 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.updater.bot = MockBot('Test') self.updater.bot = MockBot('Test')
self.updater.dispatcher.addTelegramMessageHandler( self.updater.dispatcher.addTelegramMessageHandler(
self.telegramHandlerTest) self.telegramHandlerTest)
self.updater.start_polling(0.05) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, 'Test') self.assertEqual(self.received_message, 'Test')
@ -115,7 +118,7 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.updater.bot = MockBot('Test2') self.updater.bot = MockBot('Test2')
self.updater.dispatcher.addTelegramRegexHandler(re.compile('Te.*'), self.updater.dispatcher.addTelegramRegexHandler(re.compile('Te.*'),
self.telegramHandlerTest) self.telegramHandlerTest)
self.updater.start_polling(0.05) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, 'Test2') self.assertEqual(self.received_message, 'Test2')
@ -124,7 +127,7 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.updater.bot = MockBot('/test') self.updater.bot = MockBot('/test')
self.updater.dispatcher.addTelegramCommandHandler( self.updater.dispatcher.addTelegramCommandHandler(
'test', self.telegramHandlerTest) 'test', self.telegramHandlerTest)
self.updater.start_polling(0.05) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, '/test') self.assertEqual(self.received_message, '/test')
@ -133,27 +136,27 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.updater.bot = MockBot('/test2') self.updater.bot = MockBot('/test2')
self.updater.dispatcher.addUnknownTelegramCommandHandler( self.updater.dispatcher.addUnknownTelegramCommandHandler(
self.telegramHandlerTest) self.telegramHandlerTest)
self.updater.start_polling(0.05) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, '/test2') self.assertEqual(self.received_message, '/test2')
def test_addStringRegexHandler(self): def test_addStringRegexHandler(self):
print('Testing addStringRegexHandler') print('Testing addStringRegexHandler')
self.updater.bot = MockBot('') self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addStringRegexHandler(re.compile('Te.*'), self.updater.dispatcher.addStringRegexHandler(re.compile('Te.*'),
self.stringHandlerTest) self.stringHandlerTest)
queue = self.updater.start_polling(0.05) queue = self.updater.start_polling(0.01)
queue.put('Test3') queue.put('Test3')
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, 'Test3') self.assertEqual(self.received_message, 'Test3')
def test_addStringCommandHandler(self): def test_addStringCommandHandler(self):
print('Testing addStringCommandHandler') print('Testing addStringCommandHandler')
self.updater.bot = MockBot('') self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addStringCommandHandler( self.updater.dispatcher.addStringCommandHandler(
'test3', self.stringHandlerTest) 'test3', self.stringHandlerTest)
queue = self.updater.start_polling(0.05) queue = self.updater.start_polling(0.01)
queue.put('/test3') queue.put('/test3')
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, '/test3') self.assertEqual(self.received_message, '/test3')
@ -163,34 +166,46 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.updater.bot = MockBot('/test') self.updater.bot = MockBot('/test')
self.updater.dispatcher.addUnknownStringCommandHandler( self.updater.dispatcher.addUnknownStringCommandHandler(
self.stringHandlerTest) self.stringHandlerTest)
queue = self.updater.start_polling(0.05) queue = self.updater.start_polling(0.01)
queue.put('/test4') queue.put('/test4')
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, '/test4') self.assertEqual(self.received_message, '/test4')
def test_addErrorHandler(self): def test_addErrorHandler(self):
print('Testing addErrorHandler') print('Testing addErrorHandler')
self.updater.bot = MockBot('') self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addErrorHandler(self.errorHandlerTest) self.updater.dispatcher.addErrorHandler(self.errorHandlerTest)
queue = self.updater.start_polling(0.05) queue = self.updater.start_polling(0.01)
error = TelegramError("Unauthorized.") error = TelegramError("Unauthorized.")
queue.put(error) queue.put(error)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, error) self.assertEqual(self.received_message, "Unauthorized.")
def test_errorInHandler(self):
print('Testing error in Handler')
self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addStringRegexHandler('.*',
self.errorRaisingHandlerTest)
self.updater.dispatcher.addErrorHandler(self.errorHandlerTest)
queue = self.updater.start_polling(0.01)
queue.put('Test Error 1')
sleep(.1)
self.assertEqual(self.received_message, 'Test Error 1')
def test_errorOnGetUpdates(self): def test_errorOnGetUpdates(self):
print('Testing errorOnGetUpdates') print('Testing error on getUpdates')
self.updater.bot = MockBot('', raise_error=True) self.updater.bot = MockBot('', raise_error=True)
self.updater.dispatcher.addErrorHandler(self.errorHandlerTest) self.updater.dispatcher.addErrorHandler(self.errorHandlerTest)
self.updater.start_polling(0.05) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message.message, "Test Error") self.assertEqual(self.received_message, "Test Error 2")
def test_addTypeHandler(self): def test_addTypeHandler(self):
print('Testing addTypeHandler') print('Testing addTypeHandler')
self.updater.bot = MockBot('') self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addTypeHandler(dict, self.stringHandlerTest) self.updater.dispatcher.addTypeHandler(dict, self.stringHandlerTest)
queue = self.updater.start_polling(0.05) queue = self.updater.start_polling(0.01)
payload = {"Test": 42} payload = {"Test": 42}
queue.put(payload) queue.put(payload)
sleep(.1) sleep(.1)
@ -198,29 +213,29 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
def test_runAsync(self): def test_runAsync(self):
print('Testing @run_async') print('Testing @run_async')
self.updater.bot = MockBot('Test4', messages=2) self.updater.bot = MockBot('Test5', messages=2)
self.updater.dispatcher.addTelegramMessageHandler( self.updater.dispatcher.addTelegramMessageHandler(
self.asyncHandlerTest) self.asyncHandlerTest)
self.updater.start_polling(0.01) self.updater.start_polling(0.01)
sleep(1.2) sleep(1.2)
self.assertEqual(self.received_message, 'Test4') self.assertEqual(self.received_message, 'Test5')
self.assertEqual(self.message_count, 2) self.assertEqual(self.message_count, 2)
def test_additionalArgs(self): def test_additionalArgs(self):
print('Testing additional arguments for handlers') print('Testing additional arguments for handlers')
self.updater.bot = MockBot('') self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addStringCommandHandler( self.updater.dispatcher.addStringCommandHandler(
'test5', self.additionalArgsTest) 'test5', self.additionalArgsTest)
queue = self.updater.start_polling(0.05) queue = self.updater.start_polling(0.01)
queue.put('/test5 resend') queue.put('/test5 resend')
sleep(.2) sleep(.1)
self.assertEqual(self.received_message, '/test5 noresend') self.assertEqual(self.received_message, '/test5 noresend')
self.assertEqual(self.message_count, 2) self.assertEqual(self.message_count, 2)
def test_webhook(self): def test_webhook(self):
print('Testing Webhook') print('Testing Webhook')
self.updater.bot = MockBot('Test4', messages=2) self.updater.bot = MockBot('', messages=0)
self.updater.dispatcher.addTelegramMessageHandler( self.updater.dispatcher.addTelegramMessageHandler(
self.telegramHandlerTest) self.telegramHandlerTest)
@ -261,13 +276,31 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
sleep(1) sleep(1)
self.assertEqual(self.received_message, 'Webhook Test') self.assertEqual(self.received_message, 'Webhook Test')
print("Test other webhook server functionalites...")
request = Request('http://localhost:%d/webookhandler.py' % port)
response = urlopen(request)
self.assertEqual(b'', response.read())
self.assertEqual(200, response.code)
request.get_method = lambda: 'HEAD'
response = urlopen(request)
self.assertEqual(b'', response.read())
self.assertEqual(200, response.code)
# Test multiple shutdown() calls
self.updater.httpd.shutdown()
self.updater.httpd.shutdown()
self.assertTrue(True)
def signalsender(self): def signalsender(self):
sleep(0.5) sleep(0.5)
os.kill(os.getpid(), signal.SIGTERM) os.kill(os.getpid(), signal.SIGTERM)
def test_idle(self): def test_idle(self):
print('Testing idle') print('Testing idle')
self.updater.bot = MockBot('Test4', messages=0) self.updater.bot = MockBot('Test6', messages=0)
self.updater.start_polling(poll_interval=0.01) self.updater.start_polling(poll_interval=0.01)
Thread(target=self.signalsender).start() Thread(target=self.signalsender).start()
self.updater.idle() self.updater.idle()
@ -303,7 +336,7 @@ class MockBot:
network_delay=2.): network_delay=2.):
if self.raise_error: if self.raise_error:
raise TelegramError('Test Error') raise TelegramError('Test Error 2')
elif self.send_messages >= 2: elif self.send_messages >= 2:
self.send_messages -= 2 self.send_messages -= 2
return self.mockUpdate(self.text), self.mockUpdate(self.text) return self.mockUpdate(self.text), self.mockUpdate(self.text)