o2o-plagiarism-ai/tests/test_download_life_writing_drive.py

10 lines
375 B
Python

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