From 65ae35ea024e214682fae6c9a3516741799e08ed Mon Sep 17 00:00:00 2001 From: Dineshkarthik Date: Fri, 11 Feb 2022 12:49:08 +0100 Subject: [PATCH] test: add test for video_note media type --- tests/test_media_downloader.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/tests/test_media_downloader.py b/tests/test_media_downloader.py index 0c68cd3..480d466 100644 --- a/tests/test_media_downloader.py +++ b/tests/test_media_downloader.py @@ -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")