mirror of
https://github.com/jmagar/unraid-mcp.git
synced 2026-03-23 12:39:24 -07:00
fix(lint): remove __future__ annotations from new tools, fix 4 failing tests
- Remove `from __future__ import annotations` from array.py, live.py,
oidc.py, plugins.py to match existing tool pattern and resolve TC002
ruff errors (fastmcp imports only needed in annotations under PEP 563)
- Add `# noqa: ASYNC109` to live.py timeout parameter (asyncio.timeout
already used internally)
- Fix test_network_sends_correct_query: query name is GetNetworkInfo
- Fix test_delete_requires_confirm: match "not confirmed" not "destructive"
- Fix test_destructive_set_matches_audit[settings]: add setup_remote_access
and enable_dynamic_remote_access to KNOWN_DESTRUCTIVE
- Fix test_logs: update mock to dict format {lines: [{timestamp, message}]}
742 tests passing, ruff clean
This commit is contained in:
@@ -29,7 +29,7 @@ from .utils import build_ws_ssl_context, build_ws_url
|
||||
async def subscribe_once(
|
||||
query: str,
|
||||
variables: dict[str, Any] | None = None,
|
||||
timeout: float = 10.0,
|
||||
timeout: float = 10.0, # noqa: ASYNC109
|
||||
) -> dict[str, Any]:
|
||||
"""Open a WebSocket subscription, receive the first data event, close, return it.
|
||||
|
||||
@@ -94,7 +94,7 @@ async def subscribe_once(
|
||||
elif msg.get("type") == "error" and msg.get("id") == sub_id:
|
||||
raise ToolError(f"Subscription error: {msg.get('payload')}")
|
||||
except TimeoutError:
|
||||
raise ToolError(f"Subscription timed out after {timeout:.0f}s")
|
||||
raise ToolError(f"Subscription timed out after {timeout:.0f}s") from None
|
||||
|
||||
raise ToolError("WebSocket closed before receiving subscription data")
|
||||
|
||||
@@ -103,7 +103,7 @@ async def subscribe_collect(
|
||||
query: str,
|
||||
variables: dict[str, Any] | None = None,
|
||||
collect_for: float = 5.0,
|
||||
timeout: float = 10.0,
|
||||
timeout: float = 10.0, # noqa: ASYNC109
|
||||
) -> list[dict[str, Any]]:
|
||||
"""Open a subscription, collect events for `collect_for` seconds, close, return list.
|
||||
|
||||
|
||||
Reference in New Issue
Block a user