mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-08-07 21:08:34 +03:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""Tests for subagent announce text shaping on external channel surfaces."""
|
|
|
|
from nanobot.utils.subagent_channel_display import scrub_subagent_announce_body
|
|
|
|
|
|
def test_scrub_subagent_keeps_header_and_result_only() -> None:
|
|
raw = """[Subagent 'Phase1' failed]
|
|
|
|
Task: Collect GitHub stats.
|
|
|
|
Result:
|
|
gh CLI missing.
|
|
|
|
Summarize this naturally for the user. Keep it brief."""
|
|
|
|
out = scrub_subagent_announce_body(raw)
|
|
assert out == "[Subagent 'Phase1' failed]\n\ngh CLI missing."
|
|
assert "Task:" not in out
|
|
assert "Summarize" not in out
|
|
|
|
|
|
def test_scrub_normalizes_crlf_before_result_marker() -> None:
|
|
raw = "[Subagent 'z' failed]\r\n\r\nTask: x\r\n\r\nResult:\r\none line\r\n\r\nSummarize this naturally"
|
|
out = scrub_subagent_announce_body(raw)
|
|
assert "Task:" not in out
|
|
assert out.startswith("[Subagent 'z' failed]")
|
|
assert "one line" in out
|
|
|
|
|
|
def test_scrub_truncates_very_long_result() -> None:
|
|
body = "x" * 900
|
|
raw = f"[Subagent 'z' failed]\n\nTask: t\n\nResult:\n{body}\n\nSummarize this naturally"
|
|
out = scrub_subagent_announce_body(raw)
|
|
assert out.endswith("…")
|
|
assert len(out) < len(raw)
|
|
assert body not in out
|