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-11-22 15:17:00 +01:00
Code Issues Projects Releases Packages Wiki Activity

update tests

Browse source
This commit is contained in:
Jannes Höke 2015-11-21 15:45:45 +01:00
parent c1f194a310
commit 0006294f29
1 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/test_boteventhandler.py
View file

@ -28,7 +28,7 @@ import re
sys.path.append('.')
from telegram import Update, Message, TelegramError
from tests.base import BaseTest
from base import BaseTest
class BotEventHandlerTest(BaseTest, unittest.TestCase):
@ -41,6 +41,9 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.received_message = None
self.message_count = 0
def tearDown(self):
self.beh.stop()
def telegramHandlerTest(self, bot, update):
self.received_message = update.message.text
self.message_count += 1
@ -49,6 +52,10 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
self.received_message = update
self.message_count += 1
def errorHandlerTest(self, bot, update, error):
self.received_message = error
self.message_count += 1
def test_addTelegramMessageHandler(self):
print('Testing addTelegramMessageHandler')
self.beh.bot = MockBot('Test')
@ -129,7 +136,7 @@ class BotEventHandlerTest(BaseTest, unittest.TestCase):
def test_addErrorHandler(self):
print('Testing addErrorHandler')
self.beh.bot = MockBot('')
self.beh.broadcaster.addErrorHandler(self.stringHandlerTest)
self.beh.broadcaster.addErrorHandler(self.errorHandlerTest)
queue = self.beh.start_polling(0.05)
error = TelegramError("Unauthorized.")
queue.put(error)
@ -160,7 +167,11 @@ class MockBot:
update.message = message
return update
def getUpdates(self, offset):
def getUpdates(self,
offset=None,
limit=100,
timeout=0,
network_delay=2.):
if self.send_messages >= 2:
self.send_messages -= 2