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-03-16 12:25:45 +01:00
Code Issues Projects Releases Packages Wiki Activity

test @run_async

Browse source
This commit is contained in:
Jannes Höke 2015-11-21 19:35:24 +01:00
parent 6b309397e8
commit 6626d4534e
1 changed files with 22 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/test_boteventhandler.py
View file

@ -28,7 +28,9 @@ import re
sys.path.append('.')
from telegram import Update, Message, TelegramError
from telegram.broadcaster import run_async
from tests.base import BaseTest
from threading import Lock
class BotEventHandlerTest(BaseTest, unittest.TestCase):
@ -36,10 +38,11 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
def setUp(self):
from telegram import BotEventHandler
self.beh = BotEventHandler('')
self.beh = BotEventHandler('', workers=2)
self.received_message = None
self.message_count = 0
self.lock = Lock()
def tearDown(self):
self.beh.stop()
@ -48,6 +51,13 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.received_message = update.message.text
self.message_count += 1
@run_async
def asyncHandlerTest(self, bot, update):
sleep(1)
with self.lock:
self.received_message = update.message.text
self.message_count += 1
def stringHandlerTest(self, bot, update):
self.received_message = update
self.message_count += 1
@ -153,6 +163,17 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
sleep(.1)
self.assertEqual(self.received_message, payload)
def test_runAsync(self):
print('Testing @run_async')
self.beh.bot = MockBot('Test4', messages=2)
self.beh.broadcaster.addTelegramMessageHandler(
self.asyncHandlerTest)
self.beh.start_polling(0.01)
sleep(1.2)
self.assertEqual(self.received_message, 'Test4')
self.assertEqual(self.message_count, 2)
class MockBot:
def __init__(self, text, messages=1):