229 lines
7.9 KiB
Go
229 lines
7.9 KiB
Go
package repo
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
"testing"
|
||
|
||
"github.com/agentmail/gateway/internal/db"
|
||
"github.com/google/uuid"
|
||
)
|
||
|
||
// ListSessionsFor 的 Scan 列数必须与 SELECT 一致。
|
||
//
|
||
// 这个测试存在的理由：预算两列（max_rounds/used_rounds）加进了 SELECT 却忘了加进
|
||
// Scan，于是 /me/sessions 整个 500 —— 联系人栏一条数据都拉不到，
|
||
// 而错误信息只是 "Failed to list sessions"，看不出是列数不匹配。
|
||
// 列数错位是纯结构问题，一个最小用例就能钉住。
|
||
func TestListSessionsForScanMatchesSelect(t *testing.T) {
|
||
setupTestDB(t)
|
||
|
||
if _, err := db.DB.ExecContext(context.Background(),
|
||
`INSERT INTO users (username, display_name, password_hash, role)
|
||
VALUES ('alice', 'Alice', 'x', 'user')`); err != nil {
|
||
t.Fatalf("seed user: %v", err)
|
||
}
|
||
|
||
sid := seedSessionRow(t, "list-scan")
|
||
if _, err := db.DB.ExecContext(context.Background(),
|
||
`UPDATE sessions SET max_rounds = 7, used_rounds = 3 WHERE session_id = $1`,
|
||
sid); err != nil {
|
||
t.Fatalf("set budget: %v", err)
|
||
}
|
||
seedMailIn(t, sid, "alice", "opencode", "hello")
|
||
|
||
// 无过滤（管理员 all=true 走这条）
|
||
all, err := ListSessionsFor(context.Background(), "", 50)
|
||
if err != nil {
|
||
t.Fatalf("列出全部会话失败: %v", err)
|
||
}
|
||
if len(all) != 1 {
|
||
t.Fatalf("应有 1 个会话，实际 %d", len(all))
|
||
}
|
||
// 预算两列要真的读出来，不是零值
|
||
if all[0].MaxRounds != 7 || all[0].UsedRounds != 3 {
|
||
t.Errorf("预算未读出：max=%d used=%d（期望 7/3）",
|
||
all[0].MaxRounds, all[0].UsedRounds)
|
||
}
|
||
if all[0].MailCount != 1 {
|
||
t.Errorf("邮件数应为 1，实际 %d —— 列顺序可能错位", all[0].MailCount)
|
||
}
|
||
|
||
// 带用户过滤（普通用户走这条，SQL 分支不同，要分别验）
|
||
mine, err := ListSessionsFor(context.Background(), "alice", 50)
|
||
if err != nil {
|
||
t.Fatalf("列出自己的会话失败: %v", err)
|
||
}
|
||
if len(mine) != 1 {
|
||
t.Fatalf("alice 参与过该会话，应能看到，实际 %d 个", len(mine))
|
||
}
|
||
if mine[0].MaxRounds != 7 || mine[0].MailCount != 1 {
|
||
t.Errorf("过滤分支的列顺序错位：%+v", mine[0])
|
||
}
|
||
|
||
// 与自己无关的人看不到
|
||
other, err := ListSessionsFor(context.Background(), "bob", 50)
|
||
if err != nil {
|
||
t.Fatalf("列出 bob 的会话失败: %v", err)
|
||
}
|
||
if len(other) != 0 {
|
||
t.Errorf("bob 未参与该会话，不该看到，实际 %d 个", len(other))
|
||
}
|
||
|
||
// 归档会话不出现在列表里
|
||
if _, err := db.DB.ExecContext(context.Background(),
|
||
`UPDATE sessions SET status = 'archived' WHERE session_id = $1`, sid); err != nil {
|
||
t.Fatalf("archive: %v", err)
|
||
}
|
||
after, err := ListSessionsFor(context.Background(), "", 50)
|
||
if err != nil {
|
||
t.Fatalf("归档后列出失败: %v", err)
|
||
}
|
||
if len(after) != 0 {
|
||
t.Errorf("归档会话不该出现在列表里，实际 %d 个", len(after))
|
||
}
|
||
}
|
||
|
||
// 工作列表卡片视图需要「这条线索在干什么 / 还剩几个来回 / 最新进展是什么」，
|
||
// 这些都从 ListContactsFor 一次取回 —— 否则卡片要为每条会话再打一次库。
|
||
func TestListContactsForCardFields(t *testing.T) {
|
||
setupTestDB(t)
|
||
|
||
if _, err := db.DB.ExecContext(context.Background(),
|
||
`INSERT INTO users (username, display_name, password_hash, role)
|
||
VALUES ('alice', 'Alice', 'x', 'user')`); err != nil {
|
||
t.Fatalf("seed user: %v", err)
|
||
}
|
||
|
||
sid := seedSessionRow(t, "card-fields")
|
||
if _, err := db.DB.ExecContext(context.Background(),
|
||
`UPDATE sessions SET subject = '缓存层选型评估', max_rounds = 5, used_rounds = 2
|
||
WHERE session_id = $1`, sid); err != nil {
|
||
t.Fatalf("set session: %v", err)
|
||
}
|
||
|
||
// 三封：最早一封决定联系人身份，最后一封决定「最新进展」
|
||
seedMailIn(t, sid, "alice", "opencode", "第一封")
|
||
seedMailIn(t, sid, "opencode", "alice", "第二封")
|
||
last := seedMailIn(t, sid, "opencode", "alice", "第三封")
|
||
if _, err := db.DB.ExecContext(context.Background(),
|
||
`UPDATE mails SET body = '已经跑完压测，Redis 方案在这个负载下明显更稳。'
|
||
WHERE mail_id = $1`, last); err != nil {
|
||
t.Fatalf("set body: %v", err)
|
||
}
|
||
|
||
got, err := ListContactsFor(context.Background(), "alice", false)
|
||
if err != nil {
|
||
t.Fatalf("列出联系人失败: %v", err)
|
||
}
|
||
if len(got) != 1 {
|
||
t.Fatalf("应有 1 个联系人，实际 %d", len(got))
|
||
}
|
||
c := got[0]
|
||
|
||
if c.Subject != "缓存层选型评估" {
|
||
t.Errorf("主题未带回：%q", c.Subject)
|
||
}
|
||
if c.MaxRounds != 5 || c.UsedRounds != 2 {
|
||
t.Errorf("预算未带回：%d/%d（期望 5/2）", c.UsedRounds, c.MaxRounds)
|
||
}
|
||
// 最新进展取的是【最后】一封，不是第一封
|
||
if c.LastFrom != "opencode" {
|
||
t.Errorf("最新发件人应为 opencode，实际 %q", c.LastFrom)
|
||
}
|
||
if !strings.Contains(c.LastPreview, "Redis 方案") {
|
||
t.Errorf("最新摘要应来自最后一封，实际 %q", c.LastPreview)
|
||
}
|
||
// 联系人身份仍取最早一封的对端
|
||
if c.AgentName != "opencode" {
|
||
t.Errorf("联系人应为 opencode，实际 %q", c.AgentName)
|
||
}
|
||
if c.MailCount != 3 {
|
||
t.Errorf("邮件数应为 3，实际 %d —— 列顺序可能错位", c.MailCount)
|
||
}
|
||
if c.Address != "opencode@.card-fields" && !strings.HasSuffix(c.Address, ".card-fields") {
|
||
t.Errorf("地址应带会话别名，实际 %q", c.Address)
|
||
}
|
||
}
|
||
|
||
// 摘要按字符截断，不按字节 —— 中文一字三字节，裸切会留半个字符。
|
||
func TestPreviewRunes(t *testing.T) {
|
||
cases := []struct {
|
||
in string
|
||
n int
|
||
want string
|
||
}{
|
||
{"短文本", 10, "短文本"},
|
||
{" 两边有空白 ", 10, "两边有空白"},
|
||
{"", 5, ""},
|
||
{"一二三四五六", 3, "一二三…"},
|
||
{"abcdefgh", 3, "abc…"},
|
||
}
|
||
for _, c := range cases {
|
||
if got := previewRunes(c.in, c.n); got != c.want {
|
||
t.Errorf("previewRunes(%q, %d) = %q，期望 %q", c.in, c.n, got, c.want)
|
||
}
|
||
}
|
||
|
||
// 截断结果必须是合法 UTF-8（不含替换字符）
|
||
long := strings.Repeat("汉字", 200)
|
||
got := previewRunes(long, 90)
|
||
if strings.ContainsRune(got, '\uFFFD') {
|
||
t.Error("截断产生了 U+FFFD，说明按字节切了")
|
||
}
|
||
if n := len([]rune(got)); n != 91 { // 90 + 省略号
|
||
t.Errorf("截断后应为 90 字符 + 省略号，实际 %d 字符", n)
|
||
}
|
||
}
|
||
|
||
// 时间戳精度回归：SQLite 的 CURRENT_TIMESTAMP 只有秒，同秒插入的多行排序不确定，
|
||
// 「会话里最早那封」（决定联系人身份）与「最后那封」（决定最新进展）都会取错。
|
||
// NOW() 现在返回毫秒精度，且 mails 的 INSERT 显式传它 —— 这两点都要钉住。
|
||
func TestMailTimestampSubSecond(t *testing.T) {
|
||
setupTestDB(t)
|
||
sid := seedSessionRow(t, "ts-precision")
|
||
|
||
// 连续插 8 封（不显式给时间戳，走 CreateMail 里的 NOW()）
|
||
ids := make([]uuid.UUID, 0, 8)
|
||
for i := 0; i < 8; i++ {
|
||
id, err := CreateMail(context.Background(), sid, nil,
|
||
"alice", "", "opencode", "", fmt.Sprintf("第%d封", i), "body", nil)
|
||
if err != nil {
|
||
t.Fatalf("创建邮件 %d 失败: %v", i, err)
|
||
}
|
||
ids = append(ids, id)
|
||
}
|
||
|
||
// 至少要出现亚秒差异，否则说明 NOW() 又退回秒精度
|
||
var distinct int
|
||
if err := db.DB.QueryRowContext(context.Background(),
|
||
`SELECT COUNT(DISTINCT created_at) FROM mails WHERE session_id = $1`,
|
||
sid).Scan(&distinct); err != nil {
|
||
t.Fatalf("统计不同时间戳失败: %v", err)
|
||
}
|
||
if distinct < 2 {
|
||
var sample string
|
||
db.DB.QueryRowContext(context.Background(),
|
||
`SELECT CAST(created_at AS TEXT) FROM mails WHERE session_id = $1 LIMIT 1`,
|
||
sid).Scan(&sample)
|
||
t.Fatalf("8 封邮件只有 %d 个不同时间戳（样例 %q）—— NOW() 精度不足，"+
|
||
"同秒邮件的先后顺序会由随机 UUID 决定", distinct, sample)
|
||
}
|
||
|
||
// GetSessionMails 按时间升序，顺序必须与插入顺序一致
|
||
got, err := GetSessionMails(context.Background(), sid)
|
||
if err != nil {
|
||
t.Fatalf("取会话邮件失败: %v", err)
|
||
}
|
||
if len(got) != len(ids) {
|
||
t.Fatalf("应有 %d 封，实际 %d", len(ids), len(got))
|
||
}
|
||
for i, m := range got {
|
||
if m.ID != ids[i] {
|
||
t.Errorf("第 %d 封顺序错位：期望 %s，实际 %s（主题 %q）",
|
||
i, ids[i], m.ID, m.Subject)
|
||
}
|
||
}
|
||
}
|