from vk_parser_app.workers.vk_storage_uploader import video_provider


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


if __name__ == "__main__":
    test_video_provider()