mirror of
https://gitcode.com/JianFeeeee/homeagent-sdk.git
synced 2026-10-31 14:27:30 +00:00
28c162cf81
## 缺陷（用户实测反馈） 「commondev 无法取消正在运行的 session，agent 反馈它无法取消错误的 pi session，只能等它跑完。」 日志佐证：agent 在 09:07 收到 interrupt 请求后只能如实回答无法取消。 ## 根因不是"没实现取消"，而是取消能力从来没被暴露 pi 的 AgentSession **本身就有** abort()（agent-session.d.ts:598， "Abort current operation and wait for agent to become idle"）。 但 commondev 注册的 30 个工具里一个 cancel/stop 都没有， sidecar 也只有 ping/prompt/state/close 四个 method。 这是异步投递（2026-10-17）带来的新缺口：改动之前 prompt 是同步的， 调用方阻塞到任务结束，没有"中途要停"这个需求；改成异步投递后， 任务在后台长跑，用户却失去了停止它的手段。 ## 改动 **sidecar 三个平台各有各的取消方式：** pi → session.abort()（协作式中断，必须 await：不 await 会立刻 回"已中断"而实际还在跑 = 谎报成功） zcode → 整组杀掉当前回合的子进程（没有 abort API） spawn 时 detached=true 建了独立进程组，killTree 整组杀 **interrupt 语义如实回报：** interrupted=true 确实有任务在跑并被终止 interrupted=false 当时没有在跑的任务（不算错误） ok=false + error 该 CLI 没实现 / 调用失败 **三个收尾分支都清 currentChild**（timeout / error / close）： 不清的话 interrupt 会拿到已死的子进程去杀，或更糟 —— 误杀下一轮。 ## 判据（4 条 + 「不谎报」） · pi 的两个 sidecar 都有 interrupt 且用 await session.abort() · zcode 的 interrupt 真的 killTree 且用 currentChild · 三个 sidecar 无重复声明（我写的时候真撞了：zcode 已有 currentChild） · **不谎报成功**：sidecar 没跑 / CLI 没实现时必须 interrupted=false ## 变异验证 pi 删 interrupt case → 咬 pi 去掉 await → 咬（"会谎报已中断"） zcode 不 killTree → 咬（"只回 ok 不真杀"） ## 顺带 既有判据 TestEveryRegisteredToolHasHandler 立刻抓到我加了 RegisterTool 但没在 handlerFor 映射里登记 —— 它比我先发现漏了接线。
341 lines
12 KiB
Go
341 lines
12 KiB
Go
package main
|
||
|
||
// 对齐审计：**逐个调用**已注册工具，检查它们真的能工作。
|
||
//
|
||
// 为何要这个：注册了名字 ≠ 能用。上一轮补了工具，但「能注册」只证明
|
||
// 编译通过。本文件对每个工具做一次最小真实调用，把「写了但没接上」
|
||
// （空返回、nil handler、路径写死错、参数名不匹配）暴露出来。
|
||
//
|
||
// 判定原则：
|
||
// - 返回 error ≠ 失败（很多工具对缺参数就是返回 error，那是正确的响亮行为）
|
||
// - 真正的失败是：panic、handler 为 nil、或**参数齐全却依然拒绝**
|
||
// - 这里的目标是发现"连正确调用都走不通"的工具
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
// listRegisteredTools 从 plugin.go 里抽出注册的工具名。
|
||
// 用源码而非运行期 sdk，因为 SDK 的 mock 不暴露注册表。
|
||
func listRegisteredTools(t *testing.T) []string {
|
||
t.Helper()
|
||
b, err := os.ReadFile("plugin.go")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
var out []string
|
||
for _, line := range strings.Split(string(b), "\n") {
|
||
line = strings.TrimSpace(line)
|
||
const p = `p.sdk.RegisterTool("`
|
||
if i := strings.Index(line, p); i >= 0 {
|
||
rest := line[i+len(p):]
|
||
if j := strings.Index(rest, `"`); j > 0 {
|
||
out = append(out, rest[:j])
|
||
}
|
||
}
|
||
}
|
||
return out
|
||
}
|
||
|
||
// handlerFor 通过名字映射到实际 handler 函数。
|
||
//
|
||
// ★ 这里刻意**手写映射**而不是反射：如果新增了工具却忘了在这里登记，
|
||
// 下面的覆盖率判据会报出来 —— 这正好是"新工具没接 handler"的探测。
|
||
func handlerFor(p *Plugin, name string) func(map[string]interface{}) (interface{}, error) {
|
||
switch name {
|
||
case "commondev_project_add", "commondev_project_open":
|
||
return p.handleProjectAdd
|
||
case "commondev_project_list":
|
||
return p.handleProjectList
|
||
case "commondev_branch_open":
|
||
return p.handleBranchOpen
|
||
case "commondev_branch_status":
|
||
return p.handleBranchStatus
|
||
case "commondev_branch_close":
|
||
return p.handleBranchClose
|
||
case "commondev_job_status":
|
||
return p.handleJobStatus
|
||
case "commondev_agent_up":
|
||
return p.handleAgentUp
|
||
case "commondev_agent_state":
|
||
return p.handleAgentState
|
||
case "commondev_agent_down":
|
||
return p.handleAgentDown
|
||
case "commondev_workdir_prepare":
|
||
return p.handleWorkdirPrepare
|
||
case "commondev_workdir_clean":
|
||
return p.handleWorkdirClean
|
||
case "commondev_test_define":
|
||
return p.handleTestDefine
|
||
case "commondev_test_list":
|
||
return p.handleTestList
|
||
case "commondev_test_run":
|
||
return p.handleTestRun
|
||
case "commondev_test_mutate":
|
||
return p.handleTestMutate
|
||
case "commondev_review_submit":
|
||
return p.handleReviewSubmit
|
||
case "commondev_review_result":
|
||
return p.handleReviewResult
|
||
case "commondev_conflict_check":
|
||
return p.handleConflictCheck
|
||
case "commondev_parallel_status":
|
||
return p.handleParallelStatus
|
||
case "commondev_pi_config_get":
|
||
return p.handlePiConfigGet
|
||
case "commondev_pi_config_set":
|
||
return p.handlePiConfigSet
|
||
case "commondev_pi_source_add":
|
||
return p.handlePiSourceAdd
|
||
case "commondev_pi_source_list":
|
||
return p.handlePiSourceList
|
||
case "commondev_pi_source_remove":
|
||
return p.handlePiSourceRemove
|
||
case "commondev_pi_source_default":
|
||
return p.handlePiSourceDefault
|
||
case "commondev_pi_source_import":
|
||
return p.handlePiSourceImport
|
||
case "commondev_pi_package_install", "commondev_pi_plugin_install":
|
||
return p.handlePiPackageInstall
|
||
case "commondev_pi_plugin_list":
|
||
return p.handlePiPluginList
|
||
case "commondev_pi_plugin_remove":
|
||
return p.handlePiPluginRemove
|
||
case "commondev_pi_plugin_update":
|
||
return p.handlePiPluginUpdate
|
||
case "commondev_pi_mcp_list":
|
||
return p.handlePiMCPList
|
||
case "commondev_pi_mcp_add":
|
||
return p.handlePiMCPAdd
|
||
case "commondev_pi_mcp_remove":
|
||
return p.handlePiMCPRemove
|
||
case "enc_store":
|
||
return p.handleEncStore
|
||
case "enc_list":
|
||
return p.handleEncList
|
||
case "enc_show":
|
||
return p.handleEncShow
|
||
case "enc_rm":
|
||
return p.handleEncRm
|
||
case "enc_shell":
|
||
return p.handleEncryptedShell
|
||
case "commondev_endpoint_set":
|
||
return p.handleEndpointSet
|
||
case "commondev_endpoint_list":
|
||
return p.handleEndpointList
|
||
case "commondev_endpoint_remove":
|
||
return p.handleEndpointRemove
|
||
case "commondev_agent_interrupt":
|
||
return p.handleAgentInterrupt
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ★★ 判据：**每个注册的工具都必须有对应 handler**。
|
||
//
|
||
// 这条能抓出「加了 RegisterTool 却指向不存在的 handler」或
|
||
// 「加了工具但这里没同步登记」（后者是需要人补映射，属于审计信号）。
|
||
func TestEveryRegisteredToolHasHandler(t *testing.T) {
|
||
p := newAuditPlugin(t)
|
||
tools := listRegisteredTools(t)
|
||
if len(tools) < 40 {
|
||
t.Fatalf("注册工具数异常偏少（%d），插件文件可能没被正确解析", len(tools))
|
||
}
|
||
var missing []string
|
||
for _, name := range tools {
|
||
if handlerFor(p, name) == nil {
|
||
missing = append(missing, name)
|
||
}
|
||
}
|
||
if len(missing) > 0 {
|
||
t.Fatalf("★ 这些工具没有可用 handler（注册了但没接上）: %v", missing)
|
||
}
|
||
t.Logf("已注册 %d 个工具，全部有 handler", len(tools))
|
||
}
|
||
|
||
// ★★ 判据：**无参数调用不得 panic**。
|
||
//
|
||
// 工具会遇到模型乱传参数（缺字段、类型错）。正确行为是返回 error，
|
||
// 不是崩掉整个插件进程。
|
||
func TestNoToolPanicsOnEmptyArgs(t *testing.T) {
|
||
p := newAuditPlugin(t)
|
||
for _, name := range listRegisteredTools(t) {
|
||
h := handlerFor(p, name)
|
||
if h == nil {
|
||
continue
|
||
}
|
||
func() {
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
t.Errorf("★ %s 空参数 panic: %v", name, r)
|
||
}
|
||
}()
|
||
// 空 map + nil map 都要试：模型可能传 null
|
||
h(map[string]interface{}{})
|
||
h(nil)
|
||
}()
|
||
}
|
||
}
|
||
|
||
// ★★★ 判据：**本轮补的工具必须真的能完成一次正常调用**。
|
||
//
|
||
// 与前两条不同：这里给足参数，检查业务是否真走通。
|
||
// 「注册 + 不 panic」还不够 —— 一条写死的错误路径也能满足那两条。
|
||
func TestNewlyAddedToolsActuallyWork(t *testing.T) {
|
||
p := newAuditPlugin(t)
|
||
p.ledger.Projects["proj"] = &Project{
|
||
Name: "proj", RemoteURL: "https://example.invalid/o/r.git", DefaultBranch: "main"}
|
||
dir := t.TempDir()
|
||
p.ledger.Branches["br-1"] = &Branch{
|
||
ID: "br-1", ProjectID: "proj", Name: "feat/x", Workdir: dir, Status: "open"}
|
||
|
||
// ① agent_state 应报出真实状态（此前完全不存在）
|
||
res, err := p.handleAgentState(map[string]interface{}{"branch_id": "br-1"})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
m := res.(map[string]interface{})
|
||
if m["branch_id"] != "br-1" || m["agent_alive"] != false {
|
||
t.Errorf("agent_state 返回不对: %v", m)
|
||
}
|
||
|
||
// ② agent_down 应成功停掉（无进程时也幂等成功）
|
||
res2, err := p.handleAgentDown(map[string]interface{}{"branch_id": "br-1"})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if res2.(map[string]interface{})["status"] != "ok" {
|
||
t.Errorf("agent_down 未成功: %v", res2)
|
||
}
|
||
|
||
// ③ workdir_clean 默认 dry_run，且能识别孤儿
|
||
root := p.programRoot()
|
||
os.MkdirAll(filepath.Join(root, "deadbeef-feat_x"), 0o755)
|
||
res3, err := p.handleWorkdirClean(map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
m3 := res3.(map[string]interface{})
|
||
if m3["status"] != "dry_run" {
|
||
t.Errorf("workdir_clean 默认应为 dry_run: %v", m3)
|
||
}
|
||
if n, _ := m3["count"].(int); n != 1 {
|
||
t.Errorf("应识别 1 个孤儿目录，实际 %v", m3["orphans"])
|
||
}
|
||
|
||
// ④ pi_config_get 必须能读到结构（即使没有任何 provider）
|
||
res4, err := p.handlePiConfigGet(map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, ok := res4.(map[string]interface{})["providers"]; !ok {
|
||
t.Errorf("pi_config_get 缺少 providers 字段: %v", res4)
|
||
}
|
||
}
|
||
|
||
// ★★ 判据：**设计文档点名的工具必须真的存在**。
|
||
//
|
||
// 这条把「文档说做了」与「代码里有」绑起来 —— 上一轮就是靠它
|
||
// 发现 pr_open / agent_state 等工具根本不存在。
|
||
//
|
||
// ⚠ 2026-10-15 起：**git 托管平台专有的工具已不在清单里**
|
||
// （account_* / repo_search / pr_* / protect_branch）—— 那些能力进 skill，
|
||
// 由 agent 用 safeshell + curl 完成。
|
||
func TestDesignNamedToolsExist(t *testing.T) {
|
||
tools := listRegisteredTools(t)
|
||
have := map[string]bool{}
|
||
for _, n := range tools {
|
||
have[n] = true
|
||
}
|
||
want := []string{
|
||
"commondev_project_add", "commondev_project_list",
|
||
"commondev_branch_open", "commondev_branch_status", "commondev_branch_close",
|
||
"commondev_workdir_prepare", "commondev_workdir_clean",
|
||
"commondev_agent_up", "commondev_agent_state", "commondev_agent_down",
|
||
"commondev_pi_plugin_install", "commondev_pi_plugin_remove", "commondev_pi_plugin_list",
|
||
"commondev_pi_source_import", "commondev_pi_source_add", "commondev_pi_source_list",
|
||
"commondev_pi_source_default", "commondev_pi_config_get",
|
||
"commondev_test_define", "commondev_test_run", "commondev_test_mutate",
|
||
"commondev_review_submit",
|
||
}
|
||
var missing []string
|
||
for _, w := range want {
|
||
if !have[w] {
|
||
missing = append(missing, w)
|
||
}
|
||
}
|
||
if len(missing) > 0 {
|
||
t.Fatalf("设计点名的工具缺失: %v", missing)
|
||
}
|
||
}
|
||
|
||
// ★ 判据：**git 平台专有的工具已全部移出插件**。
|
||
//
|
||
// 这条钉住 2026-10-15 的架构决定：平台知识进 skill。若有人又把
|
||
// account_* / pr_* / protect_branch 加回插件，这里会报红。
|
||
func TestPlatformToolsAreNotInPlugin(t *testing.T) {
|
||
tools := listRegisteredTools(t)
|
||
have := map[string]bool{}
|
||
for _, n := range tools {
|
||
have[n] = true
|
||
}
|
||
for _, forbidden := range []string{
|
||
"commondev_account_add", "commondev_account_list", "commondev_account_test",
|
||
"commondev_account_remove", "commondev_repo_search",
|
||
"commondev_pr_open", "commondev_pr_status", "commondev_pr_approve",
|
||
"commondev_pr_merge", "commondev_protect_branch",
|
||
} {
|
||
if have[forbidden] {
|
||
t.Errorf("★ %s 属 git 平台专有能力，不应再注册在插件里（应进 skill + safeshell）", forbidden)
|
||
}
|
||
}
|
||
}
|
||
|
||
// ★★ 判据：**平台专有代码不得回到插件里**。
|
||
//
|
||
// 钉住 2026-10-15 的架构决定（用户拍板）：git 托管平台的知识进 skill，
|
||
// 插件只保留与平台无关的机制。若有人把 platformSpec / 按平台构造认证头
|
||
// 又加回来，这里会报红。
|
||
//
|
||
// 为什么要用判据而不只是文档：文档说了还会有人（包括将来的我）图省事加回去，
|
||
// 而那种回退在**功能上看起来完全正常** —— 只是把可移植性换成了硬编码。
|
||
func TestNoPlatformSpecificCodeInPlugin(t *testing.T) {
|
||
// 逐文件扫源码（排除测试与注释性提及由人工判断）
|
||
files := []string{"plugin.go", "credentials.go", "pi_tools.go", "criteria.go",
|
||
"encstore.go", "encshell.go", "conflict.go", "workdir.go", "workdir_security.go",
|
||
"util.go", "llmsource.go", "sidecar.go"}
|
||
forbidden := map[string]string{
|
||
"platformSpec(": "按平台构造请求（认证头/端点差异）应进 skill",
|
||
"plaftformSpec": "平台适配结构体",
|
||
"DefaultAPI": "硬编码平台 API 前缀",
|
||
"AuthHeader:": "硬编码平台认证头",
|
||
"RoleExecutor": "平台角色语义",
|
||
"handleAccountAdd": "平台凭据工具",
|
||
"handleRepoSearch": "平台搜索工具",
|
||
"handlePROpen": "平台 PR 工具",
|
||
"handleProtectBranch": "平台分支保护工具",
|
||
}
|
||
for _, f := range files {
|
||
b, err := os.ReadFile(f)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
src := string(b)
|
||
for sym, why := range forbidden {
|
||
// 只认代码出现，不认注释里的解释性提及
|
||
for i, line := range strings.Split(src, "\n") {
|
||
trimmed := strings.TrimSpace(line)
|
||
if strings.HasPrefix(trimmed, "//") {
|
||
continue // 注释里说明"已移出"是允许的
|
||
}
|
||
if strings.Contains(line, sym) {
|
||
t.Errorf("★ %s:%d 含平台专有代码 %q（%s）", f, i+1, sym, why)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|