"""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