mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-05-06 08:26:45 -03:00
45 lines
1.1 KiB
Python
45 lines
1.1 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"}))
|
|
payload = json.loads(response.text)
|
|
|
|
assert payload["success"] is True
|
|
assert payload["base_models"] == [
|
|
{"name": "SDXL", "count": 2},
|
|
{"name": "LTXV 2.3", "count": 1},
|
|
]
|