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)
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from py.routes.handlers.misc_handlers import WildcardsHandler
|
|
|
|
|
|
class FakeRequest:
|
|
def __init__(self, query=None):
|
|
self.query = query or {}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_wildcards_returns_results():
|
|
class StubService:
|
|
def get_metadata(self, create_dir=False):
|
|
assert create_dir is True
|
|
return SimpleNamespace(
|
|
has_wildcards=True,
|
|
wildcards_dir="/tmp/settings/wildcards",
|
|
supported_formats=(".txt", ".yaml", ".yml", ".json"),
|
|
)
|
|
|
|
def search_keys(self, search_term, limit, offset):
|
|
assert search_term == "cat"
|
|
assert limit == 25
|
|
assert offset == 2
|
|
return ["animals/cat"]
|
|
|
|
handler = WildcardsHandler(service=StubService())
|
|
response = await handler.search_wildcards(
|
|
FakeRequest(query={"search": "cat", "limit": "25", "offset": "2"}) # pyright: ignore[reportArgumentType]
|
|
)
|
|
text = response.text
|
|
assert text is not None
|
|
payload = json.loads(text)
|
|
|
|
assert response.status == 200
|
|
assert payload == {
|
|
"success": True,
|
|
"words": ["animals/cat"],
|
|
"meta": {
|
|
"has_wildcards": True,
|
|
"wildcards_dir": "/tmp/settings/wildcards",
|
|
"supported_formats": [".txt", ".yaml", ".yml", ".json"],
|
|
},
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_wildcards_handles_errors():
|
|
class StubService:
|
|
def get_metadata(self, create_dir=False):
|
|
return SimpleNamespace(
|
|
has_wildcards=False,
|
|
wildcards_dir="/tmp/settings/wildcards",
|
|
supported_formats=(".txt",),
|
|
)
|
|
|
|
def search_keys(self, search_term, limit, offset):
|
|
raise RuntimeError("boom")
|
|
|
|
handler = WildcardsHandler(service=StubService())
|
|
response = await handler.search_wildcards(
|
|
FakeRequest(query={"search": "cat"}) # pyright: ignore[reportArgumentType]
|
|
)
|
|
text = response.text
|
|
assert text is not None
|
|
payload = json.loads(text)
|
|
|
|
assert response.status == 500
|
|
assert payload["error"] == "boom"
|