test(example-images): expand coverage for routes and utilities

tests/utils/test_example_images_download_manager_unit.py

@@ -0,0 +1,132 @@
+from __future__ import annotations
+
+import asyncio
+from typing import Any, Dict
+
+import pytest
+
+from py.services.settings_manager import settings
+from py.utils import example_images_download_manager as download_module
+
+
+class RecordingWebSocketManager:
+    def __init__(self) -> None:
+        self.payloads: list[Dict[str, Any]] = []
+
+    async def broadcast(self, payload: Dict[str, Any]) -> None:
+        self.payloads.append(payload)
+
+
+@pytest.fixture(autouse=True)
+def restore_settings() -> None:
+    original = settings.settings.copy()
+    try:
+        yield
+    finally:
+        settings.settings.clear()
+        settings.settings.update(original)
+
+
+async def test_start_download_requires_configured_path(monkeypatch: pytest.MonkeyPatch) -> None:
+    manager = download_module.DownloadManager(ws_manager=RecordingWebSocketManager())
+
+    with pytest.raises(download_module.ExampleImagesDownloadError) as exc_info:
+        await manager.start_download({})
+
+    assert isinstance(exc_info.value.__cause__, download_module.DownloadConfigurationError)
+    assert "not configured" in str(exc_info.value)
+
+    result = await manager.start_download({"auto_mode": True})
+    assert result["success"] is True
+    assert "skipping auto download" in result["message"]
+
+
+async def test_start_download_bootstraps_progress_and_task(monkeypatch: pytest.MonkeyPatch, tmp_path) -> None:
+    settings.settings["example_images_path"] = str(tmp_path)
+    settings.settings["libraries"] = {"default": {}}
+    settings.settings["active_library"] = "default"
+
+    ws_manager = RecordingWebSocketManager()
+    manager = download_module.DownloadManager(ws_manager=ws_manager)
+
+    started = asyncio.Event()
+    release = asyncio.Event()
+
+    async def fake_download(self, output_dir, optimize, model_types, delay, library_name):
+        started.set()
+        await release.wait()
+        async with self._state_lock:
+            self._is_downloading = False
+            self._download_task = None
+        self._progress["status"] = "completed"
+
+    monkeypatch.setattr(
+        download_module.DownloadManager,
+        "_download_all_example_images",
+        fake_download,
+    )
+
+    result = await manager.start_download({"model_types": ["lora"], "delay": 0})
+    assert result["success"] is True
+    assert manager._is_downloading is True
+
+    await asyncio.wait_for(started.wait(), timeout=1)
+    assert ws_manager.payloads[0]["status"] == "running"
+
+    task = manager._download_task
+    assert task is not None
+    release.set()
+    await asyncio.wait_for(task, timeout=1)
+    assert manager._is_downloading is False
+    assert manager._progress["status"] == "completed"
+
+
+async def test_pause_and_resume_flow(monkeypatch: pytest.MonkeyPatch, tmp_path) -> None:
+    settings.settings["example_images_path"] = str(tmp_path)
+    settings.settings["libraries"] = {"default": {}}
+    settings.settings["active_library"] = "default"
+
+    ws_manager = RecordingWebSocketManager()
+    manager = download_module.DownloadManager(ws_manager=ws_manager)
+
+    started = asyncio.Event()
+    release = asyncio.Event()
+
+    async def fake_download(self, *_args):
+        started.set()
+        await release.wait()
+        async with self._state_lock:
+            self._is_downloading = False
+            self._download_task = None
+
+    monkeypatch.setattr(
+        download_module.DownloadManager,
+        "_download_all_example_images",
+        fake_download,
+    )
+
+    await manager.start_download({})
+    await asyncio.wait_for(started.wait(), timeout=1)
+
+    pause_response = await manager.pause_download(object())
+    assert pause_response == {"success": True, "message": "Download paused"}
+    assert manager._progress["status"] == "paused"
+
+    resume_response = await manager.resume_download(object())
+    assert resume_response == {"success": True, "message": "Download resumed"}
+    assert manager._progress["status"] == "running"
+
+    task = manager._download_task
+    assert task is not None
+    release.set()
+    await asyncio.wait_for(task, timeout=1)
+
+
+async def test_pause_or_resume_without_running_download(monkeypatch: pytest.MonkeyPatch) -> None:
+    manager = download_module.DownloadManager(ws_manager=RecordingWebSocketManager())
+
+    with pytest.raises(download_module.DownloadNotRunningError):
+        await manager.pause_download(object())
+
+    with pytest.raises(download_module.DownloadNotRunningError):
+        await manager.resume_download(object())