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-17 04:39:55 +01:00
Code Issues Projects Releases Packages Wiki Activity

test_audio done

Browse source 
* Changed to new method of non-static file_id
* removed obsolete tests.
* changed some tests in all file
This commit is contained in:
Eldin 2017-06-21 17:59:35 +02:00
parent 281bd67298
commit c92ebc4046
6 changed files with 114 additions and 204 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

219
tests/test_audio.py
View file

@ -18,58 +18,58 @@
# along with this program. If not, see [http://www.gnu.org/licenses/].
"""This module contains an object that represents Tests for Telegram Audio"""
import sys
import unittest
import os
import unittest
from flaky import flaky
sys.path.append('.')
import telegram
from tests.base import BaseTest, timeout
from tests.bots import get_bot
class AudioTest(BaseTest, unittest.TestCase):
"""This object represents Tests for Telegram Audio."""
@classmethod
def setUpClass(cls):
cls.caption = "Test audio"
cls.performer = 'Leandro Toledo'
cls.title = 'Teste'
cls.audio_file_url = 'https://raw.githubusercontent.com/python-telegram-bot/python-telegram-bot/master/tests/data/telegram.mp3'
bot_info = get_bot()
cls._chat_id = bot_info['chat_id']
cls._bot = telegram.Bot(bot_info['token'])
audio_file = open('tests/data/telegram.mp3', 'rb')
audio = cls._bot.send_audio(cls._chat_id, audio=audio_file, timeout=10).audio
cls.audio = audio
# Make sure file has been uploaded.
# Simple assertions PY2 Only
assert isinstance(cls.audio, telegram.Audio)
assert isinstance(cls.audio.file_id, str)
assert cls.audio.file_id is not ''
def setUp(self):
self.audio_file = open('tests/data/telegram.mp3', 'rb')
self.audio_file_id = 'CQADAQADDwADHyP1B6PSPq2HjX8kAg'
self.audio_file_url = 'https://raw.githubusercontent.com/python-telegram-bot/python-telegram-bot/master/tests/data/telegram.mp3'
self.duration = 4
self.performer = 'Leandro Toledo'
self.title = 'Teste'
self.caption = "Test audio"
self.mime_type = 'audio/mpeg'
self.file_size = 28232
self.json_dict = {
'file_id': self.audio_file_id,
'duration': self.duration,
'file_id': self.audio.file_id,
'duration': self.audio.duration,
'performer': self.performer,
'title': self.title,
'caption': self.caption,
'mime_type': self.mime_type,
'file_size': self.file_size
'mime_type': self.audio.mime_type,
'file_size': self.audio.file_size
}
@flaky(3, 1)
@timeout(10)
def test_send_audio_required_args_only(self):
message = self._bot.sendAudio(self._chat_id, self.audio_file)
self.assertEqual(message.caption, None)
audio = message.audio
self.assertTrue(isinstance(audio.file_id, str))
self.assertNotEqual(audio.file_id, None)
# self.assertEqual(audio.duration, 4)
self.assertEqual(audio.performer, None)
self.assertEqual(audio.title, None)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
def test_expected_values(self):
self.assertEqual(self.audio.duration, 4)
self.assertEqual(self.audio.performer, None)
self.assertEqual(self.audio.title, None)
self.assertEqual(self.audio.mime_type, 'audio/ogg')
self.assertEqual(self.audio.file_size, 28232)
@flaky(3, 1)
@timeout(10)
@ -77,85 +77,38 @@ class AudioTest(BaseTest, unittest.TestCase):
message = self._bot.sendAudio(
self._chat_id,
self.audio_file,
duration=self.duration,
performer=self.performer,
title=self.title,
caption=self.caption,
mime_type=self.mime_type,
file_size=self.file_size)
self.assertEqual(message.caption, self.caption)
audio = message.audio
self.assertTrue(isinstance(audio.file_id, str))
self.assertNotEqual(audio.file_id, None)
self.assertEqual(audio.duration, self.duration)
self.assertEqual(audio.performer, self.performer)
self.assertEqual(audio.title, self.title)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
@flaky(3, 1)
@timeout(10)
def test_send_audio_mp3_file(self):
message = self._bot.sendAudio(
chat_id=self._chat_id,
audio=self.audio_file,
duration=self.duration,
duration=self.audio.duration,
performer=self.performer,
title=self.title,
caption=self.caption)
disable_notification=False)
self.assertEqual(message.caption, self.caption)
audio = message.audio
self.assertTrue(isinstance(audio.file_id, str))
self.assertIsInstance(audio, telegram.Audio)
self.assertIsInstance(audio.file_id, str)
self.assertNotEqual(audio.file_id, None)
self.assertEqual(audio.duration, self.duration)
self.assertEqual(audio.duration, self.audio.duration)
self.assertEqual(audio.performer, self.performer)
self.assertEqual(audio.title, self.title)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
@flaky(3, 1)
@timeout(10)
def test_send_audio_mp3_file_custom_filename(self):
message = self._bot.sendAudio(
chat_id=self._chat_id,
audio=self.audio_file,
duration=self.duration,
performer=self.performer,
title=self.title,
caption=self.caption,
filename='telegram_custom.mp3')
self.assertEqual(message.caption, self.caption)
audio = message.audio
self.assertTrue(isinstance(audio.file_id, str))
self.assertNotEqual(audio.file_id, None)
self.assertEqual(audio.duration, self.duration)
self.assertEqual(audio.performer, self.performer)
self.assertEqual(audio.title, self.title)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
self.assertEqual(audio.mime_type, self.audio.mime_type)
self.assertEqual(audio.file_size, self.audio.file_size)
@flaky(3, 1)
@timeout(10)
def test_send_audio_mp3_url_file(self):
message = self._bot.sendAudio(
chat_id=self._chat_id, audio=self.audio_file_url, duration=self.duration)
message = self._bot.sendAudio(chat_id=self._chat_id, audio=self.audio_file_url)
audio = message.audio
self.assertTrue(isinstance(audio.file_id, str))
self.assertIsInstance(audio, telegram.Audio)
self.assertIsInstance(audio.file_id, str)
self.assertNotEqual(audio.file_id, None)
self.assertEqual(audio.duration, self.duration)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
self.assertEqual(audio.duration, self.audio.duration)
self.assertEqual(audio.mime_type, self.audio.mime_type)
self.assertEqual(audio.file_size, self.audio.file_size)
@flaky(3, 1)
@timeout(10)
@ -163,65 +116,54 @@ class AudioTest(BaseTest, unittest.TestCase):
message = self._bot.sendAudio(
chat_id=self._chat_id,
audio=self.audio_file_url,
duration=self.duration,
caption=self.caption)
self.assertEqual(message.caption, self.caption)
audio = message.audio
self.assertTrue(isinstance(audio.file_id, str))
self.assertIsInstance(audio, telegram.Audio)
self.assertIsInstance(audio.file_id, str)
self.assertNotEqual(audio.file_id, None)
self.assertEqual(audio.duration, self.duration)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
self.assertEqual(audio.duration, self.audio.duration)
self.assertEqual(audio.mime_type, self.audio.mime_type)
self.assertEqual(audio.file_size, self.audio.file_size)
@flaky(3, 1)
@timeout(10)
def test_send_audio_resend(self):
message = self._bot.sendAudio(
chat_id=self._chat_id,
audio=self.audio_file_id,
duration=self.duration,
performer=self.performer,
title=self.title,
caption=self.caption)
self.assertEqual(message.caption, self.caption)
audio=self.audio.file_id)
audio = message.audio
self.assertEqual(audio.file_id, self.audio_file_id)
self.assertEqual(audio.duration, self.duration)
self.assertEqual(audio.performer, self.performer)
self.assertEqual(audio.title, self.title)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio, self.audio)
def test_audio_de_json(self):
audio = telegram.Audio.de_json(self.json_dict, self._bot)
self.assertEqual(audio.file_id, self.audio_file_id)
self.assertEqual(audio.duration, self.duration)
self.assertIsInstance(audio, telegram.Audio)
self.assertEqual(audio.file_id, self.audio.file_id)
self.assertEqual(audio.duration, self.audio.duration)
self.assertEqual(audio.performer, self.performer)
self.assertEqual(audio.title, self.title)
self.assertEqual(audio.mime_type, self.mime_type)
self.assertEqual(audio.file_size, self.file_size)
self.assertEqual(audio.mime_type, self.audio.mime_type)
self.assertEqual(audio.file_size, self.audio.file_size)
def test_audio_to_json(self):
audio = telegram.Audio.de_json(self.json_dict, self._bot)
self.assertTrue(self.is_json(audio.to_json()))
self.assertTrue(self.is_json(self.audio.to_json()))
def test_audio_to_dict(self):
audio = telegram.Audio.de_json(self.json_dict, self._bot)
audio = self.audio.to_dict()
self.assertTrue(self.is_dict(audio.to_dict()))
self.assertEqual(audio['file_id'], self.audio_file_id)
self.assertEqual(audio['duration'], self.duration)
self.assertTrue(self.is_dict(audio))
self.assertEqual(audio['file_id'], self.audio.file_id)
self.assertEqual(audio['duration'], self.audio.duration)
self.assertEqual(audio['performer'], self.performer)
self.assertEqual(audio['title'], self.title)
self.assertEqual(audio['mime_type'], self.mime_type)
self.assertEqual(audio['file_size'], self.file_size)
self.assertEqual(audio['mime_type'], self.audio.mime_type)
self.assertEqual(audio['file_size'], self.audio.file_size)
@flaky(3, 1)
@timeout(10)
@ -231,9 +173,8 @@ class AudioTest(BaseTest, unittest.TestCase):
del (json_dict['file_id'])
json_dict['audio'] = open(os.devnull, 'rb')
self.assertRaises(
telegram.TelegramError,
lambda: self._bot.sendAudio(chat_id=self._chat_id, **json_dict))
with self.assertRaises(telegram.TelegramError):
self._bot.sendAudio(chat_id=self._chat_id, **json_dict)
@flaky(3, 1)
@timeout(10)
@ -243,9 +184,8 @@ class AudioTest(BaseTest, unittest.TestCase):
del (json_dict['file_id'])
json_dict['audio'] = ''
self.assertRaises(
telegram.TelegramError,
lambda: self._bot.sendAudio(chat_id=self._chat_id, **json_dict))
with self.assertRaises(telegram.TelegramError):
self._bot.sendAudio(chat_id=self._chat_id, **json_dict)
@flaky(3, 1)
@timeout(10)
@ -255,25 +195,26 @@ class AudioTest(BaseTest, unittest.TestCase):
del (json_dict['file_id'])
del (json_dict['duration'])
self.assertRaises(
TypeError,
lambda: self._bot.sendAudio(chat_id=self._chat_id, **json_dict))
with self.assertRaises(TypeError):
self._bot.sendAudio(chat_id=self._chat_id, **json_dict)
@flaky(3, 1)
@timeout(10)
def test_reply_audio(self):
"""Test for Message.reply_audio"""
message = self._bot.sendMessage(self._chat_id, '.')
message = message.reply_audio(self.audio_file)
audio = message.reply_audio(self.audio_file).audio
self.assertNotEqual(message.audio.file_id, None)
self.assertIsInstance(audio, telegram.Audio)
self.assertIsInstance(audio.file_id, str)
self.assertNotEqual(audio.file_id, None)
def test_equality(self):
a = telegram.Audio(self.audio_file_id, self.duration)
b = telegram.Audio(self.audio_file_id, self.duration)
c = telegram.Audio(self.audio_file_id, 0)
d = telegram.Audio("", self.duration)
e = telegram.Voice(self.audio_file_id, self.duration)
a = telegram.Audio(self.audio.file_id, self.audio.duration)
b = telegram.Audio(self.audio.file_id, self.audio.duration)
c = telegram.Audio(self.audio.file_id, 0)
d = telegram.Audio("", self.audio.duration)
e = telegram.Voice(self.audio.file_id, self.audio.duration)
self.assertEqual(a, b)
self.assertEqual(hash(a), hash(b))

12
tests/test_document.py
View file

@ -122,19 +122,13 @@ class DocumentTest(BaseTest, unittest.TestCase):
def test_document_de_json(self):
document = telegram.Document.de_json(self.json_dict, self._bot)
self.assertEqual(document.file_id, self.document.file_id)
self.assertTrue(isinstance(document.thumb, telegram.PhotoSize))
self.assertEqual(document.file_name, self.document.file_name)
self.assertEqual(document.mime_type, self.document.mime_type)
self.assertEqual(document.file_size, self.document.file_size)
self.assertEqual(document, self.document)
def test_document_to_json(self):
document = telegram.Document.de_json(self.json_dict, self._bot)
self.assertTrue(self.is_json(document.to_json()))
self.assertTrue(self.is_json(self.document.to_json()))
def test_document_to_dict(self):
document = telegram.Document.de_json(self.json_dict, self._bot).to_dict()
document = self.document.to_dict()
self.assertTrue(self.is_dict(document))
self.assertEqual(document['file_id'], self.document.file_id)

44
tests/test_photo.py
View file

@ -78,15 +78,15 @@ class PhotoTest(BaseTest, unittest.TestCase):
message = self._bot.sendPhoto(self._chat_id, self.photo_file, caption=self.caption, disable_notification=False)
thumb, photo = message.photo
self.assertTrue(isinstance(thumb, telegram.PhotoSize))
self.assertTrue(isinstance(thumb.file_id, str))
self.assertIsInstance(thumb, telegram.PhotoSize)
self.assertIsInstance(thumb.file_id, str)
self.assertNotEqual(thumb.file_id, '')
self.assertEqual(thumb.width, self.thumb.width)
self.assertEqual(thumb.height, self.thumb.height)
self.assertEqual(thumb.file_size, self.thumb.file_size)
self.assertTrue(isinstance(photo, telegram.PhotoSize))
self.assertTrue(isinstance(photo.file_id, str))
self.assertIsInstance(photo, telegram.PhotoSize)
self.assertIsInstance(photo.file_id, str)
self.assertNotEqual(photo.file_id, '')
self.assertEqual(photo.width, self.photo.width)
self.assertEqual(photo.height, self.photo.height)
@ -101,15 +101,15 @@ class PhotoTest(BaseTest, unittest.TestCase):
thumb, photo = message.photo
self.assertTrue(isinstance(thumb, telegram.PhotoSize))
self.assertTrue(isinstance(thumb.file_id, str))
self.assertIsInstance(thumb, telegram.PhotoSize)
self.assertIsInstance(thumb.file_id, str)
self.assertNotEqual(thumb.file_id, '')
self.assertEqual(thumb.width, self.thumb.width)
self.assertEqual(thumb.height, self.thumb.height)
self.assertEqual(thumb.file_size, self.thumb.file_size)
self.assertTrue(isinstance(photo, telegram.PhotoSize))
self.assertTrue(isinstance(photo.file_id, str))
self.assertIsInstance(photo, telegram.PhotoSize)
self.assertIsInstance(photo.file_id, str)
self.assertNotEqual(photo.file_id, '')
self.assertEqual(photo.width, self.photo.width)
self.assertEqual(photo.height, self.photo.height)
@ -134,9 +134,9 @@ class PhotoTest(BaseTest, unittest.TestCase):
raw_bytes = BytesIO(open(self.photo_bytes_jpg_no_standard, 'rb').read())
message = self._bot.sendPhoto(self._chat_id, photo=raw_bytes)
photo = message.photo[-1]
self.assertTrue(isinstance(photo.file_id, str))
self.assertIsInstance(photo.file_id, str)
self.assertNotEqual(photo.file_id, '')
self.assertTrue(isinstance(photo, telegram.PhotoSize))
self.assertIsInstance(photo, telegram.PhotoSize)
self.assertEqual(photo.width, 1920)
self.assertEqual(photo.height, 1080)
self.assertEqual(photo.file_size, 30907)
@ -148,13 +148,13 @@ class PhotoTest(BaseTest, unittest.TestCase):
thumb, photo = message.photo
self.assertTrue(isinstance(thumb, telegram.PhotoSize))
self.assertIsInstance(thumb, telegram.PhotoSize)
self.assertEqual(thumb.file_id, self.thumb.file_id)
self.assertEqual(thumb.width, self.thumb.width)
self.assertEqual(thumb.height, self.thumb.height)
self.assertEqual(thumb.file_size, self.thumb.file_size)
self.assertTrue(isinstance(photo, telegram.PhotoSize))
self.assertIsInstance(photo, telegram.PhotoSize)
self.assertEqual(photo.file_id, self.photo.file_id)
self.assertEqual(photo.width, self.photo.width)
self.assertEqual(photo.height, self.photo.height)
@ -163,19 +163,13 @@ class PhotoTest(BaseTest, unittest.TestCase):
def test_photo_de_json(self):
photo = telegram.PhotoSize.de_json(self.json_dict, self._bot)
self.assertTrue(isinstance(photo, telegram.PhotoSize))
self.assertEqual(photo.file_id, self.photo.file_id)
self.assertEqual(photo.width, self.photo.width)
self.assertEqual(photo.height, self.photo.height)
self.assertEqual(photo.file_size, self.photo.file_size)
self.assertEqual(photo, self.photo)
def test_photo_to_json(self):
photo = telegram.PhotoSize.de_json(self.json_dict, self._bot)
self.assertTrue(self.is_json(photo.to_json()))
self.assertTrue(self.is_json(self.photo.to_json()))
def test_photo_to_dict(self):
photo = telegram.PhotoSize.de_json(self.json_dict, self._bot).to_dict()
photo = self.photo.to_dict()
self.assertTrue(self.is_dict(photo))
self.assertEqual(photo['file_id'], self.photo.file_id)
@ -224,12 +218,12 @@ class PhotoTest(BaseTest, unittest.TestCase):
message = self._bot.sendMessage(self._chat_id, '.')
thumb, photo = message.reply_photo(self.photo_file).photo
self.assertTrue(isinstance(thumb, telegram.PhotoSize))
self.assertTrue(isinstance(thumb.file_id, str))
self.assertIsInstance(thumb, telegram.PhotoSize)
self.assertIsInstance(thumb.file_id, str)
self.assertNotEqual(thumb.file_id, '')
self.assertTrue(isinstance(photo, telegram.PhotoSize))
self.assertTrue(isinstance(photo.file_id, str))
self.assertIsInstance(photo, telegram.PhotoSize)
self.assertIsInstance(photo.file_id, str)
self.assertNotEqual(photo.file_id, '')
def test_equality(self):

16
tests/test_video.py
View file

@ -138,23 +138,15 @@ class VideoTest(BaseTest, unittest.TestCase):
def test_video_de_json(self):
video = telegram.Video.de_json(self.json_dict, self._bot)
self.assertEqual(video.file_id, self.video.file_id)
self.assertEqual(video.width, self.video.width)
self.assertEqual(video.height, self.video.height)
self.assertEqual(video.duration, self.video.duration)
self.assertEqual(video.thumb, self.video.thumb)
self.assertEqual(video.mime_type, self.video.mime_type)
self.assertEqual(video.file_size, self.video.file_size)
self.assertEqual(video, self.video)
def test_video_to_json(self):
video = telegram.Video.de_json(self.json_dict, self._bot)
self.assertTrue(self.is_json(video.to_json()))
self.assertTrue(self.is_json(self.video.to_json()))
def test_video_to_dict(self):
video = telegram.Video.de_json(self.json_dict, self._bot)
video = self.video.to_dict()
self.assertTrue(self.is_dict(video.to_dict()))
self.assertTrue(self.is_dict(video))
self.assertEqual(video['file_id'], self.video.file_id)
self.assertEqual(video['width'], self.video.width)
self.assertEqual(video['height'], self.video.height)

14
tests/test_videonote.py
View file

@ -104,21 +104,15 @@ class VideoNoteTest(BaseTest, unittest.TestCase):
def test_videonote_de_json(self):
videonote = telegram.VideoNote.de_json(self.json_dict, self._bot)
self.assertEqual(videonote.file_id, self.videonote.file_id)
self.assertEqual(videonote.duration, self.videonote.duration)
self.assertEqual(videonote.thumb, self.videonote.thumb)
self.assertEqual(videonote.length, self.videonote.length)
self.assertEqual(videonote.file_size, self.videonote.file_size)
self.assertEqual(videonote, self.videonote)
def test_videonote_to_json(self):
videonote = telegram.VideoNote.de_json(self.json_dict, self._bot)
self.assertTrue(self.is_json(videonote.to_json()))
self.assertTrue(self.is_json(self.videonote.to_json()))
def test_videonote_to_dict(self):
videonote = telegram.VideoNote.de_json(self.json_dict, self._bot)
videonote = self.videonote.to_dict()
self.assertTrue(self.is_dict(videonote.to_dict()))
self.assertTrue(self.is_dict(videonote))
self.assertEqual(videonote['file_id'], self.videonote.file_id)
self.assertEqual(videonote['duration'], self.videonote.duration)
self.assertEqual(videonote['length'], self.videonote.length)

13
tests/test_voice.py
View file

@ -139,20 +139,15 @@ class VoiceTest(BaseTest, unittest.TestCase):
def test_voice_de_json(self):
voice = telegram.Voice.de_json(self.json_dict, self._bot)
self.assertEqual(voice.file_id, self.voice.file_id)
self.assertEqual(voice.duration, self.voice.duration)
self.assertEqual(voice.mime_type, self.voice.mime_type)
self.assertEqual(voice.file_size, self.voice.file_size)
self.assertEqual(voice, self.voice)
def test_voice_to_json(self):
voice = telegram.Voice.de_json(self.json_dict, self._bot)
self.assertTrue(self.is_json(voice.to_json()))
self.assertTrue(self.is_json(self.voice.to_json()))
def test_voice_to_dict(self):
voice = telegram.Voice.de_json(self.json_dict, self._bot)
voice = self.voice.to_dict()
self.assertTrue(self.is_dict(voice.to_dict()))
self.assertTrue(self.is_dict(voice))
self.assertEqual(voice['file_id'], self.voice.file_id)
self.assertEqual(voice['duration'], self.voice.duration)
self.assertEqual(voice['mime_type'], self.voice.mime_type)