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_drafts_difference_test.go
+++ b/internal/rpc/messages_drafts_difference_test.go
@ -0,0 +1,115 @@
+package rpc
+
+import (
+	"context"
+	"testing"
+
+	"github.com/gotd/td/clock"
+	"github.com/gotd/td/tg"
+	"go.uber.org/zap/zaptest"
+
+	appdialogs "telesrv/internal/app/dialogs"
+	appupdates "telesrv/internal/app/updates"
+	appusers "telesrv/internal/app/users"
+	"telesrv/internal/domain"
+	"telesrv/internal/store/memory"
+)
+
+// 回归：云草稿变更必须可被离线设备经 updates.getDifference 恢复（draft_message
+// durable 事件，重放时按 peer 重载当前权威态）。
+
+func newDraftDifferenceTestRouter(t *testing.T) (*Router, domain.User, domain.User) {
+	t.Helper()
+	ctx := context.Background()
+	userStore := memory.NewUserStore()
+	owner, _ := userStore.Create(ctx, domain.User{AccessHash: 21, Phone: "15550008801", FirstName: "Owner"})
+	friend, _ := userStore.Create(ctx, domain.User{AccessHash: 22, Phone: "15550008802", FirstName: "Friend"})
+	dialogStore := memory.NewDialogStore()
+	r := New(Config{DC: 2, IP: "127.0.0.1", Port: 2398}, Deps{
+		Users:    appusers.NewService(userStore),
+		Dialogs:  appdialogs.NewService(dialogStore, memory.NewChannelStore()),
+		Updates:  appupdates.NewService(memory.NewUpdateStateStore(), memory.NewUpdateEventStore()),
+		Sessions: &captureSessions{},
+	}, zaptest.NewLogger(t), clock.System)
+	return r, owner, friend
+}
+
+func draftUpdatesFromDifference(t *testing.T, diff tg.UpdatesDifferenceClass) []*tg.UpdateDraftMessage {
+	t.Helper()
+	full, ok := diff.(*tg.UpdatesDifference)
+	if !ok {
+		t.Fatalf("expected *tg.UpdatesDifference, got %T", diff)
+	}
+	out := make([]*tg.UpdateDraftMessage, 0, 1)
+	for _, u := range full.OtherUpdates {
+		if dm, ok := u.(*tg.UpdateDraftMessage); ok {
+			out = append(out, dm)
+		}
+	}
+	return out
+}
+
+func TestDraftMessageDifferenceReplay(t *testing.T) {
+	ctx := context.Background()
+	r, owner, friend := newDraftDifferenceTestRouter(t)
+	ownerCtx := WithUserID(ctx, owner.ID)
+	peer := &tg.InputPeerUser{UserID: friend.ID, AccessHash: friend.AccessHash}
+
+	// 设备 A 存草稿。
+	saveReq := &tg.MessagesSaveDraftRequest{Peer: peer, Message: "typing offline sync"}
+	if _, err := r.onMessagesSaveDraft(ownerCtx, saveReq); err != nil {
+		t.Fatalf("saveDraft: %v", err)
+	}
+
+	// 离线设备从 pts=0 拉差分：必须拿到 updateDraftMessage（内容为当前权威草稿）。
+	diff, err := r.onUpdatesGetDifference(ownerCtx, &tg.UpdatesGetDifferenceRequest{Pts: 0})
+	if err != nil {
+		t.Fatalf("getDifference: %v", err)
+	}
+	drafts := draftUpdatesFromDifference(t, diff)
+	if len(drafts) != 1 {
+		t.Fatalf("difference draft updates = %d, want 1", len(drafts))
+	}
+	dm, ok := drafts[0].Draft.(*tg.DraftMessage)
+	if !ok {
+		t.Fatalf("difference draft = %T, want *tg.DraftMessage", drafts[0].Draft)
+	}
+	if dm.Message != "typing offline sync" {
+		t.Fatalf("difference draft message = %q, want saved text", dm.Message)
+	}
+	if peerUser, ok := drafts[0].Peer.(*tg.PeerUser); !ok || peerUser.UserID != friend.ID {
+		t.Fatalf("difference draft peer = %#v, want friend", drafts[0].Peer)
+	}
+	full := diff.(*tg.UpdatesDifference)
+	if full.State.Pts < 1 {
+		t.Fatalf("difference state pts = %d, want advanced by draft event", full.State.Pts)
+	}
+	ptsAfterSave := full.State.Pts
+
+	// 清空草稿（空 saveDraft）→ 差分重放变为 draftMessageEmpty。
+	if _, err := r.onMessagesSaveDraft(ownerCtx, &tg.MessagesSaveDraftRequest{Peer: peer}); err != nil {
+		t.Fatalf("clear draft: %v", err)
+	}
+	diff, err = r.onUpdatesGetDifference(ownerCtx, &tg.UpdatesGetDifferenceRequest{Pts: ptsAfterSave})
+	if err != nil {
+		t.Fatalf("getDifference after clear: %v", err)
+	}
+	drafts = draftUpdatesFromDifference(t, diff)
+	if len(drafts) != 1 {
+		t.Fatalf("difference draft updates after clear = %d, want 1", len(drafts))
+	}
+	if _, ok := drafts[0].Draft.(*tg.DraftMessageEmpty); !ok {
+		t.Fatalf("difference draft after clear = %T, want DraftMessageEmpty", drafts[0].Draft)
+	}
+
+	// 历史事件重放也按"当前权威态"渲染：从 0 重拉，两条事件都应是 empty（草稿已删）。
+	diff, err = r.onUpdatesGetDifference(ownerCtx, &tg.UpdatesGetDifferenceRequest{Pts: 0})
+	if err != nil {
+		t.Fatalf("getDifference replay: %v", err)
+	}
+	for _, dm := range draftUpdatesFromDifference(t, diff) {
+		if _, ok := dm.Draft.(*tg.DraftMessageEmpty); !ok {
+			t.Fatalf("replayed draft = %T, want DraftMessageEmpty (authority deleted)", dm.Draft)
+		}
+	}
+}