287 lines
9.1 KiB
Go
287 lines
9.1 KiB
Go
package repo
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/agentmail/gateway/internal/db"
|
||
"github.com/google/uuid"
|
||
)
|
||
|
||
func TestPreviewTruncatesOnUTF8Boundary(t *testing.T) {
|
||
// 「巡」是 3 字节；在 max=4 处切会切进第 2 个字符中间
|
||
s := "巡检报告"
|
||
got := preview(s, 4)
|
||
if got != "巡..." {
|
||
t.Fatalf("按 UTF-8 边界截断失败：%q", got)
|
||
}
|
||
for i, r := range got {
|
||
if r == 0xFFFD {
|
||
t.Fatalf("位置 %d 出现替换符，说明切在了字符中间", i)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestPreviewKeepsShortBodyIntact(t *testing.T) {
|
||
if got := preview("短正文", 240); got != "短正文" {
|
||
t.Fatalf("未超长却被改动：%q", got)
|
||
}
|
||
}
|
||
|
||
// seedReply 插一封回复：parent 指向来信。
|
||
func seedReply(t *testing.T, sessionID uuid.UUID, parent uuid.UUID, from, to, subject string) uuid.UUID {
|
||
t.Helper()
|
||
var id uuid.UUID
|
||
err := db.DB.QueryRowContext(context.Background(), `
|
||
INSERT INTO mails (session_id, parent_mail_id, from_name, from_workspace,
|
||
to_name, to_workspace, subject, body, cc_list, created_at)
|
||
VALUES ($1, $2, $3, '', $4, '', $5, 'body', '[]', $6)
|
||
RETURNING mail_id
|
||
`, sessionID, parent, from, to, subject, nextSeedTime()).Scan(&id)
|
||
if err != nil {
|
||
t.Fatalf("seed reply: %v", err)
|
||
}
|
||
return id
|
||
}
|
||
|
||
// 线索根定位：整棵树从根展开，所以这一步错了后面全错。
|
||
func TestThreadRootOf(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "root-of")
|
||
|
||
root := seedMailIn(t, sid, "admin", "dsh", "原件")
|
||
mid := seedReply(t, sid, root, "dsh", "admin", "Re: 原件")
|
||
leaf := seedReply(t, sid, mid, "admin", "dsh", "Re: Re: 原件")
|
||
|
||
cases := []struct {
|
||
name string
|
||
from uuid.UUID
|
||
depth int
|
||
}{
|
||
{"从根本身出发", root, 0},
|
||
{"从中间一封出发", mid, 1},
|
||
{"从叶子出发", leaf, 2},
|
||
}
|
||
for _, c := range cases {
|
||
t.Run(c.name, func(t *testing.T) {
|
||
gotRoot, gotDepth, err := ThreadRootOf(context.Background(), c.from)
|
||
if err != nil {
|
||
t.Fatalf("ThreadRootOf: %v", err)
|
||
}
|
||
if gotRoot != root {
|
||
t.Errorf("根定位错误：得到 %s，期望 %s", gotRoot, root)
|
||
}
|
||
if gotDepth != c.depth {
|
||
t.Errorf("层数错误：得到 %d，期望 %d", gotDepth, c.depth)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// 这个测试是对话树那次故障的回归：
|
||
// 一封抄送给两个 Agent 的邮件收到两个回复，它们互为**兄弟**。
|
||
// 旧实现从锚点分「祖先方向 + 子孙方向」两路展开，兄弟既不是锚点的祖先
|
||
// 也不是它的子孙，于是整条分支在树里根本不出现。
|
||
// 从线索根 BFS 之后，兄弟都是根的子孙，必须一次全出来。
|
||
func TestDescendantsFromRootIncludesSiblings(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "siblings")
|
||
|
||
root := seedMailIn(t, sid, "admin", "dsh", "测试抄送")
|
||
replyA := seedReply(t, sid, root, "dsh", "admin", "Re: 测试抄送")
|
||
replyB := seedReply(t, sid, root, "opencode", "admin", "Re: 测试抄送")
|
||
|
||
// 从 replyA 出发定位根，再从根整树展开
|
||
gotRoot, anchorDepth, err := ThreadRootOf(context.Background(), replyA)
|
||
if err != nil {
|
||
t.Fatalf("ThreadRootOf: %v", err)
|
||
}
|
||
if gotRoot != root || anchorDepth != 1 {
|
||
t.Fatalf("根定位错误：root=%s depth=%d", gotRoot, anchorDepth)
|
||
}
|
||
|
||
nodes, hasMore, err := DescendantsRaw(context.Background(), gotRoot, 0, 50)
|
||
if err != nil {
|
||
t.Fatalf("DescendantsRaw: %v", err)
|
||
}
|
||
if hasMore {
|
||
t.Error("三封邮件不该报 hasMore")
|
||
}
|
||
|
||
byID := map[uuid.UUID]TreeMail{}
|
||
for _, n := range nodes {
|
||
byID[n.ID] = n
|
||
}
|
||
for name, id := range map[string]uuid.UUID{"根": root, "回复A": replyA, "回复B": replyB} {
|
||
if _, ok := byID[id]; !ok {
|
||
t.Errorf("%s 不在树里 —— 兄弟分支又丢了", name)
|
||
}
|
||
}
|
||
if byID[root].Depth != 0 {
|
||
t.Errorf("根的深度应为 0，实际 %d", byID[root].Depth)
|
||
}
|
||
if byID[replyA].Depth != 1 || byID[replyB].Depth != 1 {
|
||
t.Errorf("两个回复都应在深度 1：A=%d B=%d", byID[replyA].Depth, byID[replyB].Depth)
|
||
}
|
||
}
|
||
|
||
// 转发落在**另一个会话**里，但 parent 仍指向原件。
|
||
// 树必须跨会话展开，否则「这条线索转发给谁了」就看不见了。
|
||
func TestDescendantsCrossSession(t *testing.T) {
|
||
setupTestDB(t)
|
||
srcSession := seedSessionRow(t, "fwd-src")
|
||
dstSession := seedSessionRow(t, "fwd-dst")
|
||
|
||
root := seedMailIn(t, srcSession, "admin", "dsh", "原件")
|
||
// 转发：新会话，parent 仍指原件
|
||
fwd := seedReply(t, dstSession, root, "admin", "opencode", "Fwd: 原件")
|
||
// 转发的下游回复，还在新会话里
|
||
fwdReply := seedReply(t, dstSession, fwd, "opencode", "admin", "Re: Fwd: 原件")
|
||
|
||
nodes, _, err := DescendantsRaw(context.Background(), root, 0, 50)
|
||
if err != nil {
|
||
t.Fatalf("DescendantsRaw: %v", err)
|
||
}
|
||
found := map[uuid.UUID]int{}
|
||
for _, n := range nodes {
|
||
found[n.ID] = n.Depth
|
||
}
|
||
if _, ok := found[fwd]; !ok {
|
||
t.Error("转发不在树里 —— 跨会话展开失效")
|
||
}
|
||
if _, ok := found[fwdReply]; !ok {
|
||
t.Error("转发的下游回复不在树里")
|
||
}
|
||
if found[fwd] != 1 || found[fwdReply] != 2 {
|
||
t.Errorf("跨会话深度错误：fwd=%d fwdReply=%d（期望 1/2）", found[fwd], found[fwdReply])
|
||
}
|
||
// 会话不同 → session_id 必须如实反映，否则前端无法标出「线索去了别的会话」
|
||
for _, n := range nodes {
|
||
if n.ID == fwd && n.SessionID != dstSession {
|
||
t.Errorf("转发的 session_id 错误：%s，期望 %s", n.SessionID, dstSession)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 分页：BFS 顺序稳定，两页拼起来等于一次全取。
|
||
func TestDescendantsPaginationStable(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "paging")
|
||
|
||
root := seedMailIn(t, sid, "admin", "dsh", "原件")
|
||
for i := 0; i < 5; i++ {
|
||
seedReply(t, sid, root, "dsh", "admin", "Re: 原件")
|
||
}
|
||
|
||
full, hasMoreFull, err := DescendantsRaw(context.Background(), root, 0, 50)
|
||
if err != nil {
|
||
t.Fatalf("全取: %v", err)
|
||
}
|
||
if hasMoreFull {
|
||
t.Error("6 封邮件一次取完不该报 hasMore")
|
||
}
|
||
if len(full) != 6 {
|
||
t.Fatalf("应有 6 个节点，实际 %d", len(full))
|
||
}
|
||
|
||
page1, hasMore1, err := DescendantsRaw(context.Background(), root, 0, 4)
|
||
if err != nil {
|
||
t.Fatalf("第一页: %v", err)
|
||
}
|
||
if !hasMore1 {
|
||
t.Error("还有 2 封没取，hasMore 应为真")
|
||
}
|
||
page2, hasMore2, err := DescendantsRaw(context.Background(), root, 4, 4)
|
||
if err != nil {
|
||
t.Fatalf("第二页: %v", err)
|
||
}
|
||
if hasMore2 {
|
||
t.Error("第二页已取完，hasMore 应为假")
|
||
}
|
||
|
||
joined := append(append([]TreeMail{}, page1...), page2...)
|
||
if len(joined) != len(full) {
|
||
t.Fatalf("两页拼接 %d 个，全取 %d 个", len(joined), len(full))
|
||
}
|
||
for i := range full {
|
||
if joined[i].ID != full[i].ID {
|
||
t.Fatalf("第 %d 个节点顺序不一致：分页 %s，全取 %s —— BFS 顺序不稳定，"+
|
||
"分页加载会重复或漏掉节点", i, joined[i].ID, full[i].ID)
|
||
}
|
||
}
|
||
}
|
||
|
||
// TreeMailByID 是「锚点没落进 BFS 首页」时的回填手段，深度由调用方给。
|
||
func TestTreeMailByIDUsesGivenDepth(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "by-id")
|
||
id := seedMailIn(t, sid, "admin", "dsh", "某封")
|
||
|
||
got, err := TreeMailByID(context.Background(), id, 7)
|
||
if err != nil {
|
||
t.Fatalf("TreeMailByID: %v", err)
|
||
}
|
||
if got.ID != id {
|
||
t.Errorf("取错了邮件：%s", got.ID)
|
||
}
|
||
if got.Depth != 7 {
|
||
t.Errorf("深度应取调用方给的 7，实际 %d", got.Depth)
|
||
}
|
||
// 树视图只要预览，全文必须被清空 —— 否则整条线索会把几百 KB 正文塞给前端
|
||
if got.Body != "" {
|
||
t.Errorf("Body 应清空，实际 %q", got.Body)
|
||
}
|
||
if got.BodyPreview == "" {
|
||
t.Error("BodyPreview 应有内容")
|
||
}
|
||
}
|
||
|
||
// 抄送列表必须原样带出来：树上两个兄弟节点为什么并列，
|
||
// 唯一的解释就是父邮件抄送给了两个人。丢了 cc_list 前端就没法说明。
|
||
func TestTreeCarriesCCList(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "cc-carry")
|
||
|
||
var root uuid.UUID
|
||
err := db.DB.QueryRowContext(context.Background(), `
|
||
INSERT INTO mails (session_id, from_name, from_workspace, to_name, to_workspace,
|
||
subject, body, cc_list, created_at)
|
||
VALUES ($1, 'admin', '', 'dsh', '', '抄送两人', 'body',
|
||
'[{"name":"opencode","path":"/home","session":"new","raw":"opencode@/home.new"}]', $2)
|
||
RETURNING mail_id
|
||
`, sid, nextSeedTime()).Scan(&root)
|
||
if err != nil {
|
||
t.Fatalf("seed: %v", err)
|
||
}
|
||
|
||
nodes, _, err := DescendantsRaw(context.Background(), root, 0, 10)
|
||
if err != nil {
|
||
t.Fatalf("DescendantsRaw: %v", err)
|
||
}
|
||
if len(nodes) != 1 {
|
||
t.Fatalf("应有 1 个节点，实际 %d", len(nodes))
|
||
}
|
||
if len(nodes[0].CCList) != 1 {
|
||
t.Fatalf("抄送应有 1 人，实际 %d —— cc_list 没带出来", len(nodes[0].CCList))
|
||
}
|
||
if nodes[0].CCList[0].Raw != "opencode@/home.new" {
|
||
t.Errorf("抄送 raw 错误：%q", nodes[0].CCList[0].Raw)
|
||
}
|
||
}
|
||
|
||
// 无抄送时 cc_list 必须是空数组而不是 null：
|
||
// Go 的 nil slice 会序列化成 null，前端 node.cc_list.length 直接抛异常。
|
||
func TestTreeEmptyCCIsArrayNotNull(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "cc-empty")
|
||
root := seedMailIn(t, sid, "admin", "dsh", "无抄送")
|
||
|
||
nodes, _, err := DescendantsRaw(context.Background(), root, 0, 10)
|
||
if err != nil {
|
||
t.Fatalf("DescendantsRaw: %v", err)
|
||
}
|
||
if nodes[0].CCList == nil {
|
||
t.Error("cc_list 为 nil，会序列化成 null")
|
||
}
|
||
}
|