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/rpc/messages_edit_rpc_test.go
+++ b/internal/rpc/messages_edit_rpc_test.go
@ -0,0 +1,156 @@
+package rpc
+
+import (
+	"context"
+	"github.com/gotd/td/bin"
+	"github.com/gotd/td/clock"
+	"github.com/gotd/td/tg"
+	"go.uber.org/zap/zaptest"
+	"strings"
+	"telesrv/internal/domain"
+	"testing"
+)
+
+func TestUpdatesDifferenceIncludesEditMessage(t *testing.T) {
+	msg := domain.Message{
+		ID:          4,
+		OwnerUserID: 1000000001,
+		Peer:        domain.Peer{Type: domain.PeerTypeUser, ID: 1000000002},
+		From:        domain.Peer{Type: domain.PeerTypeUser, ID: 1000000001},
+		Date:        1700000200,
+		EditDate:    1700000300,
+		Out:         true,
+		Body:        "edited",
+		Pts:         8,
+	}
+	got, ok := tgUpdatesDifference(0, domain.UpdateDifference{
+		State: domain.UpdateState{Pts: 8, Date: 1700000300, Seq: 0},
+		Events: []domain.UpdateEvent{{
+			Type:     domain.UpdateEventEditMessage,
+			Pts:      8,
+			PtsCount: 1,
+			Date:     1700000300,
+			Message:  msg,
+		}},
+	}).(*tg.UpdatesDifference)
+	if !ok {
+		t.Fatalf("difference = %T, want *tg.UpdatesDifference", got)
+	}
+	if len(got.OtherUpdates) != 1 {
+		t.Fatalf("other updates = %+v, want one edit update", got.OtherUpdates)
+	}
+	update, ok := got.OtherUpdates[0].(*tg.UpdateEditMessage)
+	if !ok || update.Pts != 8 || update.PtsCount != 1 {
+		t.Fatalf("edit update = %T %+v, want pts=8", got.OtherUpdates[0], got.OtherUpdates[0])
+	}
+	edited, ok := update.Message.(*tg.Message)
+	if !ok || edited.Message != "edited" || edited.EditDate != 1700000300 {
+		t.Fatalf("edited message = %#v, want text and edit_date", update.Message)
+	}
+}
+
+func TestMessagesEditMessageReturnsUpdateAndRecordsOwnerContext(t *testing.T) {
+	const (
+		userID = int64(1000000001)
+		peerID = int64(1000000002)
+	)
+	var authKeyID [8]byte
+	authKeyID[0] = 9
+	messages := &captureMessages{}
+	r := New(Config{}, Deps{Messages: messages}, zaptest.NewLogger(t), clock.System)
+	req := &tg.MessagesEditMessageRequest{
+		Peer: &tg.InputPeerUser{UserID: peerID, AccessHash: 22},
+		ID:   3,
+	}
+	req.SetMessage("edited")
+	req.SetEntities([]tg.MessageEntityClass{&tg.MessageEntityBold{Offset: 0, Length: 6}})
+	var in bin.Buffer
+	if err := req.Encode(&in); err != nil {
+		t.Fatalf("encode request: %v", err)
+	}
+
+	enc, err := r.Dispatch(WithUserID(context.Background(), userID), authKeyID, 77, &in)
+	if err != nil {
+		t.Fatalf("dispatch: %v", err)
+	}
+	box, ok := enc.(*tg.UpdatesBox)
+	if !ok {
+		t.Fatalf("response = %T, want *tg.UpdatesBox", enc)
+	}
+	got, ok := box.Updates.(*tg.Updates)
+	if !ok || len(got.Updates) != 1 {
+		t.Fatalf("boxed updates = %T %+v, want one update", box.Updates, box.Updates)
+	}
+	edit, ok := got.Updates[0].(*tg.UpdateEditMessage)
+	if !ok || edit.Pts != 7 || edit.PtsCount != 1 {
+		t.Fatalf("edit update = %#v, want pts=7 count=1", got.Updates[0])
+	}
+	msg, ok := edit.Message.(*tg.Message)
+	if !ok || msg.ID != 3 || msg.Message != "edited" {
+		t.Fatalf("edited message = %#v, want id=3 text edited", edit.Message)
+	}
+	if messages.editReq.OwnerUserID != userID || messages.editReq.Peer.ID != peerID || messages.editReq.ID != 3 || messages.editReq.OriginAuthKeyID != authKeyID || messages.editReq.OriginSessionID != 77 {
+		t.Fatalf("edit request = %+v, want owner peer message id and origin", messages.editReq)
+	}
+	if len(messages.editReq.Entities) != 1 || messages.editReq.Entities[0].Type != domain.MessageEntityBold {
+		t.Fatalf("edit entities = %+v, want bold", messages.editReq.Entities)
+	}
+}
+
+func TestMessagesEditMessageOptionBoundaries(t *testing.T) {
+	const (
+		userID = int64(1000000001)
+		peerID = int64(1000000002)
+	)
+	ctx := WithUserID(context.Background(), userID)
+	r := New(Config{}, Deps{Messages: &captureMessages{}}, zaptest.NewLogger(t), clock.System)
+
+	webPreviewReq := &tg.MessagesEditMessageRequest{
+		Peer: &tg.InputPeerUser{UserID: peerID, AccessHash: 22},
+		ID:   3,
+	}
+	webPreviewReq.SetMessage("edited with link")
+	webPreviewReq.SetMedia(&tg.InputMediaWebPage{URL: "https://example.test/"})
+	if _, err := r.onMessagesEditMessage(ctx, webPreviewReq); err != nil {
+		t.Fatalf("edit with webpage media err = %v, want text-only downgrade", err)
+	}
+
+	cases := []struct {
+		name string
+		req  *tg.MessagesEditMessageRequest
+		want string
+	}{
+		{
+			name: "quick reply shortcut",
+			req: func() *tg.MessagesEditMessageRequest {
+				req := &tg.MessagesEditMessageRequest{Peer: &tg.InputPeerUser{UserID: peerID, AccessHash: 22}, ID: 3}
+				req.SetMessage("edited")
+				req.SetQuickReplyShortcutID(11)
+				return req
+			}(),
+			want: "MESSAGE_ID_INVALID",
+		},
+		{
+			name: "unsupported media",
+			req: func() *tg.MessagesEditMessageRequest {
+				req := &tg.MessagesEditMessageRequest{Peer: &tg.InputPeerUser{UserID: peerID, AccessHash: 22}, ID: 3}
+				req.SetMessage("edited")
+				req.SetMedia(&tg.InputMediaUploadedPhoto{})
+				return req
+			}(),
+			want: "MEDIA_INVALID",
+		},
+		{
+			name: "message flag missing",
+			req:  &tg.MessagesEditMessageRequest{Peer: &tg.InputPeerUser{UserID: peerID, AccessHash: 22}, ID: 3},
+			want: "MESSAGE_EMPTY",
+		},
+	}
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			if _, err := r.onMessagesEditMessage(ctx, tc.req); err == nil || !strings.Contains(err.Error(), tc.want) {
+				t.Fatalf("edit err = %v, want %s", err, tc.want)
+			}
+		})
+	}
+}