chore: refresh gramsrv public release

2026-06-30 14:37:43 +08:00 · 2026-06-30 14:37:43 +08:00 · 70b6820474
commit 70b6820474
parent 75cebe8dbf
1274 changed files with 378751 additions and 59919 deletions
--- a/internal/store/memory/message_webpage_resolve_test.go
+++ b/internal/store/memory/message_webpage_resolve_test.go
@ -0,0 +1,111 @@
+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)
+	}
+}