MailUI4Agents/server/internal/repo/session_scope_test.go

package repo

import (
	"context"
	"testing"

	"github.com/agentmail/gateway/internal/db"
	"github.com/agentmail/gateway/internal/models"
	"github.com/google/uuid"
)

/*
会话维度：一个 Agent 同时在多条会话里干活时，**读一条不能动另一条**。

# 用户报的缺陷

「不同 session 的 agent 都可以看到全部邮件」—— 原先 `read_inbox` 按 **Agent** 列，
并且把列出来的都标成已读 ⇒ A 会话的 worker 会把 B 会话的未读标掉。

平时看不出来（SSE 事件在途时队列兜着），但桥重启/漏事件之后的补投判据是
`?status=unread` —— 被别人标掉的那封信**再也不会补投**。这不是"少看一封"，
是静默丢信。现场实例：另一条会话的来信被我在读自己收件箱时标成已读
（`mail_reads` 里那行的 reader=pi、时间就是那次读取的时刻）。

# 判据要两侧都验

只验"收窄后看不到别人的"是不够的：把列表直接返回空也能过。所以同时验
"不带会话时两条都看得到"（反向对照），以及"全标已读"两个变体。
*/

// seedMailInSession 在指定会话里造一封给 recipient 的未读邮件。
func seedMailInSession(t *testing.T, sessionID uuid.UUID, recipient string) uuid.UUID {
	t.Helper()
	var id uuid.UUID
	if err := db.DB.QueryRowContext(context.Background(),
		`INSERT INTO mails (session_id, from_name, to_name, subject, body)
		 VALUES ($1, 'sender', $2, 's', 'b') RETURNING mail_id`,
		sessionID, recipient).Scan(&id); err != nil {
		t.Fatal(err)
	}
	return id
}

func twoSessionsWithUnread(t *testing.T, agent string) (uuid.UUID, uuid.UUID, uuid.UUID, uuid.UUID) {
	t.Helper()
	ctx := context.Background()
	s1, err := CreateSession(ctx, nil, "human", "会话一", "")
	if err != nil {
		t.Fatal(err)
	}
	s2, err := CreateSession(ctx, nil, "human", "会话二", "")
	if err != nil {
		t.Fatal(err)
	}
	return s1, s2, seedMailInSession(t, s1, agent), seedMailInSession(t, s2, agent)
}

func TestInboxListIsScopedBySession(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()
	s1, s2, m1, m2 := twoSessionsWithUnread(t, "bot")

	// 反向对照：不带会话 = 整个 Agent 的收件箱，两条都看得到
	all, err := ListInbox(ctx, "bot", "unread", 50)
	if err != nil {
		t.Fatal(err)
	}
	if !hasMail(all, m1) || !hasMail(all, m2) {
		t.Fatalf("不带会话时应当两条都列出来（%d 条）", len(all))
	}

	// ★ 收窄到会话一：只有它那条
	scoped, err := ListInboxScoped(ctx, "bot", "unread", 50, s1)
	if err != nil {
		t.Fatal(err)
	}
	if !hasMail(scoped, m1) {
		t.Fatal("会话一的邮件应当在会话一的列表里")
	}
	if hasMail(scoped, m2) {
		t.Fatal("★ 会话二的邮件不该出现在会话一的列表里（用户报的缺陷）")
	}

	// 会话二同理（别只验一侧 —— 方向反了也能"过"）
	scoped2, err := ListInboxScoped(ctx, "bot", "unread", 50, s2)
	if err != nil {
		t.Fatal(err)
	}
	if hasMail(scoped2, m1) || !hasMail(scoped2, m2) {
		t.Fatalf("会话二的列表应只含会话二的邮件")
	}
}

func TestMarkAllReadCanBeScopedToSession(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()
	s1, s2, _, _ := twoSessionsWithUnread(t, "bot")

	// 只标会话一
	n, err := MarkAllInboxReadForSession(ctx, "bot", s1)
	if err != nil {
		t.Fatal(err)
	}
	if n != 1 {
		t.Fatalf("应当只标掉 1 封，实际 %d", n)
	}
	left, err := ListInboxScoped(ctx, "bot", "unread", 50, s2)
	if err != nil {
		t.Fatal(err)
	}
	if len(left) != 1 {
		t.Fatalf("★ 会话二的未读必须还在（实际剩 %d 条）—— 被跨会话标掉就是静默丢信", len(left))
	}

	// 反向对照：不带会话时两条都会被标掉
	if _, err := MarkAllInboxReadFor(ctx, "bot"); err != nil {
		t.Fatal(err)
	}
	if again, _ := ListInbox(ctx, "bot", "unread", 50); len(again) != 0 {
		t.Fatalf("不带会话的全标应当清空整个收件箱，实际剩 %d", len(again))
	}
}

func TestScopedCountsMatchScopedList(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()
	s1, s2, _, _ := twoSessionsWithUnread(t, "bot")

	// 会话内的未读数（会话列表徽标用）与"只列这条会话"的口径必须一致，
	// 否则界面上会出现"徽标 2、列表 1"这种自相矛盾。
	n1, err := CountUnreadInSession(ctx, "bot", s1)
	if err != nil {
		t.Fatal(err)
	}
	list1, err := ListInboxScoped(ctx, "bot", "unread", 50, s1)
	if err != nil {
		t.Fatal(err)
	}
	if n1 != len(list1) || n1 != 1 {
		t.Fatalf("会话一：计数 %d、列表 %d，期望都是 1", n1, len(list1))
	}
	_ = s2
}

func hasMail(list []models.Mail, id uuid.UUID) bool {
	for _, m := range list {
		if m.ID == id {
			return true
		}
	}
	return false
}