mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-11-21 22:56:38 +01:00
Adding tests
This commit is contained in:
parent
fdc913a580
commit
8c3ed7c629
3 changed files with 57 additions and 0 deletions
5
test.py
Normal file
5
test.py
Normal file
|
@ -0,0 +1,5 @@
|
|||
import unittest
|
||||
|
||||
if __name__ == '__main__':
|
||||
testsuite = unittest.TestLoader().discover('.')
|
||||
unittest.TextTestRunner(verbosity=1).run(testsuite)
|
0
tests/__init__.py
Normal file
0
tests/__init__.py
Normal file
52
tests/test_bot.py
Normal file
52
tests/test_bot.py
Normal file
|
@ -0,0 +1,52 @@
|
|||
#!/usr/bin/env python
|
||||
# encoding: utf-8
|
||||
|
||||
import os
|
||||
import telegram
|
||||
import unittest
|
||||
|
||||
from token import TOKEN
|
||||
|
||||
|
||||
@unittest.skipIf(not TOKEN, "No tokens provided")
|
||||
class BotTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
bot = telegram.Bot(token=os.environ.get('TOKEN'))
|
||||
self._bot = bot
|
||||
print 'Testing the Bot API class'
|
||||
|
||||
def testGetMe(self):
|
||||
'''Test the telegram.Bot getMe method'''
|
||||
print 'Testing getMe'
|
||||
user = self._bot.getMe()
|
||||
self.assertEqual(120405045, user.id)
|
||||
self.assertEqual('Toledo\'s Palace Bot', user.first_name)
|
||||
self.assertEqual(None, user.last_name)
|
||||
self.assertEqual('ToledosPalaceBot', user.username)
|
||||
|
||||
def testSendMessage(self):
|
||||
'''Test the telegram.Bot sendMessage method'''
|
||||
print 'Testing sendMessage'
|
||||
message = self._bot.sendMessage(chat_id=12173560,
|
||||
text=u'Моё судно на воздушной подушке полно угрей'.encode('utf8'))
|
||||
self.assertEqual(u'Моё судно на воздушной подушке полно угрей', message.text)
|
||||
|
||||
def testGetUpdates(self):
|
||||
'''Test the telegram.Bot getUpdates method'''
|
||||
print 'Testing getUpdates'
|
||||
updates = self._bot.getUpdates()
|
||||
self.assertEqual(129566481, updates[0].update_id)
|
||||
|
||||
def testForwardMessage(self):
|
||||
'''Test the telegram.Bot forwardMessage method'''
|
||||
print 'Testing forwardMessage'
|
||||
message = self._bot.forwardMessage(chat_id=12173560,
|
||||
from_chat_id=12173560,
|
||||
message_id=138)
|
||||
self.assertEqual(u'Oi', message.text)
|
||||
self.assertEqual(u'leandrotoledo', message.forward_from.username)
|
||||
|
||||
# def testSendPhoto(self):
|
||||
# '''Test the telegram.Bot sendPhoto method'''
|
||||
# print 'Testing sendPhoto'
|
||||
# message = self._bot.sendPhoto()
|
Loading…
Reference in a new issue