Your Name
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
import asyncio
|
|
from pathlib import Path
|
|
|
|
from vk_parser_app.workers.media_uploader import MediaUploader, video_provider
|
|
|
|
|
|
class FakeContent:
|
|
async def iter_chunked(self, _size):
|
|
yield b"video"
|
|
|
|
|
|
class FakeResponse:
|
|
status = 200
|
|
content = FakeContent()
|
|
|
|
async def __aenter__(self):
|
|
return self
|
|
|
|
async def __aexit__(self, *_args):
|
|
return None
|
|
|
|
|
|
class FakeSession:
|
|
def get(self, *_args, **_kwargs):
|
|
return FakeResponse()
|
|
|
|
|
|
def test_video_provider() -> None:
|
|
assert video_provider("https://vk.com/video-1_2") == "vk"
|
|
assert video_provider("https://youtu.be/dQw4w9WgXcQ") == "youtube"
|
|
assert video_provider("https://www.youtube.com/watch?v=dQw4w9WgXcQ") == "youtube"
|
|
assert video_provider("file:///etc/passwd") is None
|
|
assert video_provider("http://127.0.0.1/video.mp4") is None
|
|
|
|
|
|
async def test_http_video_download() -> None:
|
|
path = "/tmp/vkparser_tg_media_http_test.mp4"
|
|
result = await MediaUploader().download_http_video(FakeSession(), "https://example.test/a.mp4", path)
|
|
assert result["size_bytes"] == 5
|
|
assert Path(path).read_bytes() == b"video"
|
|
Path(path).unlink()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_video_provider()
|
|
asyncio.run(test_http_video_download())
|