oclaw/tests/test_ops_short_intent_and_exec_hints.py
oliver 6c5536b758 Nudge WhatsApp group ops asks that skip @mention.
When require_mention drops an ops short-intent, send one throttled @sender tip instead of silent ignore; casual chat stays quiet.

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

67 lines
2.6 KiB
Python

from __future__ import annotations
from runtime.application.gateway.ops_short_intent import (
build_group_mention_nudge_text,
detect_ops_short_intent,
maybe_ops_short_intent_system_hint,
reset_group_mention_nudge_throttle_for_tests,
should_send_group_mention_nudge,
)
from runtime.tools.tool_error_hints import enrich_exec_managed_ne_error
def test_detect_ops_short_intent_english_field() -> None:
assert detect_ops_short_intent("@bot fiber cut sites") == "fiber_cut"
assert detect_ops_short_intent("offline NE list") == "offline"
assert detect_ops_short_intent("please continue") == "continue"
assert detect_ops_short_intent("YES") == "continue"
assert detect_ops_short_intent("export excel") == "excel_export"
assert detect_ops_short_intent("critical top alarms") == "alarm_tally"
assert detect_ops_short_intent("hello there how are you doing today with something else") is None
def test_group_mention_nudge_text_english() -> None:
text = build_group_mention_nudge_text(intent="fiber_cut", lang="en", triggers=["/oclaw"])
assert "@mentioned" in text.lower() or "@me" in text.lower()
assert "/oclaw" in text
assert "fiber" in text.lower()
def test_group_mention_nudge_throttle() -> None:
reset_group_mention_nudge_throttle_for_tests()
assert should_send_group_mention_nudge(
account_id="wa", chat_id="g1", user_id="u1", now=100.0, ttl_s=60.0
)
assert not should_send_group_mention_nudge(
account_id="wa", chat_id="g1", user_id="u1", now=130.0, ttl_s=60.0
)
assert should_send_group_mention_nudge(
account_id="wa", chat_id="g1", user_id="u2", now=130.0, ttl_s=60.0
)
assert should_send_group_mention_nudge(
account_id="wa", chat_id="g1", user_id="u1", now=170.0, ttl_s=60.0
)
def test_ops_short_intent_hint_english_default() -> None:
hint = maybe_ops_short_intent_system_hint(text="LOS on these sites", lang="en")
assert "fiber" in hint.lower() or "LOS" in hint
assert "ume_alarm_xlsx_report" in hint
assert "断纤" not in hint
def test_enrich_exec_timeout() -> None:
out = enrich_exec_managed_ne_error(
{"ok": False, "error_code": "tool_timeout_or_failed", "error": "timeout"}
)
assert out["error_class"] == "timeout"
assert "read_timeout_sec" in out["hint"]
def test_enrich_exec_unreachable_nested_json() -> None:
nested = '{"ok": false, "error": "ssh_connect failed: host unreachable"}'
out = enrich_exec_managed_ne_error(
{"ok": False, "error_code": "mcp_tool_call_failed", "error": nested}
)
assert out["error_class"] == "unreachable"
assert "unreachable" in out["hint"].lower()