Dineshkarthik/telegram_media_downloader
1
1
Fork 0
mirror of https://github.com/Dineshkarthik/telegram_media_downloader.git synced 2024-12-28 01:33:00 +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", "api_hash": "hasw5Tgawsuj67",
"last_read_message_id": 0, "last_read_message_id": 0,
"chat_id": 8654123, "chat_id": 8654123,
"ids_to_retry": [], "ids_to_retry": [1],
"media_types": ["audio", "voice"], "media_types": ["audio", "voice"],
"file_formats": {"audio": ["all"], "voice": ["all"]}, "file_formats": {"audio": ["all"], "voice": ["all"]},
} }
@ -203,6 +203,18 @@ class MockClient:
mime_type="video/mov", 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 return message
async def download_media(self, *args, **kwargs): async def download_media(self, *args, **kwargs):