mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-08 23:10: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)
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from types import SimpleNamespace
|
|
|
|
import jinja2
|
|
|
|
from py.routes.handlers.model_handlers import ModelPageView
|
|
|
|
|
|
class DummySettings:
|
|
def get(self, key, default=None):
|
|
return default
|
|
|
|
|
|
class DummyI18n:
|
|
def __init__(self):
|
|
self.locale = None
|
|
|
|
def set_locale(self, locale):
|
|
self.locale = locale
|
|
|
|
def get_translation(self, key, default=None, **_kwargs):
|
|
return default or key
|
|
|
|
def create_template_filter(self):
|
|
return lambda key, *_args, **_kwargs: key
|
|
|
|
|
|
class DummyScanner:
|
|
def __init__(self):
|
|
self._cache = SimpleNamespace()
|
|
|
|
async def get_cached_data(self, *_args, **_kwargs):
|
|
return SimpleNamespace(folders=[])
|
|
|
|
|
|
class DummyService:
|
|
def __init__(self):
|
|
self.scanner = DummyScanner()
|
|
|
|
|
|
async def test_model_page_view_reads_version_per_request():
|
|
template_env = jinja2.Environment(
|
|
loader=jinja2.DictLoader({"dummy.html": "{{ version }}"}),
|
|
autoescape=True,
|
|
)
|
|
view = ModelPageView(
|
|
template_env=template_env,
|
|
template_name="dummy.html",
|
|
service=DummyService(),
|
|
settings_service=DummySettings(), # pyright: ignore[reportArgumentType]
|
|
server_i18n=DummyI18n(),
|
|
logger=logging.getLogger("test_model_page_view"),
|
|
)
|
|
|
|
view._get_app_version = lambda: "1.0.2-old"
|
|
first = await view.handle(SimpleNamespace()) # pyright: ignore[reportArgumentType]
|
|
|
|
view._get_app_version = lambda: "1.0.2-new"
|
|
second = await view.handle(SimpleNamespace()) # pyright: ignore[reportArgumentType]
|
|
|
|
assert first.text == "1.0.2-old"
|
|
assert second.text == "1.0.2-new"
|