From b689a084edecb1415b7f9cd1817ff1c6850d0eb0 Mon Sep 17 00:00:00 2001
From: Oleg Abramov <t0rr@mail.ru>
Date: Wed, 4 Apr 2018 16:00:55 +0300
Subject: [PATCH] Add sendMediaGroup test

---
 tests/test_bot.py      | 14 ++++++++++++++
 tests/types/dataset.py |  9 +++++++++
 2 files changed, 23 insertions(+)

diff --git a/tests/test_bot.py b/tests/test_bot.py
index ac0b99e6..f219f5b2 100644
--- a/tests/test_bot.py
+++ b/tests/test_bot.py
@@ -155,3 +155,17 @@ async def test_send_video_note(bot: Bot, event_loop):
                                            duration=video_note.duration, length=video_note.length,
                                            disable_notification=False)
         assert result == msg
+
+
+@pytest.mark.asyncio
+async def test_send_media_group(bot: Bot, event_loop):
+    """ sendMediaGroup method test with file_id """
+    from .types.dataset import MESSAGE_WITH_MEDIA_GROUP, PHOTO
+    msg = types.Message(**MESSAGE_WITH_MEDIA_GROUP)
+    photo = types.PhotoSize(**PHOTO)
+    media = [types.InputMediaPhoto(media=photo.file_id), types.InputMediaPhoto(media=photo.file_id)]
+
+    async with FakeTelegram(message_dict=[MESSAGE_WITH_MEDIA_GROUP, MESSAGE_WITH_MEDIA_GROUP], loop=event_loop):
+        result = await bot.send_media_group(msg.chat.id, media=media, disable_notification=False)
+        assert len(result) == len(media)
+        assert result.pop().media_group_id
diff --git a/tests/types/dataset.py b/tests/types/dataset.py
index 6336a6e4..270dc4cb 100644
--- a/tests/types/dataset.py
+++ b/tests/types/dataset.py
@@ -284,6 +284,15 @@ MESSAGE_WITH_PHOTO = {
     "caption": "photo description"
 }
 
+MESSAGE_WITH_MEDIA_GROUP = {
+    "message_id": 55966,
+    "from": USER,
+    "chat": CHAT,
+    "date": 1522843665,
+    "media_group_id": "12182749320567362",
+    "photo": [PHOTO, PHOTO, PHOTO, PHOTO]
+}
+
 MESSAGE_WITH_PINNED_MESSAGE = {}
 
 MESSAGE_WITH_REPLY_TO_MESSAGE = {}