oclaw/tests/test_tool_error_hints.py
oliver e91e4fe773 Isolate WhatsApp group turns by speaker and clarify SQL scope denials.
Always tag group senders, inject group-focus system hints, label queued follow-ups by speaker, and mark insufficient_scope failures as non-retryable with a user-facing hint.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-08-10 23:11:41 +08:00

49 lines
1.8 KiB
Python

from __future__ import annotations
from runtime.tools.tool_error_hints import (
enrich_mcp_scope_error,
format_unregistered_tool_error,
suggest_tool_names,
)
def test_suggest_tool_names_prefers_substring() -> None:
pool = [
"mcp__netx__queryUmeAlarms",
"mcp__netx__queryUmeAlarmsRaw",
"mcp__netx__aggregateUmeAlarms",
"write_xlsx",
]
hits = suggest_tool_names("mcp__netx__queryUmeAlarmsRaw", pool)
assert "mcp__netx__queryUmeAlarmsRaw" in hits or "mcp__netx__queryUmeAlarms" in hits
def test_format_unregistered_includes_suggestions() -> None:
out = format_unregistered_tool_error(
"mcp__netx__queryUmeAlarmsRaw",
["mcp__netx__queryUmeAlarms", "mcp__netx__aggregateUmeAlarms", "ume_alarm_xlsx_report"],
lang="en",
)
assert out["error_code"] == "tool_not_registered"
assert out.get("suggestions")
assert any("queryUmeAlarms" in s for s in out["suggestions"])
def test_enrich_mcp_scope_sql() -> None:
raw = {"ok": False, "error_code": "mcp_rpc_error_-32001", "error": "insufficient_scope:sql:query"}
out = enrich_mcp_scope_error(raw)
assert out["error_code"] == "insufficient_scope"
assert out.get("failure_class") == "auth"
assert out.get("retry_forbidden") is True
assert "aggregateUmeAlarms" in str(out.get("hint") or "")
assert "SQL query is not enabled" in str(out.get("user_facing_hint") or "")
assert out["required_scope"] == "sql:query"
assert "fallback_tools" in out
assert "ume_alarm_xlsx_report" in out["fallback_tools"]
def test_enrich_exec_auth() -> None:
from runtime.tools.tool_error_hints import enrich_exec_managed_ne_error
out = enrich_exec_managed_ne_error({"ok": False, "error": "authentication failed: Permission denied"})
assert out["error_class"] == "auth"