111 lines
4.3 KiB
Go
111 lines
4.3 KiB
Go
package memory
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"testing"
|
||
|
||
"telesrv/internal/domain"
|
||
)
|
||
|
||
// TestMessageStoreResolveWebPageSwapsMediaOnly 验证 WebPageResolve 模式只换 media、不碰
|
||
// body/entities/edit_date（不标记「已编辑」），双盒一致，事件为 web_page，且幂等。
|
||
func TestMessageStoreResolveWebPageSwapsMediaOnly(t *testing.T) {
|
||
ctx := context.Background()
|
||
dialogs := NewDialogStore()
|
||
messages := NewMessageStore(dialogs)
|
||
senderID := int64(1000000201)
|
||
recipientID := int64(1000000202)
|
||
const url = "https://example.com/x"
|
||
urlHash := domain.WebPageURLHash(url)
|
||
peer := domain.Peer{Type: domain.PeerTypeUser, ID: recipientID}
|
||
|
||
pending := &domain.MessageMedia{
|
||
Kind: domain.MessageMediaKindWebPage,
|
||
WebPage: &domain.MessageWebPage{State: domain.MessageWebPageStatePending, ID: urlHash, URL: url},
|
||
}
|
||
sent, err := messages.SendPrivateText(ctx, domain.SendPrivateTextRequest{
|
||
SenderUserID: senderID, RecipientUserID: recipientID, RandomID: 201,
|
||
Message: "see " + url, Date: 1700000100, Media: pending,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("SendPrivateText: %v", err)
|
||
}
|
||
|
||
done := &domain.MessageMedia{
|
||
Kind: domain.MessageMediaKindWebPage,
|
||
WebPage: &domain.MessageWebPage{State: domain.MessageWebPageStateDone, ID: urlHash, URL: url, Title: "Example"},
|
||
}
|
||
resolveReq := domain.EditMessageRequest{
|
||
OwnerUserID: senderID,
|
||
Peer: peer,
|
||
ID: sent.SenderMessage.ID,
|
||
Media: done,
|
||
WebPageResolve: true,
|
||
ExpectedWebPageID: urlHash,
|
||
}
|
||
res, err := messages.EditMessage(ctx, resolveReq)
|
||
if err != nil {
|
||
t.Fatalf("EditMessage(WebPageResolve): %v", err)
|
||
}
|
||
if len(res.Edited) != 2 {
|
||
t.Fatalf("edited boxes = %d, want 2", len(res.Edited))
|
||
}
|
||
self := res.Self()
|
||
if self.Event.Type != domain.UpdateEventWebPage {
|
||
t.Fatalf("event type = %q, want web_page", self.Event.Type)
|
||
}
|
||
if self.Message.Media == nil || self.Message.Media.WebPage == nil || self.Message.Media.WebPage.State != domain.MessageWebPageStateDone {
|
||
t.Fatalf("self media not resolved: %+v", self.Message.Media)
|
||
}
|
||
if self.Message.Body != "see "+url {
|
||
t.Fatalf("body changed to %q, want unchanged", self.Message.Body)
|
||
}
|
||
if self.Message.EditDate != 0 {
|
||
t.Fatalf("edit_date = %d, want 0 (no 'edited' marker)", self.Message.EditDate)
|
||
}
|
||
|
||
// 接收方盒子也被替换为 done。
|
||
rh, err := messages.ListByUser(ctx, recipientID, domain.MessageFilter{HasPeer: true, Peer: domain.Peer{Type: domain.PeerTypeUser, ID: senderID}, Limit: 10})
|
||
if err != nil {
|
||
t.Fatalf("recipient history: %v", err)
|
||
}
|
||
if len(rh.Messages) != 1 || rh.Messages[0].Media == nil || rh.Messages[0].Media.WebPage == nil || rh.Messages[0].Media.WebPage.State != domain.MessageWebPageStateDone {
|
||
t.Fatalf("recipient media not resolved: %+v", rh.Messages)
|
||
}
|
||
|
||
// 幂等：已解析后再次解析 → ErrMessageNotModified。
|
||
if _, err := messages.EditMessage(ctx, resolveReq); !errors.Is(err, domain.ErrMessageNotModified) {
|
||
t.Fatalf("re-resolve err = %v, want ErrMessageNotModified", err)
|
||
}
|
||
}
|
||
|
||
// TestMessageStoreResolveWebPageWrongIDNoop 验证 expectedID 不匹配时不替换。
|
||
func TestMessageStoreResolveWebPageWrongIDNoop(t *testing.T) {
|
||
ctx := context.Background()
|
||
dialogs := NewDialogStore()
|
||
messages := NewMessageStore(dialogs)
|
||
senderID := int64(1000000211)
|
||
recipientID := int64(1000000212)
|
||
const url = "https://example.com/y"
|
||
urlHash := domain.WebPageURLHash(url)
|
||
|
||
sent, err := messages.SendPrivateText(ctx, domain.SendPrivateTextRequest{
|
||
SenderUserID: senderID, RecipientUserID: recipientID, RandomID: 211, Message: url, Date: 1700000100,
|
||
Media: &domain.MessageMedia{Kind: domain.MessageMediaKindWebPage, WebPage: &domain.MessageWebPage{State: domain.MessageWebPageStatePending, ID: urlHash, URL: url}},
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("SendPrivateText: %v", err)
|
||
}
|
||
_, err = messages.EditMessage(ctx, domain.EditMessageRequest{
|
||
OwnerUserID: senderID,
|
||
Peer: domain.Peer{Type: domain.PeerTypeUser, ID: recipientID},
|
||
ID: sent.SenderMessage.ID,
|
||
Media: &domain.MessageMedia{Kind: domain.MessageMediaKindWebPage, WebPage: &domain.MessageWebPage{State: domain.MessageWebPageStateDone, ID: 999, URL: url}},
|
||
WebPageResolve: true,
|
||
ExpectedWebPageID: 999, // 与占位 id 不符
|
||
})
|
||
if !errors.Is(err, domain.ErrMessageNotModified) {
|
||
t.Fatalf("wrong-id resolve err = %v, want ErrMessageNotModified", err)
|
||
}
|
||
}
|