167 lines
6.7 KiB
Go
167 lines
6.7 KiB
Go
package postgres
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
"time"
|
||
|
||
"telesrv/internal/domain"
|
||
)
|
||
|
||
// TestDialogTopMessageCarriesFullMetadata 回归 getDialogs/getPeerDialogs 的 top-message
|
||
// 投影完整性：dialog 列表的 top message 是独立查询（dialog.sql，不复用 message.sql），
|
||
// 必须带 grouped_id / effect / reply_markup / rich_message —— TDesktop 把 getDialogs
|
||
// 返回的消息入缓存且不被后续 getHistory 覆盖，缺任一字段都让最新那条消息永久按缺失态渲染。
|
||
func TestDialogTopMessageCarriesFullMetadata(t *testing.T) {
|
||
pool := testPool(t)
|
||
ctx := context.Background()
|
||
suffix := randomSuffix(t)
|
||
|
||
users := NewUserStore(pool)
|
||
sender, err := users.Create(ctx, domain.User{AccessHash: 71, Phone: "+1997" + suffix + "01", FirstName: "MetaSender"})
|
||
if err != nil {
|
||
t.Fatalf("create sender: %v", err)
|
||
}
|
||
recipient, err := users.Create(ctx, domain.User{AccessHash: 72, Phone: "+1997" + suffix + "02", FirstName: "MetaRecipient"})
|
||
if err != nil {
|
||
t.Fatalf("create recipient: %v", err)
|
||
}
|
||
ids := []int64{sender.ID, recipient.ID}
|
||
t.Cleanup(func() {
|
||
_, _ = pool.Exec(ctx, "DELETE FROM dispatch_outbox WHERE target_user_id = ANY($1::bigint[])", ids)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM user_update_events WHERE user_id = ANY($1::bigint[])", ids)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM message_boxes WHERE owner_user_id = ANY($1::bigint[])", ids)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM private_messages WHERE sender_user_id = ANY($1::bigint[])", ids)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM dialogs WHERE user_id = ANY($1::bigint[])", ids)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM users WHERE id = ANY($1::bigint[])", ids)
|
||
})
|
||
|
||
messages := NewMessageStore(pool, WithMessageAllocators(&perUserCounterAllocator{}))
|
||
const groupedID = int64(990000000777)
|
||
const effect = int64(5104841169784993603)
|
||
markup := &domain.MessageReplyMarkup{Inline: [][]domain.MarkupButton{{
|
||
{Type: domain.MarkupButtonCallback, Text: "Press", Data: []byte{0x00, 0x01, 0xFF, 'x'}},
|
||
}}}
|
||
rich := &domain.MessageRichMessage{Rtl: true, Blocks: []byte{0x10, 0x20, 0x30}}
|
||
|
||
res, err := messages.SendPrivateText(ctx, domain.SendPrivateTextRequest{
|
||
SenderUserID: sender.ID,
|
||
RecipientUserID: recipient.ID,
|
||
RandomID: time.Now().UnixNano(),
|
||
Message: "full metadata top",
|
||
GroupedID: groupedID,
|
||
Effect: effect,
|
||
ReplyMarkup: markup,
|
||
RichMessage: rich,
|
||
Date: int(time.Now().Unix()),
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("send message: %v", err)
|
||
}
|
||
|
||
dialogs := NewDialogStore(pool)
|
||
check := func(label string, ownerID int64, boxID int) {
|
||
t.Helper()
|
||
list, err := dialogs.ListByUser(ctx, ownerID, domain.DialogFilter{Limit: 20})
|
||
if err != nil {
|
||
t.Fatalf("%s getDialogs: %v", label, err)
|
||
}
|
||
var top *domain.Message
|
||
for i := range list.Messages {
|
||
if list.Messages[i].ID == boxID {
|
||
top = &list.Messages[i]
|
||
}
|
||
}
|
||
if top == nil {
|
||
t.Fatalf("%s getDialogs missing top message box %d", label, boxID)
|
||
}
|
||
if top.GroupedID != groupedID {
|
||
t.Fatalf("%s dialog top grouped_id = %d, want %d", label, top.GroupedID, groupedID)
|
||
}
|
||
if top.Effect != effect {
|
||
t.Fatalf("%s dialog top effect = %d, want %d", label, top.Effect, effect)
|
||
}
|
||
if top.ReplyMarkup == nil || len(top.ReplyMarkup.Inline) != 1 || len(top.ReplyMarkup.Inline[0]) != 1 ||
|
||
top.ReplyMarkup.Inline[0][0].Text != "Press" {
|
||
t.Fatalf("%s dialog top reply_markup not preserved: %+v", label, top.ReplyMarkup)
|
||
}
|
||
if top.RichMessage == nil || top.RichMessage.IsZero() || !top.RichMessage.Rtl {
|
||
t.Fatalf("%s dialog top rich_message not preserved: %+v", label, top.RichMessage)
|
||
}
|
||
}
|
||
check("sender", sender.ID, res.SenderMessage.ID)
|
||
check("recipient", recipient.ID, res.RecipientMessage.ID)
|
||
}
|
||
|
||
// TestSavedDialogTopMessageCarriesFullMetadata 回归 getSavedDialogs 的 top-message 投影：
|
||
// 收藏夹子会话(Saved Messages)的 top message 是另一组独立查询(saved_dialog.sql,3 条),
|
||
// 同样不复用 message.sql，必须带 grouped_id/effect/reply_markup/rich_message(转发相册/
|
||
// bot 键盘/富文本进收藏夹后子会话最新一条会缺失)。
|
||
func TestSavedDialogTopMessageCarriesFullMetadata(t *testing.T) {
|
||
pool := testPool(t)
|
||
ctx := context.Background()
|
||
suffix := randomSuffix(t)
|
||
|
||
users := NewUserStore(pool)
|
||
alice, err := users.Create(ctx, domain.User{AccessHash: 73, Phone: "+1998" + suffix + "01", FirstName: "SavedMeta"})
|
||
if err != nil {
|
||
t.Fatalf("create alice: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
_, _ = pool.Exec(ctx, "DELETE FROM message_boxes WHERE owner_user_id = $1", alice.ID)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM private_messages WHERE sender_user_id = $1", alice.ID)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM dialogs WHERE user_id = $1", alice.ID)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM saved_dialog_pins WHERE user_id = $1", alice.ID)
|
||
_, _ = pool.Exec(ctx, "DELETE FROM users WHERE id = $1", alice.ID)
|
||
})
|
||
|
||
messages := NewMessageStore(pool, WithMessageAllocators(&perUserCounterAllocator{}))
|
||
const groupedID = int64(990000000888)
|
||
const effect = int64(5104841169784993603)
|
||
markup := &domain.MessageReplyMarkup{Inline: [][]domain.MarkupButton{{
|
||
{Type: domain.MarkupButtonCallback, Text: "Saved", Data: []byte{0x09}},
|
||
}}}
|
||
rich := &domain.MessageRichMessage{Part: true, Blocks: []byte{0x44, 0x55}}
|
||
|
||
// 直发笔记 → saved_peer = self；子会话 top 即这条。
|
||
note, err := messages.SendPrivateText(ctx, domain.SendPrivateTextRequest{
|
||
SenderUserID: alice.ID,
|
||
RecipientUserID: alice.ID,
|
||
RandomID: time.Now().UnixNano(),
|
||
Message: "self note full meta",
|
||
GroupedID: groupedID,
|
||
Effect: effect,
|
||
ReplyMarkup: markup,
|
||
RichMessage: rich,
|
||
Date: int(time.Now().Unix()),
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("send self note: %v", err)
|
||
}
|
||
|
||
list, err := messages.ListSavedDialogs(ctx, alice.ID, domain.SavedDialogsFilter{Limit: 20})
|
||
if err != nil {
|
||
t.Fatalf("ListSavedDialogs: %v", err)
|
||
}
|
||
var top *domain.Message
|
||
for i := range list.Messages {
|
||
if list.Messages[i].ID == note.SenderMessage.ID {
|
||
top = &list.Messages[i]
|
||
}
|
||
}
|
||
if top == nil {
|
||
t.Fatalf("getSavedDialogs missing top message box %d (got %d messages)", note.SenderMessage.ID, len(list.Messages))
|
||
}
|
||
if top.GroupedID != groupedID {
|
||
t.Fatalf("saved dialog top grouped_id = %d, want %d", top.GroupedID, groupedID)
|
||
}
|
||
if top.Effect != effect {
|
||
t.Fatalf("saved dialog top effect = %d, want %d", top.Effect, effect)
|
||
}
|
||
if top.ReplyMarkup == nil || len(top.ReplyMarkup.Inline) != 1 || top.ReplyMarkup.Inline[0][0].Text != "Saved" {
|
||
t.Fatalf("saved dialog top reply_markup not preserved: %+v", top.ReplyMarkup)
|
||
}
|
||
if top.RichMessage == nil || top.RichMessage.IsZero() || !top.RichMessage.Part {
|
||
t.Fatalf("saved dialog top rich_message not preserved: %+v", top.RichMessage)
|
||
}
|
||
}
|