forked from HomeLab/unraid-mcp
test: close critical coverage gaps and harden PR review fixes
Critical bug fixes from PR review agents: - client.py: eager asyncio.Lock init, Final[frozenset] for _SENSITIVE_KEYS, explicit 429 ToolError after retries exhausted, removed lazy _get_client_lock() and _RateLimiter._get_lock() patterns - exceptions.py: use builtin TimeoutError (UP041), explicit handler before broad except so asyncio timeouts get descriptive messages - docker.py: add update_all to DESTRUCTIVE_ACTIONS (was missing), remove dead _MUTATION_ACTIONS constant - manager.py: _cap_log_content returns new dict (immutable), lock write to resource_data, clean dead task from active_subscriptions after loop exits - diagnostics.py: fix inaccurate comment about semicolon injection guard - health.py: narrow except ValueError in _safe_display_url, fix TODO comment New test coverage (98 tests added, 529 → 598 passing): - test_subscription_validation.py: 27 tests for _validate_subscription_query (security-critical allow-list, forbidden keyword guards, word-boundary test) - test_subscription_manager.py: 12 tests for _cap_log_content (immutability, truncation, nesting, passthrough) - test_client.py: +57 tests — _RateLimiter (token math, refill, sleep-on-empty), _QueryCache (TTL, invalidation, is_cacheable), 429 retry loop (1/2/3 failures) - test_health.py: +10 tests for _safe_display_url (credential strip, port, path/query removal, malformed IPv6 → <unparseable>) - test_notifications.py: +7 importance enum and field length validation tests - test_rclone.py: +7 _validate_config_data security guard tests - test_storage.py: +15 (tail_lines bounds, format_kb, safe_get) - test_docker.py: update_all now requires confirm=True + new guard test - test_destructive_guards.py: update audit to include update_all Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -7,7 +7,7 @@ import pytest
|
||||
from conftest import make_tool_fn
|
||||
|
||||
from unraid_mcp.core.exceptions import ToolError
|
||||
from unraid_mcp.core.utils import format_bytes
|
||||
from unraid_mcp.core.utils import format_bytes, format_kb, safe_get
|
||||
|
||||
|
||||
# --- Unit tests for helpers ---
|
||||
@@ -77,6 +77,70 @@ class TestStorageValidation:
|
||||
result = await tool_fn(action="logs", log_path="/var/log/syslog")
|
||||
assert result["content"] == "ok"
|
||||
|
||||
async def test_logs_tail_lines_too_large(self, _mock_graphql: AsyncMock) -> None:
|
||||
tool_fn = _make_tool()
|
||||
with pytest.raises(ToolError, match="tail_lines must be between"):
|
||||
await tool_fn(action="logs", log_path="/var/log/syslog", tail_lines=10_001)
|
||||
|
||||
async def test_logs_tail_lines_zero_rejected(self, _mock_graphql: AsyncMock) -> None:
|
||||
tool_fn = _make_tool()
|
||||
with pytest.raises(ToolError, match="tail_lines must be between"):
|
||||
await tool_fn(action="logs", log_path="/var/log/syslog", tail_lines=0)
|
||||
|
||||
async def test_logs_tail_lines_at_max_accepted(self, _mock_graphql: AsyncMock) -> None:
|
||||
_mock_graphql.return_value = {"logFile": {"path": "/var/log/syslog", "content": "ok"}}
|
||||
tool_fn = _make_tool()
|
||||
result = await tool_fn(action="logs", log_path="/var/log/syslog", tail_lines=10_000)
|
||||
assert result["content"] == "ok"
|
||||
|
||||
|
||||
class TestFormatKb:
|
||||
def test_none_returns_na(self) -> None:
|
||||
assert format_kb(None) == "N/A"
|
||||
|
||||
def test_invalid_string_returns_na(self) -> None:
|
||||
assert format_kb("not-a-number") == "N/A"
|
||||
|
||||
def test_kilobytes_range(self) -> None:
|
||||
assert format_kb(512) == "512 KB"
|
||||
|
||||
def test_megabytes_range(self) -> None:
|
||||
assert format_kb(2048) == "2.00 MB"
|
||||
|
||||
def test_gigabytes_range(self) -> None:
|
||||
assert format_kb(1_048_576) == "1.00 GB"
|
||||
|
||||
def test_terabytes_range(self) -> None:
|
||||
assert format_kb(1_073_741_824) == "1.00 TB"
|
||||
|
||||
def test_boundary_exactly_1024_kb(self) -> None:
|
||||
# 1024 KB = 1 MB
|
||||
assert format_kb(1024) == "1.00 MB"
|
||||
|
||||
|
||||
class TestSafeGet:
|
||||
def test_simple_key_access(self) -> None:
|
||||
assert safe_get({"a": 1}, "a") == 1
|
||||
|
||||
def test_nested_key_access(self) -> None:
|
||||
assert safe_get({"a": {"b": "val"}}, "a", "b") == "val"
|
||||
|
||||
def test_missing_key_returns_none(self) -> None:
|
||||
assert safe_get({"a": 1}, "missing") is None
|
||||
|
||||
def test_none_intermediate_returns_default(self) -> None:
|
||||
assert safe_get({"a": None}, "a", "b") is None
|
||||
|
||||
def test_custom_default_returned(self) -> None:
|
||||
assert safe_get({}, "x", default="fallback") == "fallback"
|
||||
|
||||
def test_non_dict_intermediate_returns_default(self) -> None:
|
||||
assert safe_get({"a": "string"}, "a", "b") is None
|
||||
|
||||
def test_empty_list_default(self) -> None:
|
||||
result = safe_get({}, "missing", default=[])
|
||||
assert result == []
|
||||
|
||||
|
||||
class TestStorageActions:
|
||||
async def test_shares(self, _mock_graphql: AsyncMock) -> None:
|
||||
|
||||
Reference in New Issue
Block a user