WindyMadman/telegram_media_downloader
1
0
Fork 0
mirror of https://github.com/Dineshkarthik/telegram_media_downloader.git synced 2025-01-08 11:37:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: update mock objects to handle pyrogram V2 migration

Browse Source
This commit is contained in:
Dineshkarthik 2022-07-18 16:58:19 +02:00
parent 1702ec9788
commit 6d58a01f94
No known key found for this signature in database
GPG Key ID: E256EF0EECD6F201
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_media_downloader.py
View File

@ -53,7 +53,7 @@ class Chat:
class MockMessage:
def __init__(self, **kwargs):
self.message_id = kwargs.get("id")
self.id = kwargs.get("id")
self.media = kwargs.get("media")
self.audio = kwargs.get("audio", None)
self.document = kwargs.get("document", None)
@ -153,7 +153,7 @@ class MockClient:
async def stop(self):
pass
async def iter_history(self, *args, **kwargs):
async def get_chat_history(self, *args, **kwargs):
items = [
MockMessage(
id=1213,
@ -219,11 +219,11 @@ class MockClient:
async def download_media(self, *args, **kwargs):
mock_message = args[0]
if mock_message.message_id in [7, 8]:
if mock_message.id in [7, 8]:
raise pyrogram.errors.exceptions.bad_request_400.BadRequest
elif mock_message.message_id == 9:
elif mock_message.id == 9:
raise pyrogram.errors.exceptions.unauthorized_401.Unauthorized
elif mock_message.message_id == 11:
elif mock_message.id == 11:
raise TypeError
return kwargs["file_name"]