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 2025-01-04 02:01:48 +01:00
Code Issues Projects Releases Packages Wiki Activity

tests

Browse source
This commit is contained in:
Jannes Höke 2016-04-18 18:50:49 +02:00
parent b6fceefc80
commit de5619f3ca
1 changed files with 55 additions and 115 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

170
tests/test_updater.py
View file

@ -48,7 +48,7 @@ except ImportError:
sys.path.append('.') sys.path.append('.')
from telegram import Update, Message, TelegramError, User, Chat, Bot from telegram import Update, Message, TelegramError, User, Chat, Bot
from telegram.ext import Updater from telegram.ext import *
from telegram.ext.dispatcher import run_async from telegram.ext.dispatcher import run_async
from telegram.error import Unauthorized, InvalidToken from telegram.error import Unauthorized, InvalidToken
from tests.base import BaseTest from tests.base import BaseTest
@ -60,7 +60,8 @@ root.setLevel(logging.INFO)
ch = logging.StreamHandler(sys.stdout) ch = logging.StreamHandler(sys.stdout)
ch.setLevel(logging.WARN) ch.setLevel(logging.WARN)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s '
'- %(message)s')
ch.setFormatter(formatter) ch.setFormatter(formatter)
root.addHandler(ch) root.addHandler(ch)
@ -99,7 +100,7 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self.message_count += 1 self.message_count += 1
@run_async @run_async
def asyncHandlerTest(self, bot, update, **kwargs): def asyncHandlerTest(self, bot, update):
sleep(1) sleep(1)
with self.lock: with self.lock:
self.received_message = update.message.text self.received_message = update.message.text
@ -109,7 +110,7 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self.received_message = update self.received_message = update
self.message_count += 1 self.message_count += 1
def regexGroupHandlerTest(self, bot, update, groups=None, groupdict=None): def regexGroupHandlerTest(self, bot, update, groups, groupdict):
self.received_message = (groups, groupdict) self.received_message = (groups, groupdict)
self.message_count += 1 self.message_count += 1
@ -121,14 +122,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
elif args[0] == 'noresend': elif args[0] == 'noresend':
pass pass
def contextTest(self, bot, update, context):
self.received_message = update
self.message_count += 1
self.context = context
@run_async @run_async
def asyncAdditionalHandlerTest(self, bot, update, update_queue=None, def asyncAdditionalHandlerTest(self, bot, update, update_queue=None):
**kwargs):
sleep(1) sleep(1)
with self.lock: with self.lock:
if update_queue is not None: if update_queue is not None:
@ -142,37 +137,18 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self.received_message = error.message self.received_message = error.message
self.message_count += 1 self.message_count += 1
def test_importLegacyUpdater(self):
from telegram import Updater as legacyUpdater
lu = legacyUpdater(workers=2, bot=Bot('123:abcd'))
self.assertIsInstance(lu, Updater)
lu.stop()
def test_importLegacyDispatcher(self):
from telegram.ext import Dispatcher
from telegram.utils.updatequeue import UpdateQueue
from telegram import Dispatcher as legacyDispatcher
lp = legacyDispatcher(bot=Bot('123:abcd'), update_queue=UpdateQueue())
self.assertIsInstance(lp, Dispatcher)
lp.stop()
def test_addRemoveTelegramMessageHandler(self): def test_addRemoveTelegramMessageHandler(self):
self._setup_updater('Test') self._setup_updater('Test')
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramMessageHandler( from telegram.ext import filters
self.telegramHandlerTest) handler = MessageHandler([filters.TEXT], self.telegramHandlerTest)
d.addHandler(handler)
self.updater.start_polling(0.01) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, 'Test') self.assertEqual(self.received_message, 'Test')
# Remove handler # Remove handler
d.removeTelegramMessageHandler(self.telegramHandlerTest) d.removeHandler(handler)
self.reset() self.reset()
self.updater.bot.send_messages = 1 self.updater.bot.send_messages = 1
@ -181,8 +157,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_addTelegramMessageHandlerMultipleMessages(self): def test_addTelegramMessageHandlerMultipleMessages(self):
self._setup_updater('Multiple', 100) self._setup_updater('Multiple', 100)
self.updater.dispatcher.addTelegramMessageHandler( self.updater.dispatcher.addHandler(
self.telegramHandlerTest) MessageHandler([], self.telegramHandlerTest))
self.updater.start_polling(0.0) self.updater.start_polling(0.0)
sleep(2) sleep(2)
self.assertEqual(self.received_message, 'Multiple') self.assertEqual(self.received_message, 'Multiple')
@ -192,14 +168,14 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self._setup_updater('Test2') self._setup_updater('Test2')
d = self.updater.dispatcher d = self.updater.dispatcher
regobj = re.compile('Te.*') regobj = re.compile('Te.*')
self.updater.dispatcher.addTelegramRegexHandler(regobj, handler = RegexHandler(regobj, self.telegramHandlerTest)
self.telegramHandlerTest) self.updater.dispatcher.addHandler(handler)
self.updater.start_polling(0.01) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, 'Test2') self.assertEqual(self.received_message, 'Test2')
# Remove handler # Remove handler
d.removeTelegramRegexHandler(regobj, self.telegramHandlerTest) d.removeHandler(handler)
self.reset() self.reset()
self.updater.bot.send_messages = 1 self.updater.bot.send_messages = 1
@ -209,31 +185,14 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_addRemoveTelegramCommandHandler(self): def test_addRemoveTelegramCommandHandler(self):
self._setup_updater('/test') self._setup_updater('/test')
d = self.updater.dispatcher d = self.updater.dispatcher
self.updater.dispatcher.addTelegramCommandHandler( handler = CommandHandler('test', self.telegramHandlerTest)
'test', self.telegramHandlerTest) self.updater.dispatcher.addHandler(handler)
self.updater.start_polling(0.01) self.updater.start_polling(0.01)
sleep(.1) sleep(.1)
self.assertEqual(self.received_message, '/test') self.assertEqual(self.received_message, '/test')
# Remove handler # Remove handler
d.removeTelegramCommandHandler('test', self.telegramHandlerTest) d.removeHandler(handler)
self.reset()
self.updater.bot.send_messages = 1
sleep(.1)
self.assertTrue(None is self.received_message)
def test_addRemoveUnknownTelegramCommandHandler(self):
self._setup_updater('/test2')
d = self.updater.dispatcher
self.updater.dispatcher.addUnknownTelegramCommandHandler(
self.telegramHandlerTest)
self.updater.start_polling(0.01)
sleep(.1)
self.assertEqual(self.received_message, '/test2')
# Remove handler
d.removeUnknownTelegramCommandHandler(self.telegramHandlerTest)
self.reset() self.reset()
self.updater.bot.send_messages = 1 self.updater.bot.send_messages = 1
@ -243,14 +202,15 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_addRemoveStringRegexHandler(self): def test_addRemoveStringRegexHandler(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addStringRegexHandler('Te.*', self.stringHandlerTest) handler = StringRegexHandler('Te.*', self.stringHandlerTest)
d.addHandler(handler)
queue = self.updater.start_polling(0.01) 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')
# Remove handler # Remove handler
d.removeStringRegexHandler('Te.*', self.stringHandlerTest) d.removeHandler(handler)
self.reset() self.reset()
queue.put('Test3') queue.put('Test3')
@ -260,8 +220,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_addRemoveStringCommandHandler(self): def test_addRemoveStringCommandHandler(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addStringCommandHandler( handler = StringCommandHandler('test3', self.stringHandlerTest)
'test3', self.stringHandlerTest) d.addHandler(handler)
queue = self.updater.start_polling(0.01) queue = self.updater.start_polling(0.01)
queue.put('/test3') queue.put('/test3')
@ -269,31 +229,13 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self.assertEqual(self.received_message, '/test3') self.assertEqual(self.received_message, '/test3')
# Remove handler # Remove handler
d.removeStringCommandHandler('test3', self.stringHandlerTest) d.removeHandler(handler)
self.reset() self.reset()
queue.put('/test3') queue.put('/test3')
sleep(.1) sleep(.1)
self.assertTrue(None is self.received_message) self.assertTrue(None is self.received_message)
def test_addRemoveUnknownStringCommandHandler(self):
self._setup_updater('/test')
d = self.updater.dispatcher
d.addUnknownStringCommandHandler(
self.stringHandlerTest)
queue = self.updater.start_polling(0.01)
queue.put('/test4')
sleep(.1)
self.assertEqual(self.received_message, '/test4')
# Remove handler
d.removeUnknownStringCommandHandler(self.stringHandlerTest)
self.reset()
self.updater.bot.send_messages = 1
sleep(.1)
self.assertTrue(None is self.received_message)
def test_addRemoveErrorHandler(self): def test_addRemoveErrorHandler(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
@ -315,7 +257,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_errorInHandler(self): def test_errorInHandler(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addStringRegexHandler('.*', self.errorRaisingHandlerTest) handler = StringRegexHandler('.*', self.errorRaisingHandlerTest)
d.addHandler(handler)
self.updater.dispatcher.addErrorHandler(self.errorHandlerTest) self.updater.dispatcher.addErrorHandler(self.errorHandlerTest)
queue = self.updater.start_polling(0.01) queue = self.updater.start_polling(0.01)
@ -326,7 +269,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_cleanBeforeStart(self): def test_cleanBeforeStart(self):
self._setup_updater('') self._setup_updater('')
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramMessageHandler(self.telegramHandlerTest) handler = MessageHandler([], self.telegramHandlerTest)
d.addHandler(handler)
self.updater.start_polling(0.01, clean=True) self.updater.start_polling(0.01, clean=True)
sleep(.1) sleep(.1)
self.assertEqual(self.message_count, 0) self.assertEqual(self.message_count, 0)
@ -343,7 +287,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_addRemoveTypeHandler(self): def test_addRemoveTypeHandler(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTypeHandler(dict, self.stringHandlerTest) handler = TypeHandler(dict, self.stringHandlerTest)
d.addHandler(handler)
queue = self.updater.start_polling(0.01) queue = self.updater.start_polling(0.01)
payload = {"Test": 42} payload = {"Test": 42}
queue.put(payload) queue.put(payload)
@ -351,18 +296,21 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self.assertEqual(self.received_message, payload) self.assertEqual(self.received_message, payload)
# Remove handler # Remove handler
d.removeTypeHandler(dict, self.stringHandlerTest) d.removeHandler(handler)
self.reset() self.reset()
queue.put(payload) queue.put(payload)
sleep(.1) sleep(.1)
self.assertTrue(None is self.received_message) self.assertTrue(None is self.received_message)
def test_addRemoveInlineHandlerQuery(self): def test_addRemoveInlineQueryHandler(self):
print('Testing add/removeInlineHandler') print('Testing add/remove InlineQueryHandler')
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramInlineHandler(self.telegramInlineHandlerTest) handler = InlineQueryHandler(self.telegramInlineHandlerTest)
handler2 = ChosenInlineResultHandler(self.telegramInlineHandlerTest)
d.addHandler(handler)
d.addHandler(handler2)
queue = self.updater.start_polling(0.01) queue = self.updater.start_polling(0.01)
update = Update(update_id=0, inline_query="testquery") update = Update(update_id=0, inline_query="testquery")
update2 = Update(update_id=0, chosen_inline_result="testresult") update2 = Update(update_id=0, chosen_inline_result="testresult")
@ -375,7 +323,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
self.assertEqual(self.received_message[1], "testresult") self.assertEqual(self.received_message[1], "testresult")
# Remove handler # Remove handler
d.removeTelegramInlineHandler(self.telegramInlineHandlerTest) d.removeHandler(handler)
d.removeHandler(handler2)
self.reset() self.reset()
queue.put(update) queue.put(update)
@ -385,8 +334,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_runAsync(self): def test_runAsync(self):
self._setup_updater('Test5', messages=2) self._setup_updater('Test5', messages=2)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramMessageHandler( handler = MessageHandler([], self.asyncHandlerTest)
self.asyncHandlerTest) d.addHandler(handler)
self.updater.start_polling(0.01) self.updater.start_polling(0.01)
sleep(1.2) sleep(1.2)
self.assertEqual(self.received_message, 'Test5') self.assertEqual(self.received_message, 'Test5')
@ -394,8 +343,9 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_additionalArgs(self): def test_additionalArgs(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
self.updater.dispatcher.addStringCommandHandler( handler = StringCommandHandler('test5', self.additionalArgsTest,
'test5', self.additionalArgsTest) pass_update_queue=True, pass_args=True)
self.updater.dispatcher.addHandler(handler)
queue = self.updater.start_polling(0.01) queue = self.updater.start_polling(0.01)
queue.put('/test5 resend') queue.put('/test5 resend')
@ -403,24 +353,13 @@ class UpdaterTest(BaseTest, unittest.TestCase):
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_context(self):
context = "context_data"
self._setup_updater('', messages=0)
self.updater.dispatcher.addStringCommandHandler(
'test_context', self.contextTest)
queue = self.updater.start_polling(0.01)
queue.put('/test_context', context=context)
sleep(.5)
self.assertEqual(self.received_message, '/test_context')
self.assertEqual(self.message_count, 1)
self.assertEqual(self.context, context)
def test_regexGroupHandler(self): def test_regexGroupHandler(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addStringRegexHandler('^(This).*?(?P<testgroup>regex group).*', handler = StringRegexHandler('^(This).*?(?P<testgroup>regex group).*',
self.regexGroupHandlerTest) self.regexGroupHandlerTest,
pass_groupdict=True, pass_groups=True)
d.addHandler(handler)
queue = self.updater.start_polling(0.01) queue = self.updater.start_polling(0.01)
queue.put('This is a test message for regex group matching.') queue.put('This is a test message for regex group matching.')
sleep(.1) sleep(.1)
@ -430,8 +369,9 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_runAsyncWithAdditionalArgs(self): def test_runAsyncWithAdditionalArgs(self):
self._setup_updater('Test6', messages=2) self._setup_updater('Test6', messages=2)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramMessageHandler( handler = MessageHandler([], self.asyncAdditionalHandlerTest,
self.asyncAdditionalHandlerTest) pass_update_queue=True)
d.addHandler(handler)
self.updater.start_polling(0.01) self.updater.start_polling(0.01)
sleep(1.2) sleep(1.2)
self.assertEqual(self.received_message, 'Test6') self.assertEqual(self.received_message, 'Test6')
@ -440,8 +380,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_webhook(self): def test_webhook(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramMessageHandler( handler = MessageHandler([], self.telegramHandlerTest)
self.telegramHandlerTest) d.addHandler(handler)
ip = '127.0.0.1' ip = '127.0.0.1'
port = randrange(1024, 49152) # Select random port for travis port = randrange(1024, 49152) # Select random port for travis
@ -485,8 +425,8 @@ class UpdaterTest(BaseTest, unittest.TestCase):
def test_webhook_no_ssl(self): def test_webhook_no_ssl(self):
self._setup_updater('', messages=0) self._setup_updater('', messages=0)
d = self.updater.dispatcher d = self.updater.dispatcher
d.addTelegramMessageHandler( handler = MessageHandler([], self.telegramHandlerTest)
self.telegramHandlerTest) d.addHandler(handler)
ip = '127.0.0.1' ip = '127.0.0.1'
port = randrange(1024, 49152) # Select random port for travis port = randrange(1024, 49152) # Select random port for travis