from scripts.download_life_writing_drive import drive_kind


def test_drive_kind_parses_file_and_folder():
    assert drive_kind("https://drive.google.com/file/d/abc123/view") == ("file", "abc123")
    assert drive_kind("https://drive.google.com/drive/folders/xyz789?usp=drive_link") == (
        "folder", "xyz789"
    )
    assert drive_kind("https://example.com") is None