mirror of
https://github.com/Dineshkarthik/telegram_media_downloader.git
synced 2024-09-20 07:16:32 +08:00
test: add test for video_note media type
This commit is contained in:
parent
2c597b9734
commit
65ae35ea02
|
@ -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")
|
||||
|
|
Loading…
Reference in a new issue