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())