mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
feat(tests): introduce ROUTE_CALLS_KEY for organizing route calls
Addressed the aiohttp warnings by aligning the test scaffolding with current best practices. Added an AppKey constant and stored the route tracking list through it to satisfy aiohttp’s NotAppKeyWarning expectations. Swapped the websocket lambdas for async no-op handlers so the registered routes now point to coroutine callables, clearing the deprecation warning about bare functions.
This commit is contained in:
@@ -13,6 +13,9 @@ from aiohttp import web
|
||||
from py.utils.settings_paths import ensure_settings_file
|
||||
|
||||
|
||||
ROUTE_CALLS_KEY: web.AppKey[List[Tuple[str, dict]]] = web.AppKey("route_calls")
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def standalone_module(monkeypatch) -> ModuleType:
|
||||
"""Load the ``standalone`` module with a lightweight ``LoraManager`` stub."""
|
||||
@@ -106,7 +109,7 @@ def test_standalone_lora_manager_registers_routes(monkeypatch, tmp_path, standal
|
||||
|
||||
app = web.Application()
|
||||
route_calls: List[Tuple[str, dict]] = []
|
||||
app["route_calls"] = route_calls
|
||||
app[ROUTE_CALLS_KEY] = route_calls
|
||||
|
||||
locales_dir = tmp_path / "locales"
|
||||
locales_dir.mkdir()
|
||||
@@ -180,10 +183,13 @@ def test_standalone_lora_manager_registers_routes(monkeypatch, tmp_path, standal
|
||||
monkeypatch.setattr("py.routes.preview_routes.PreviewRoutes", DummyPreviewRoutes)
|
||||
monkeypatch.setattr("py.routes.stats_routes.StatsRoutes", DummyStatsRoutes)
|
||||
|
||||
async def _noop_ws_handler(request):
|
||||
return web.Response(status=204)
|
||||
|
||||
ws_manager_stub = SimpleNamespace(
|
||||
handle_connection=lambda request: None,
|
||||
handle_download_connection=lambda request: None,
|
||||
handle_init_connection=lambda request: None,
|
||||
handle_connection=_noop_ws_handler,
|
||||
handle_download_connection=_noop_ws_handler,
|
||||
handle_init_connection=_noop_ws_handler,
|
||||
)
|
||||
monkeypatch.setattr("py.services.websocket_manager.ws_manager", ws_manager_stub)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user