mirror of
https://github.com/Dineshkarthik/telegram_media_downloader.git
synced 2024-09-20 15:26:32 +08:00
upd: adapt unittest
This commit is contained in:
parent
43f961eb8e
commit
09198411dd
|
@ -37,32 +37,32 @@ class MockMessage:
|
||||||
|
|
||||||
class MockAudio:
|
class MockAudio:
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
self.file_id = kwargs["file_id"]
|
self.file_ref = kwargs["file_ref"]
|
||||||
self.file_name = kwargs["file_name"]
|
self.file_name = kwargs["file_name"]
|
||||||
|
|
||||||
|
|
||||||
class MockDocument:
|
class MockDocument:
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
self.file_id = kwargs["file_id"]
|
self.file_ref = kwargs["file_ref"]
|
||||||
self.file_name = kwargs["file_name"]
|
self.file_name = kwargs["file_name"]
|
||||||
|
|
||||||
|
|
||||||
class MockPhoto:
|
class MockPhoto:
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
self.file_id = kwargs["file_id"]
|
self.file_ref = kwargs["file_ref"]
|
||||||
self.date = kwargs["date"]
|
self.date = kwargs["date"]
|
||||||
|
|
||||||
|
|
||||||
class MockVoice:
|
class MockVoice:
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
self.file_id = kwargs["file_id"]
|
self.file_ref = kwargs["file_ref"]
|
||||||
self.mime_type = kwargs["mime_type"]
|
self.mime_type = kwargs["mime_type"]
|
||||||
self.date = kwargs["date"]
|
self.date = kwargs["date"]
|
||||||
|
|
||||||
|
|
||||||
class MockVideo:
|
class MockVideo:
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
self.file_id = kwargs["file_id"]
|
self.file_ref = kwargs["file_ref"]
|
||||||
self.file_name = kwargs["file_name"]
|
self.file_name = kwargs["file_name"]
|
||||||
|
|
||||||
|
|
||||||
|
@ -76,7 +76,7 @@ class MockClient:
|
||||||
id=1213,
|
id=1213,
|
||||||
media=True,
|
media=True,
|
||||||
voice=MockVoice(
|
voice=MockVoice(
|
||||||
file_id="AwADBQADbwAD2oTRVeHe5eXRFftfAg",
|
file_ref="AwADBQADbwAD2oTRVeHe5eXRFftfAg",
|
||||||
mime_type="audio/ogg",
|
mime_type="audio/ogg",
|
||||||
date=1564066430,
|
date=1564066430,
|
||||||
),
|
),
|
||||||
|
@ -99,7 +99,7 @@ class MediaDownloaderTestCase(unittest.TestCase):
|
||||||
id=1,
|
id=1,
|
||||||
media=True,
|
media=True,
|
||||||
voice=MockVoice(
|
voice=MockVoice(
|
||||||
file_id="AwADBQADbwAD2oTRVeHe5eXRFftfAg",
|
file_ref="AwADBQADbwAD2oTRVeHe5eXRFftfAg",
|
||||||
mime_type="audio/ogg",
|
mime_type="audio/ogg",
|
||||||
date=1564066430,
|
date=1564066430,
|
||||||
),
|
),
|
||||||
|
@ -118,7 +118,7 @@ class MediaDownloaderTestCase(unittest.TestCase):
|
||||||
id=2,
|
id=2,
|
||||||
media=True,
|
media=True,
|
||||||
photo=MockPhoto(
|
photo=MockPhoto(
|
||||||
file_id="AgADBQAD5KkxG_FPQValJzQsJPyzhHcC", date=1565015712
|
file_ref="AgADBQAD5KkxG_FPQValJzQsJPyzhHcC", date=1565015712
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
result = _get_media_meta(message.photo, "photo")
|
result = _get_media_meta(message.photo, "photo")
|
||||||
|
@ -132,7 +132,7 @@ class MediaDownloaderTestCase(unittest.TestCase):
|
||||||
id=3,
|
id=3,
|
||||||
media=True,
|
media=True,
|
||||||
document=MockDocument(
|
document=MockDocument(
|
||||||
file_id="AQADAgADq7LfMgAEIdy5DwAE4w4AAwI",
|
file_ref="AQADAgADq7LfMgAEIdy5DwAE4w4AAwI",
|
||||||
file_name="sample_document.pdf",
|
file_name="sample_document.pdf",
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
@ -150,7 +150,7 @@ class MediaDownloaderTestCase(unittest.TestCase):
|
||||||
id=4,
|
id=4,
|
||||||
media=True,
|
media=True,
|
||||||
audio=MockAudio(
|
audio=MockAudio(
|
||||||
file_id="AQADAgADq7LfMgAEIdy5DwAE5Q4AAgEC",
|
file_ref="AQADAgADq7LfMgAEIdy5DwAE5Q4AAgEC",
|
||||||
file_name="sample_audio.mp3",
|
file_name="sample_audio.mp3",
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
@ -168,7 +168,7 @@ class MediaDownloaderTestCase(unittest.TestCase):
|
||||||
id=5,
|
id=5,
|
||||||
media=True,
|
media=True,
|
||||||
video=MockVideo(
|
video=MockVideo(
|
||||||
file_id="CQADBQADeQIAAlL60FUCNMBdK8OjlAI",
|
file_ref="CQADBQADeQIAAlL60FUCNMBdK8OjlAI",
|
||||||
file_name="sample_video.mp4",
|
file_name="sample_video.mp4",
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in a new issue