120 lines
3.8 KiB
Go
120 lines
3.8 KiB
Go
package repo
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/agentmail/gateway/internal/db"
|
||
)
|
||
|
||
// 权限询问的决策者必须是人：Agent 收不到 SendToUser，桥的 await 永不 resolve。
|
||
// 生产事故：pi 把任务派给自己的另一条会话 → 那条会话要跑 bash → 权限邮件发给 "pi"
|
||
// → pi 不是人类用户 → 整条会话永久阻塞。
|
||
//
|
||
// 修法是顺着会话的邮件链上溯找最近的人类 —— 派活的人才是该点头的人。
|
||
func TestNearestHumanInThread(t *testing.T) {
|
||
setupTestDB(t)
|
||
ctx := context.Background()
|
||
|
||
mustExec(t, ctx, `INSERT INTO users (username, display_name, password_hash, role)
|
||
VALUES ('alice','Alice','x','user')`)
|
||
for _, a := range []string{"opencode", "dsh", "pi"} {
|
||
if err := CreateOrUpdateAgent(ctx, a, "s", "test", nil); err != nil {
|
||
t.Fatalf("注册 %s: %v", a, err)
|
||
}
|
||
}
|
||
|
||
t.Run("沿链上溯找到派活的人", func(t *testing.T) {
|
||
sid, err := CreateSession(ctx, nil, "opencode", "任务链", "/home")
|
||
if err != nil {
|
||
t.Fatalf("建会话: %v", err)
|
||
}
|
||
// alice → opencode → dsh，dsh 触发权限询问
|
||
m1, err := CreateMail(ctx, sid, nil, "alice", "", "opencode", "", "任务", "请帮忙", nil)
|
||
if err != nil {
|
||
t.Fatalf("封1: %v", err)
|
||
}
|
||
m2, err := CreateMail(ctx, sid, &m1, "opencode", "", "dsh", "", "转派", "你来看", nil)
|
||
if err != nil {
|
||
t.Fatalf("封2: %v", err)
|
||
}
|
||
if _, err := CreateMail(ctx, sid, &m2, "dsh", "", "opencode", "", "进展", "做了一半", nil); err != nil {
|
||
t.Fatalf("封3: %v", err)
|
||
}
|
||
|
||
for _, agent := range []string{"dsh", "opencode"} {
|
||
got, err := NearestHumanInThread(ctx, sid, agent)
|
||
if err != nil {
|
||
t.Fatalf("%s: %v", agent, err)
|
||
}
|
||
if got != "alice" {
|
||
t.Errorf("%s 触发权限时应路由到 alice，得到 %q", agent, got)
|
||
}
|
||
}
|
||
})
|
||
|
||
t.Run("全 Agent 链返回空串", func(t *testing.T) {
|
||
// 没有人类参与的链条：调用方据此拒绝请求，而不是转给一个
|
||
// 对上下文一无所知的管理员。
|
||
sid, err := CreateSession(ctx, nil, "opencode", "纯 Agent", "/tmp")
|
||
if err != nil {
|
||
t.Fatalf("建会话: %v", err)
|
||
}
|
||
m1, err := CreateMail(ctx, sid, nil, "opencode", "", "dsh", "", "干活", "go", nil)
|
||
if err != nil {
|
||
t.Fatalf("封1: %v", err)
|
||
}
|
||
if _, err := CreateMail(ctx, sid, &m1, "dsh", "", "opencode", "", "好", "ok", nil); err != nil {
|
||
t.Fatalf("封2: %v", err)
|
||
}
|
||
|
||
got, err := NearestHumanInThread(ctx, sid, "dsh")
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
if got != "" {
|
||
t.Errorf("链上没有人类时应返回空串，得到 %q", got)
|
||
}
|
||
})
|
||
|
||
t.Run("skipAgent 是自己时不会把自己当人", func(t *testing.T) {
|
||
// pi 给自己的另一条会话派活正是事故场景：链上只有 pi 一个名字。
|
||
sid, err := CreateSession(ctx, nil, "pi", "自派", "/home")
|
||
if err != nil {
|
||
t.Fatalf("建会话: %v", err)
|
||
}
|
||
if _, err := CreateMail(ctx, sid, nil, "pi", "", "pi", "", "拆任务", "自己干", nil); err != nil {
|
||
t.Fatalf("封1: %v", err)
|
||
}
|
||
|
||
got, err := NearestHumanInThread(ctx, sid, "pi")
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
if got != "" {
|
||
t.Errorf("自派链上没有人类，应返回空串，得到 %q", got)
|
||
}
|
||
})
|
||
|
||
t.Run("空会话不报错", func(t *testing.T) {
|
||
sid, err := CreateSession(ctx, nil, "opencode", "空", "/tmp")
|
||
if err != nil {
|
||
t.Fatalf("建会话: %v", err)
|
||
}
|
||
got, err := NearestHumanInThread(ctx, sid, "opencode")
|
||
if err != nil {
|
||
t.Fatalf("空会话应返回空串而非报错，得到 err=%v", err)
|
||
}
|
||
if got != "" {
|
||
t.Errorf("空会话应返回空串，得到 %q", got)
|
||
}
|
||
})
|
||
}
|
||
|
||
func mustExec(t *testing.T, ctx context.Context, q string, args ...any) {
|
||
t.Helper()
|
||
if _, err := db.DB.ExecContext(ctx, q, args...); err != nil {
|
||
t.Fatalf("exec %s: %v", q, err)
|
||
}
|
||
}
|