Initial commit

tests/test_office_ui_message_sanitization.py

@@ -0,0 +1,103 @@
+from __future__ import annotations
+
+from datetime import datetime
+from types import SimpleNamespace
+
+from opc.plugins.office_ui.snapshot_builder import (
+    _sanitize_ui_message_dict,
+    _strip_trailing_verification_footer,
+    _transcript_item_to_ui_message,
+)
+
+
+def test_strip_trailing_verification_footer_removes_only_footer() -> None:
+    content = (
+        "I can help with many executable tasks.\n\n"
+        "Verification: not required because no code edits or risky runtime actions were detected."
+    )
+
+    body, footer = _strip_trailing_verification_footer(content)
+
+    assert body == "I can help with many executable tasks."
+    assert footer == "Verification: not required because no code edits or risky runtime actions were detected."
+
+
+def test_strip_trailing_verification_footer_keeps_footer_only_message() -> None:
+    content = "Verification: verified by verify."
+
+    body, footer = _strip_trailing_verification_footer(content)
+
+    assert body == content
+    assert footer is None
+
+
+def test_sanitize_ui_message_dict_strips_assistant_footer_and_preserves_metadata() -> None:
+    message = {
+        "message_id": "m1",
+        "channel_id": "session:t1",
+        "sender": "assistant",
+        "sender_name": "OPC",
+        "content": (
+            "Completed the task.\n\n"
+            "Verification: verified by verify."
+        ),
+        "created_at": 123.0,
+        "metadata": {"source": "engine", "role": "assistant"},
+    }
+
+    sanitized = _sanitize_ui_message_dict(message)
+
+    assert sanitized["content"] == "Completed the task."
+    assert sanitized["metadata"]["verification_verdict"] == "Verification: verified by verify."
+    assert sanitized["metadata"]["role"] == "assistant"
+
+
+def test_transcript_item_to_ui_message_hides_runtime_internal_user_turns() -> None:
+    item = {
+        "message": SimpleNamespace(
+            message_id="runtime-user-1",
+            role="user",
+            agent_id="",
+            created_at=datetime.now(),
+            summary_flag=False,
+            metadata={"kind": "runtime_v2_user_turn"},
+        ),
+        "parts": [
+            SimpleNamespace(
+                part_type="text",
+                payload={"text": "## Task\nWhat tasks can you help with?"},
+            ),
+        ],
+    }
+
+    assert _transcript_item_to_ui_message(item, channel_id="session:t1", task_id="t1") is None
+
+
+def test_transcript_item_to_ui_message_strips_trailing_verification_footer() -> None:
+    item = {
+        "message": SimpleNamespace(
+            message_id="reply-1",
+            role="assistant",
+            agent_id="",
+            created_at=datetime.now(),
+            summary_flag=False,
+            metadata={"kind": "top_level_reply"},
+        ),
+        "parts": [
+            SimpleNamespace(
+                part_type="text",
+                payload={
+                    "text": (
+                        "Here is the answer.\n\n"
+                        "Verification: not required because no code edits or risky runtime actions were detected."
+                    ),
+                },
+            ),
+        ],
+    }
+
+    mapped = _transcript_item_to_ui_message(item, channel_id="session:t1", task_id="t1")
+
+    assert mapped is not None
+    assert mapped["content"] == "Here is the answer."
+    assert mapped["metadata"]["verification_verdict"].startswith("Verification:")