chore: directory migration - gateway→server, web→client/electron

server/internal/repo/usersession_time_test.go

@@ -0,0 +1,195 @@
+package repo
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/agentmail/gateway/internal/db"
+	"github.com/google/uuid"
+)
+
+/**
+ * 会话令牌的时间比较。
+ *
+ * 这些用例锁的是一个真实的安全 bug：SQLite 驱动默认把 time.Time 写成
+ * Go 的 t.String()（"2026-09-10 15:10:36.12 +0800 HKT m=+607182.88"），
+ * 而 NOW() 返回 "2006-01-02 15:04:05.000000" 的 UTC 串。
+ * 两者做字符串比较时 'T'/'+' 与数字的字典序、以及 +0800 与 UTC 的偏移
+ * 双重错位，结果是：
+ *   - expires_at > NOW() 恒为真 → 令牌永不过期
+ *   - DELETE WHERE expires_at < NOW() 删 0 行 → 过期令牌永久堆积
+ * 生产库实测两条 user_sessions 都是这个状态。
+ *
+ * 修法在 db.sqliteDSN 的 _time_format=sqlite&_timezone=UTC，
+ * 所以这里必须用真实的 db.Connect 路径来验（setupTestDB 就是）。
+ */
+
+func seedUserForSession(t *testing.T, ctx context.Context, username string) uuid.UUID {
+	t.Helper()
+	id := uuid.New()
+	_, err := db.DB.ExecContext(ctx, `
+		INSERT INTO users (user_id, username, password_hash, display_name, role, status)
+		VALUES ($1, $2, 'x', $2, 'user', 'active')`, id, username)
+	if err != nil {
+		t.Fatalf("seed user %s: %v", username, err)
+	}
+	return id
+}
+
+// 存进去的时间必须能被 SQLite 的时间函数解析。
+// datetime() 返回 NULL 意味着所有 SQL 侧时间运算（过期判断、日历到点判断）全废。
+func TestTimeBindingIsParsableBySQLite(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	uid := seedUserForSession(t, ctx, "alice")
+
+	if _, _, err := CreateUserSession(ctx, uid, "probe"); err != nil {
+		t.Fatalf("create session: %v", err)
+	}
+
+	var bad int
+	err := db.DB.QueryRowContext(ctx,
+		`SELECT COUNT(*) FROM user_sessions WHERE datetime(expires_at) IS NULL`).Scan(&bad)
+	if err != nil {
+		t.Fatalf("probe: %v", err)
+	}
+	if bad != 0 {
+		var raw string
+		db.DB.QueryRowContext(ctx,
+			`SELECT CAST(expires_at AS TEXT) FROM user_sessions LIMIT 1`).Scan(&raw)
+		t.Fatalf("expires_at 无法被 datetime() 解析（存储为 %q）——"+
+			"所有 SQL 侧时间比较都会静默失效", raw)
+	}
+}
+
+// 新令牌必须被认作有效，且滑动续期后仍然有效。
+func TestFreshSessionResolves(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	uid := seedUserForSession(t, ctx, "bob")
+
+	token, expires, err := CreateUserSession(ctx, uid, "ua")
+	if err != nil {
+		t.Fatalf("create: %v", err)
+	}
+	if !expires.After(time.Now()) {
+		t.Fatalf("新令牌的过期时间应在将来，得到 %v", expires)
+	}
+
+	u, err := ResolveUserSession(ctx, token)
+	if err != nil {
+		t.Fatalf("resolve: %v", err)
+	}
+	if u.Username != "bob" {
+		t.Fatalf("解析出的用户应为 bob，得到 %q", u.Username)
+	}
+
+	// 滑动续期写回的也是 time.Time，格式同样必须正确
+	var bad int
+	db.DB.QueryRowContext(ctx,
+		`SELECT COUNT(*) FROM user_sessions WHERE datetime(expires_at) IS NULL`).Scan(&bad)
+	if bad != 0 {
+		t.Fatal("滑动续期写回的 expires_at 格式不可解析")
+	}
+}
+
+// 过期令牌必须被拒。这是 bug 的核心症状：格式错位时 expires_at > NOW()
+// 恒为真，一个 2020 年就过期的令牌照样能登录。
+func TestExpiredSessionRejected(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	uid := seedUserForSession(t, ctx, "carol")
+
+	token, _, err := CreateUserSession(ctx, uid, "ua")
+	if err != nil {
+		t.Fatalf("create: %v", err)
+	}
+
+	// 手工把过期时间推到过去（走同一条 time.Time 绑定路径）
+	past := time.Now().Add(-1 * time.Hour)
+	if _, err := db.DB.ExecContext(ctx,
+		`UPDATE user_sessions SET expires_at = $2 WHERE token = $1`, token, past); err != nil {
+		t.Fatalf("backdate: %v", err)
+	}
+
+	if _, err := ResolveUserSession(ctx, token); err == nil {
+		var raw string
+		db.DB.QueryRowContext(ctx,
+			`SELECT CAST(expires_at AS TEXT) FROM user_sessions WHERE token=$1`, token).Scan(&raw)
+		t.Fatalf("过期令牌竟然解析成功（expires_at=%q）—— 会话永不过期", raw)
+	}
+}
+
+// 顺手清理必须真的删掉过期行。删 0 行意味着 user_sessions 无限增长，
+// 且被盗令牌永远有效。
+func TestExpiredSessionsGetPurged(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	uid := seedUserForSession(t, ctx, "dave")
+
+	stale, _, err := CreateUserSession(ctx, uid, "ua")
+	if err != nil {
+		t.Fatalf("create: %v", err)
+	}
+	past := time.Now().Add(-48 * time.Hour)
+	if _, err := db.DB.ExecContext(ctx,
+		`UPDATE user_sessions SET expires_at = $2 WHERE token = $1`, stale, past); err != nil {
+		t.Fatalf("backdate: %v", err)
+	}
+
+	// 再建一个会话，CreateUserSession 内部会顺手清理过期令牌
+	if _, _, err := CreateUserSession(ctx, uid, "ua2"); err != nil {
+		t.Fatalf("second create: %v", err)
+	}
+
+	var n int
+	if err := db.DB.QueryRowContext(ctx,
+		`SELECT COUNT(*) FROM user_sessions WHERE token = $1`, stale).Scan(&n); err != nil {
+		t.Fatalf("count: %v", err)
+	}
+	if n != 0 {
+		t.Fatal("过期令牌没被清理 —— DELETE ... WHERE expires_at < NOW() 匹配不到行")
+	}
+
+	// 没过期的那条不能被误删
+	var alive int
+	db.DB.QueryRowContext(ctx, `SELECT COUNT(*) FROM user_sessions`).Scan(&alive)
+	if alive != 1 {
+		t.Fatalf("应剩 1 条有效会话，得到 %d", alive)
+	}
+}
+
+// 时间能原样读回，且亚秒精度不丢。
+//
+// 精度是有代价地保住的：_time_format=datetime 也能让 datetime() 正常工作，
+// 但它把时间截断到秒 —— 同秒插入的多封邮件排序就不确定，
+// 「会话里最早那封」（决定联系人身份）会取错行。
+func TestTimeRoundTripKeepsSubSecond(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+
+	_, err := db.DB.ExecContext(ctx,
+		`INSERT INTO agents (agent_name, secret, platform) VALUES ('probe','x','test')`)
+	if err != nil {
+		t.Fatalf("seed agent: %v", err)
+	}
+
+	want := time.Date(2026, 9, 3, 8, 45, 38, 123456000, time.UTC)
+	if _, err := db.DB.ExecContext(ctx,
+		`UPDATE agents SET last_seen = $1 WHERE agent_name = 'probe'`, want); err != nil {
+		t.Fatalf("update: %v", err)
+	}
+
+	var got time.Time
+	if err := db.DB.QueryRowContext(ctx,
+		`SELECT last_seen FROM agents WHERE agent_name = 'probe'`).Scan(&got); err != nil {
+		t.Fatalf("scan: %v", err)
+	}
+	if !got.UTC().Equal(want) {
+		t.Fatalf("时间往返不一致：写入 %v，读回 %v", want, got.UTC())
+	}
+	if got.UTC().Nanosecond() == 0 {
+		t.Fatal("亚秒精度被截断 —— 同秒插入的多行排序会不确定")
+	}
+}