mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-09 07:20:15 -03:00
d2f955266d
Fix ~790 basedpyright errors across the test suite: - Type stub subclasses of real production classes with super().__init__() - Add missing generic type arguments and Dict[str, Any] annotations - Add None guards before subscript/member access - Adapt tests to production API changes (removed dead handlers, PersistentModelCache.get_default, _i18n_filter_added location)
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
import json
|
|
import logging
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from py.routes.handlers.recipe_handlers import RecipeQueryHandler
|
|
|
|
|
|
async def _noop():
|
|
return None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_recipe_query_handler_base_models_limit_zero_returns_all():
|
|
cache = SimpleNamespace(
|
|
raw_data=[
|
|
{"base_model": "SDXL"},
|
|
{"base_model": "LTXV 2.3"},
|
|
{"base_model": "SDXL"},
|
|
]
|
|
)
|
|
scanner = SimpleNamespace(get_cached_data=lambda: None)
|
|
|
|
async def get_cached_data():
|
|
return cache
|
|
|
|
scanner.get_cached_data = get_cached_data
|
|
|
|
handler = RecipeQueryHandler(
|
|
ensure_dependencies_ready=_noop,
|
|
recipe_scanner_getter=lambda: scanner,
|
|
format_recipe_file_url=lambda value: value,
|
|
logger=logging.getLogger(__name__),
|
|
)
|
|
|
|
response = await handler.get_base_models(
|
|
SimpleNamespace(query={"limit": "0"}) # pyright: ignore[reportArgumentType]
|
|
)
|
|
text = response.text
|
|
assert text is not None
|
|
payload = json.loads(text)
|
|
|
|
assert payload["success"] is True
|
|
assert payload["base_models"] == [
|
|
{"name": "SDXL", "count": 2},
|
|
{"name": "LTXV 2.3", "count": 1},
|
|
]
|