mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-11-23 07:38:58 +01:00
208 lines
7.6 KiB
Python
208 lines
7.6 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
#
|
|
# A library that provides a Python interface to the Telegram Bot API
|
|
# Copyright (C) 2015-2016
|
|
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see [http://www.gnu.org/licenses/].
|
|
|
|
"""This module contains a object that represents Tests for Telegram Bot"""
|
|
|
|
import io
|
|
import os
|
|
from datetime import datetime
|
|
import sys
|
|
from flaky import flaky
|
|
|
|
if sys.version_info[0:2] == (2, 6):
|
|
import unittest2 as unittest
|
|
else:
|
|
import unittest
|
|
|
|
sys.path.append('.')
|
|
|
|
import telegram
|
|
from tests.base import BaseTest, timeout
|
|
|
|
|
|
class BotTest(BaseTest, unittest.TestCase):
|
|
"""This object represents Tests for Telegram Bot."""
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testGetMe(self):
|
|
bot = self._bot.getMe()
|
|
|
|
self.assertTrue(self.is_json(bot.to_json()))
|
|
self.assertEqual(bot.id, 133505823)
|
|
self.assertEqual(bot.first_name, 'PythonTelegramBot')
|
|
self.assertEqual(bot.last_name, '')
|
|
self.assertEqual(bot.username, 'PythonTelegramBot')
|
|
self.assertEqual(bot.name, '@PythonTelegramBot')
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendMessage(self):
|
|
message = self._bot.sendMessage(chat_id=self._chat_id,
|
|
text='Моё судно на воздушной подушке полно угрей')
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.text, u'Моё судно на воздушной подушке полно угрей')
|
|
self.assertTrue(isinstance(message.date, datetime))
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSilentSendMessage(self):
|
|
message = self._bot.sendMessage(chat_id=self._chat_id,
|
|
text='Моё судно на воздушной подушке полно угрей',
|
|
disable_notification=True)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.text, u'Моё судно на воздушной подушке полно угрей')
|
|
self.assertTrue(isinstance(message.date, datetime))
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testGetUpdates(self):
|
|
updates = self._bot.getUpdates()
|
|
|
|
if updates:
|
|
self.assertTrue(self.is_json(updates[0].to_json()))
|
|
self.assertTrue(isinstance(updates[0], telegram.Update))
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testForwardMessage(self):
|
|
message = self._bot.forwardMessage(chat_id=self._chat_id,
|
|
from_chat_id=self._chat_id,
|
|
message_id=2398)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.text, 'teste')
|
|
self.assertEqual(message.forward_from.username, 'leandrotoledo')
|
|
self.assertTrue(isinstance(message.forward_date, datetime))
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendPhoto(self):
|
|
message = self._bot.sendPhoto(photo=open('tests/data/telegram.png', 'rb'),
|
|
caption='testSendPhoto',
|
|
chat_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_size, 1451)
|
|
self.assertEqual(message.caption, 'testSendPhoto')
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSilentSendPhoto(self):
|
|
message = self._bot.sendPhoto(photo=open('tests/data/telegram.png', 'rb'),
|
|
caption='testSendPhoto',
|
|
chat_id=self._chat_id,
|
|
disable_notification=True)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_size, 1451)
|
|
self.assertEqual(message.caption, 'testSendPhoto')
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testResendPhoto(self):
|
|
message = self._bot.sendPhoto(photo='AgADAQADyKcxGx8j9Qdp6d-gpUsw4Gja1i8ABEVJsVqQk8LfJ3wAAgI',
|
|
chat_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_id, 'AgADAQADyKcxGx8j9Qdp6d-gpUsw4Gja1i8ABEVJsVqQk8LfJ3wAAgI')
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendJPGURLPhoto(self):
|
|
message = self._bot.sendPhoto(photo='http://dummyimage.com/600x400/000/fff.jpg&text=telegram',
|
|
chat_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_size, 822)
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendPNGURLPhoto(self):
|
|
message = self._bot.sendPhoto(photo='http://dummyimage.com/600x400/000/fff.png&text=telegram',
|
|
chat_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_size, 684)
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendGIFURLPhoto(self):
|
|
message = self._bot.sendPhoto(photo='http://dummyimage.com/600x400/000/fff.gif&text=telegram',
|
|
chat_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_size, 684)
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendBufferedReaderPhoto(self):
|
|
photo = open('tests/data/telegram.png', 'rb')
|
|
br_photo = io.BufferedReader(io.BytesIO(photo.read()))
|
|
message = self._bot.sendPhoto(photo=br_photo,
|
|
chat_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(message.to_json()))
|
|
self.assertEqual(message.photo[0].file_size, 1451)
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testSendChatAction(self):
|
|
self._bot.sendChatAction(action=telegram.ChatAction.TYPING,
|
|
chat_id=self._chat_id)
|
|
|
|
@flaky(3, 1)
|
|
@timeout(10)
|
|
def testGetUserProfilePhotos(self):
|
|
upf = self._bot.getUserProfilePhotos(user_id=self._chat_id)
|
|
|
|
self.assertTrue(self.is_json(upf.to_json()))
|
|
self.assertEqual(upf.photos[0][0].file_size, 12421)
|
|
|
|
def _test_invalid_token(self, token):
|
|
self.assertRaisesRegexp(telegram.error.InvalidToken, 'Invalid token', telegram.Bot, token)
|
|
|
|
def testInvalidToken1(self):
|
|
self._test_invalid_token('123')
|
|
|
|
def testInvalidToken2(self):
|
|
self._test_invalid_token('12a:')
|
|
|
|
def testInvalidToken3(self):
|
|
self._test_invalid_token('12:')
|
|
|
|
def testUnauthToken(self):
|
|
with self.assertRaisesRegexp(telegram.error.Unauthorized, 'Unauthorized'):
|
|
bot = telegram.Bot('1234:abcd1234')
|
|
bot.getMe()
|
|
|
|
def testInvalidSrvResp(self):
|
|
with self.assertRaisesRegexp(telegram.TelegramError, 'Invalid server response'):
|
|
# bypass the valid token check
|
|
bot = telegram.Bot.__new__(telegram.Bot)
|
|
bot.base_url = 'https://api.telegram.org/bot{0}'.format('12')
|
|
|
|
bot.getMe()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|