python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

test_reply_media_group fix

Browse source
This commit is contained in:
Oleg A 2018-08-08 03:14:14 +03:00 • committed by t0rr
parent 3d6219f441
commit 2678a6afca
2 changed files with 16 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/test_message.py
View file

@ -379,22 +379,29 @@ class TestMessageVideoNote:
class TestMessageMediaGroup:
photo = types.PhotoSize(**dataset.PHOTO)
media = [types.InputMediaPhoto(media=photo.file_id), types.InputMediaPhoto(media=photo.file_id)]
async def test_reply_media_group(self, message, bot, monkeypatch, event_loop):
""" Message.reply_media_group method test """
msg = types.Message(**dataset.MESSAGE_WITH_MEDIA_GROUP_AND_REPLY)
photo = types.InputMediaPhoto
async with FakeTelegram(message_dict=dataset.MESSAGE_WITH_MEDIA_GROUP_AND_REPLY,
loop=event_loop, bot=bot, monkeypatch=monkeypatch):
result = await message.reply_media_group(media=types.MediaGroup())
assert result == msg
async with FakeTelegram(message_dict=[dataset.MESSAGE_WITH_MEDIA_GROUP_AND_REPLY,
dataset.MESSAGE_WITH_MEDIA_GROUP_AND_REPLY],
loop=event_loop, bot=bot, monkeypatch=monkeypatch):
result = await message.reply_media_group(media=self.media)
assert len(result) == len(self.media)
assert result[0] == msg
async def test_reply_video_note_without_reply(self, message, bot, monkeypatch, event_loop):
""" Message.reply_media_group method test (without reply_to_message) """
msg = types.Message(**dataset.MESSAGE_WITH_MEDIA_GROUP)
async with FakeTelegram(message_dict=dataset.MESSAGE_WITH_MEDIA_GROUP,
async with FakeTelegram(message_dict=[dataset.MESSAGE_WITH_MEDIA_GROUP,
dataset.MESSAGE_WITH_MEDIA_GROUP],
loop=event_loop, bot=bot, monkeypatch=monkeypatch):
result = await message.reply_media_group(media=types.MediaGroup(), reply=False)
result = await message.reply_media_group(media=self.media, reply=False)
assert result == msg
assert len(result) == len(self.media)
assert result[0] == msg