host
4cd9b5eaa4
新增 run_tests.py：三套件统一入口，只加仓库根（deploy 套件 再加 deploy/ 目录在根之前），各套件走独立子进程；verify 作 业改走运行器，coverage 作业因子进程测不到覆盖率保持原 discover 命令。tests/ 13 个文件的 sys.path 前置 import 全部 改回正常 import（含 test_host_trace 的 try/except 回落）； deploy 的 deploy 目录插入与 verify-multiuser 的 spec 加载按 陷阱 2 如实保留（仅删 verify-multiuser 冗余的根插入）。 .coveragerc omit 加 */run_tests.py，使新运行器文件不计入 生产覆盖率。门槛未动（仍 75%），未补新测试。
188 lines
8.5 KiB
Python
188 lines
8.5 KiB
Python
"""实时链路（host 侧）单测：通知摘要器 + 链路文件追加。
|
||
|
||
不启动 app-server：Server 实例用 object.__new__ 绕开 __init__
|
||
（__init__ 会拉起 codex 子进程），只测与进程无关的方法。
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import shutil
|
||
import tempfile
|
||
import unittest
|
||
from host import Server, TRACE_MAX_LINES, _summarize_trace_item
|
||
|
||
|
||
|
||
def _msg(item, ts=1700000000000):
|
||
return {"method": "item/completed",
|
||
"params": {"item": item, "completedAtMs": ts}}
|
||
|
||
|
||
class SummarizeTraceItemTestCase(unittest.TestCase):
|
||
def test_exec_truncates_command(self):
|
||
entries = _summarize_trace_item(
|
||
"item/completed",
|
||
{"item": {"type": "commandExecution", "id": "c1",
|
||
"command": "x" * 600, "cwd": "/w",
|
||
"exitCode": 0, "durationMs": 12},
|
||
"completedAtMs": 1700000000000})
|
||
self.assertEqual(len(entries), 1)
|
||
self.assertEqual(entries[0]["kind"], "exec")
|
||
self.assertEqual(len(entries[0]["command"]), 500)
|
||
self.assertEqual(entries[0]["exitCode"], 0)
|
||
self.assertEqual(entries[0]["ts"], 1700000000000)
|
||
|
||
def test_file_change_split_per_change(self):
|
||
entries = _summarize_trace_item(
|
||
"item/completed",
|
||
{"item": {"type": "fileChange", "id": "f1", "status": "completed",
|
||
"changes": [{"path": "a.py", "kind": "update"},
|
||
{"path": "b.py", "kind": "add"}]},
|
||
"startedAtMs": 1700000000001})
|
||
self.assertEqual([e["kind"] for e in entries], ["file", "file"])
|
||
self.assertEqual(entries[0]["id"], "f1:a.py")
|
||
self.assertEqual(entries[1]["path"], "b.py")
|
||
|
||
def test_message_reasoning_tool_call_plan(self):
|
||
cases = [
|
||
({"type": "agentMessage", "id": "m1", "phase": "final",
|
||
"text": "done"}, "message"),
|
||
({"type": "reasoning", "id": "r1",
|
||
"summary": ["a", "b"]}, "reasoning"),
|
||
({"type": "mcpToolCall", "id": "t1", "server": "s",
|
||
"tool": "read", "status": "completed"}, "tool"),
|
||
({"type": "functionCallOutput", "id": "o1",
|
||
"name": "read_file"}, "call"),
|
||
({"type": "plan", "id": "p1", "text": "step"}, "plan"),
|
||
]
|
||
for item, kind in cases:
|
||
entries = _summarize_trace_item("item/completed",
|
||
{"item": item})
|
||
self.assertEqual(len(entries), 1)
|
||
self.assertEqual(entries[0]["kind"], kind)
|
||
|
||
def test_message_truncation_is_flagged(self):
|
||
# 2026-10-29 实弹：468 字 final_answer 在链路里只剩前
|
||
# 300 字、断在“fu”后，控制台照原样当助手回复展示，
|
||
# 用户误以为模型没写完。截断本身是设计（防链路文件
|
||
# 写爆），但必须带标记，短消息不带标记。
|
||
long_entries = _summarize_trace_item(
|
||
"item/completed",
|
||
{"item": {"type": "agentMessage", "id": "m-long",
|
||
"phase": "final_answer", "text": "x" * 468}})
|
||
self.assertEqual(len(long_entries[0]["text"]), 300)
|
||
self.assertTrue(long_entries[0].get("truncated"))
|
||
short_entries = _summarize_trace_item(
|
||
"item/completed",
|
||
{"item": {"type": "agentMessage", "id": "m-short",
|
||
"phase": "final_answer", "text": "done"}})
|
||
self.assertNotIn("truncated", short_entries[0])
|
||
|
||
def test_skipped_notifications(self):
|
||
self.assertEqual(_summarize_trace_item(
|
||
"item/completed", {"item": {"type": "userMessage", "id": "u1",
|
||
"text": "hi"}}), [])
|
||
self.assertEqual(_summarize_trace_item(
|
||
"item/completed", {"item": {"type": "agentMessage", "id": "m2",
|
||
"text": ""}}), [])
|
||
self.assertEqual(_summarize_trace_item("item/completed", {}), [])
|
||
self.assertEqual(_summarize_trace_item(
|
||
"item/completed", {"item": {"type": "unknownThing",
|
||
"id": "z"}}), [])
|
||
|
||
|
||
class TraceFileTestCase(unittest.TestCase):
|
||
def _bare_server(self):
|
||
srv = object.__new__(Server)
|
||
srv.trace_path = None
|
||
srv._trace_lines = 0
|
||
return srv
|
||
|
||
def test_set_trace_file_and_append(self):
|
||
with tempfile.TemporaryDirectory() as d:
|
||
state_path = os.path.join(d, "host-state.json")
|
||
srv = self._bare_server()
|
||
srv.set_trace_file(state_path, "abc/def?id=1")
|
||
self.assertEqual(
|
||
srv.trace_path,
|
||
os.path.join(d, "host-traces", "abc-def-id-1.jsonl"))
|
||
srv._append_trace(_msg({"type": "commandExecution", "id": "c1",
|
||
"command": "ls", "cwd": "/w",
|
||
"exitCode": 0, "durationMs": 3}))
|
||
srv._append_trace(_msg({"type": "userMessage", "id": "u1",
|
||
"text": "hi"}))
|
||
with open(srv.trace_path, encoding="utf-8") as f:
|
||
lines = f.read().strip().split("\n")
|
||
self.assertEqual(len(lines), 1)
|
||
self.assertEqual(json.loads(lines[0])["kind"], "exec")
|
||
|
||
def test_set_trace_file_failure_keeps_none(self):
|
||
with tempfile.TemporaryDirectory() as d:
|
||
# host-traces 已被普通文件占住，建目录失败时不抛、不影响回合。
|
||
blocker = os.path.join(d, "host-traces")
|
||
with open(blocker, "w", encoding="utf-8") as f:
|
||
f.write("x")
|
||
srv = self._bare_server()
|
||
srv.set_trace_file(os.path.join(d, "host-state.json"), "t1")
|
||
self.assertIsNone(srv.trace_path)
|
||
|
||
def test_append_stops_at_max_lines(self):
|
||
with tempfile.TemporaryDirectory() as d:
|
||
srv = self._bare_server()
|
||
srv.trace_path = os.path.join(d, "t.jsonl")
|
||
srv._trace_lines = TRACE_MAX_LINES - 1
|
||
srv._append_trace(_msg({"type": "commandExecution", "id": "c1",
|
||
"command": "a"}))
|
||
srv._append_trace(_msg({"type": "commandExecution", "id": "c2",
|
||
"command": "b"}))
|
||
with open(srv.trace_path, encoding="utf-8") as f:
|
||
lines = f.read().strip().split("\n")
|
||
self.assertEqual(len(lines), 1)
|
||
self.assertEqual(json.loads(lines[0])["id"], "c1")
|
||
|
||
def test_summarize_carries_thread(self):
|
||
# 模型内子代理与主任务同文件时，服务端凭 thread 归属条目。
|
||
entries = _summarize_trace_item(
|
||
"item/completed",
|
||
{"item": {"type": "commandExecution", "id": "c1",
|
||
"command": "ls"},
|
||
"threadId": "th-sub", "completedAtMs": 1700000000000})
|
||
self.assertEqual(entries[0]["thread"], "th-sub")
|
||
entries = _summarize_trace_item(
|
||
"item/completed",
|
||
{"item": {"type": "commandExecution", "id": "c2",
|
||
"command": "ls"},
|
||
"completedAtMs": 1700000000000})
|
||
self.assertIsNone(entries[0]["thread"])
|
||
|
||
def test_append_trace_marker_to_parent_file(self):
|
||
# delegate 向父链路报到：路径清洗与 set_trace_file 一致，
|
||
# 写失败不抛（尽力而为，不影响回合）。
|
||
from host import _append_trace_marker, _trace_file_path
|
||
with tempfile.TemporaryDirectory() as d:
|
||
state_path = os.path.join(d, "host-state.json")
|
||
self.assertEqual(
|
||
_trace_file_path(state_path, "abc/def?id=1"),
|
||
os.path.join(d, "host-traces", "abc-def-id-1.jsonl"))
|
||
_append_trace_marker(
|
||
state_path, "parent task!",
|
||
{"kind": "agent-start", "id": "agent:rev",
|
||
"name": "rev", "prompt": "review"})
|
||
path = os.path.join(d, "host-traces", "parent-task-.jsonl")
|
||
with open(path, encoding="utf-8") as f:
|
||
entry = json.loads(f.read().strip())
|
||
self.assertEqual(entry["kind"], "agent-start")
|
||
self.assertEqual(entry["name"], "rev")
|
||
# 目录被文件占住时也不抛。
|
||
blocker = os.path.join(d, "host-traces")
|
||
shutil.rmtree(blocker)
|
||
with open(blocker, "w", encoding="utf-8") as f:
|
||
f.write("x")
|
||
_append_trace_marker(state_path, "parent task!",
|
||
{"kind": "agent-done", "id": "agent:rev:done",
|
||
"name": "rev", "status": "completed"})
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|