Dineshkarthik/telegram_media_downloader
1
1
Fork 0
mirror of https://github.com/Dineshkarthik/telegram_media_downloader.git synced 2024-12-27 09:11:56 +08:00
Code Issues Projects Releases Packages Wiki Activity

test: add test to download skipped files logic

Browse source
This commit is contained in:
Dineshkarthik 2022-02-28 20:52:57 +01:00
parent 5e34342cb9
commit 6c23daaa89
No known key found for this signature in database
GPG key ID: E256EF0EECD6F201
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_media_downloader.py
View file

@ -29,7 +29,7 @@ MOCK_CONF = {
"api_hash": "hasw5Tgawsuj67",
"last_read_message_id": 0,
"chat_id": 8654123,
"ids_to_retry": [],
"ids_to_retry": [1],
"media_types": ["audio", "voice"],
"file_formats": {"audio": ["all"], "voice": ["all"]},
}
@ -203,6 +203,18 @@ class MockClient:
mime_type="video/mov",
),
)
elif kwargs["message_ids"] == [1]:
message = [
MockMessage(
id=1,
media=True,
chat_id=234568,
video=MockVideo(
file_name="sample_video.mov",
mime_type="video/mov",
),
)
]
return message
async def download_media(self, *args, **kwargs):