test: add test for video_note media type

This commit is contained in:
Dineshkarthik 2022-02-11 12:49:08 +01:00
parent 2c597b9734
commit 65ae35ea02
No known key found for this signature in database
GPG key ID: E256EF0EECD6F201

View file

@ -60,6 +60,7 @@ class MockMessage:
self.photo = kwargs.get("photo", None)
self.video = kwargs.get("video", None)
self.voice = kwargs.get("voice", None)
self.video_note = kwargs.get("video_note", None)
self.chat = Chat(kwargs.get("chat_id", None))
@ -91,6 +92,12 @@ class MockVideo:
self.mime_type = kwargs["mime_type"]
class MockVideoNote:
def __init__(self, **kwargs):
self.mime_type = kwargs["mime_type"]
self.date = kwargs["date"]
class MockEventLoop:
def __init__(self):
pass
@ -317,6 +324,28 @@ class MediaDownloaderTestCase(unittest.TestCase):
result,
)
# Test VideoNote
message = MockMessage(
id=6,
media=True,
video_note=MockVideoNote(
mime_type="video/mp4",
date=1564066430,
),
)
result = self.loop.run_until_complete(
async_get_media_meta(message.video_note, "video_note")
)
self.assertEqual(
(
platform_generic_path(
"/root/project/video_note/video_note_2019-07-25T14:53:50.mp4"
),
"mp4",
),
result,
)
@mock.patch("media_downloader.THIS_DIR", new=MOCK_DIR)
@mock.patch("media_downloader.asyncio.sleep", return_value=None)
@mock.patch("media_downloader.logger")