test: more tests

Author: 2mal3

pytailwindcss_extra/main.py

@@ -25,7 +25,8 @@
 def main() -> int:
     bin_dir_path = Path(user_data_dir("pytailwindcss-extra")) / "bin"
 
-    version = get_version(environ.get("PYTAILWINDCSS_EXTRA_VERSION", "major"))
+    cache_file_path = Path(gettempdir()) / "tailwindcss-extra-cache.json"
+    version = get_version(environ.get("PYTAILWINDCSS_EXTRA_VERSION", "major"), cache_file_path)
 
     bin_path = bin_dir_path / f"tailwindcss-extra-{version.replace('.', '-')}"
     if not bin_path.exists():
@@ -38,11 +39,10 @@ def main() -> int:
     return result.returncode
 
 
-def get_version(specifier: str) -> str:
+def get_version(specifier: str, cache_file_path: Path) -> str:
     if specifier not in ["latest", "major"]:
         return specifier
 
-    cache_file_path = Path(gettempdir()) / "tailwindcss-extra-cache.json"
     cached_version = get_cached_version(specifier, cache_file_path)
     if cached_version:
         return cached_version
@@ -66,6 +66,8 @@ def get_cached_version(key: str, cache_file_path: Path) -> str | None:
         cache = loads(file.read())
 
     cache_for_key = cache.get(key)
+    if not cache_for_key:
+        return None
 
     if datetime.fromtimestamp(int(cache_for_key["time"])) + timedelta(hours=CACHE_EXPIRATION_HOURS) < datetime.now():
         return None

tests/test_main.py

@@ -1,5 +1,28 @@
-from pytailwindcss_extra.main import get_latest_major_version_tag
+from pathlib import Path
+from typing import Generator, Callable
+from os import remove
+
+import pytest
+
+from pytailwindcss_extra.main import get_latest_major_version_tag, get_download_url, get_version
 
 
 def test_get_latest_major_version_tag() -> None:
     assert get_latest_major_version_tag(1) == "v1.7.27"
+
+
+def test_get_download_url() -> None:
+    assert get_download_url("v1.7.27")
+
+
+@pytest.fixture(scope="module")
+def cleanup_cache() -> Generator:
+    yield
+    remove(Path("cache.json").absolute())
+
+
+@pytest.mark.parametrize("specifier", ["latest", "major", "v1.7.27"])
+def test_get_version(cleanup_cache: Callable, specifier: str) -> None:
+    cache_file_path = Path("cache.json").absolute()
+
+    assert get_version(specifier, cache_file_path).startswith("v")