chore: refresh gramsrv public release

internal/rpc/messages_stickers_personal_rpc_test.go

@@ -0,0 +1,149 @@
+package rpc
+
+import (
+	"context"
+	"testing"
+
+	"github.com/gotd/td/clock"
+	"github.com/gotd/td/tg"
+	"github.com/gotd/td/tgerr"
+	"go.uber.org/zap/zaptest"
+
+	appaccount "telesrv/internal/app/account"
+	"telesrv/internal/domain"
+	"telesrv/internal/store/memory"
+)
+
+func stickerCollectionRouter(t *testing.T) *Router {
+	t.Helper()
+	files := &fakeFiles{docs: map[int64]domain.Document{
+		101: {ID: 101, AccessHash: 11, Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrSticker}}},
+		102: {ID: 102, AccessHash: 12, Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrSticker}}},
+		201: {ID: 201, AccessHash: 21, Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrAnimated}}},
+		301: {ID: 301, AccessHash: 31, Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrAudio}}},
+	}}
+	passwordStore := memory.NewPasswordStore()
+	return New(Config{}, Deps{
+		Account:  appaccount.NewService(passwordStore, appaccount.WithStickerCollections(passwordStore)),
+		Files:    files,
+		Sessions: &captureSessions{},
+	}, zaptest.NewLogger(t), clock.System)
+}
+
+func inputDoc(id, accessHash int64) *tg.InputDocument {
+	return &tg.InputDocument{ID: id, AccessHash: accessHash}
+}
+
+// TestFavedStickersRoundTrip 回归：faveSticker/getFavedStickers 此前未注册/返空。
+func TestFavedStickersRoundTrip(t *testing.T) {
+	r := stickerCollectionRouter(t)
+	ctx := WithUserID(context.Background(), 1000000001)
+
+	// 非贴纸文档拒绝。
+	if ok, err := r.onMessagesFaveSticker(ctx, &tg.MessagesFaveStickerRequest{ID: inputDoc(301, 31)}); ok || !tgerr.Is(err, "STICKER_DOCUMENT_INVALID") {
+		t.Fatalf("fave non-sticker = ok %v err %v, want STICKER_DOCUMENT_INVALID", ok, err)
+	}
+
+	// fave 101 → 102（最新在前）。
+	for _, id := range []int64{101, 102} {
+		if ok, err := r.onMessagesFaveSticker(ctx, &tg.MessagesFaveStickerRequest{ID: inputDoc(id, id%100+10)}); err != nil || !ok {
+			t.Fatalf("fave %d = ok %v err %v", id, ok, err)
+		}
+	}
+	faved := favedStickerIDs(t, r, ctx, 0)
+	if len(faved) != 2 || faved[0] != 102 || faved[1] != 101 {
+		t.Fatalf("faved = %v, want [102 101]（最新在前）", faved)
+	}
+
+	// not-modified：用返回 hash 再请求。
+	full, err := r.onMessagesGetFavedStickers(ctx, 0)
+	if err != nil {
+		t.Fatalf("get faved: %v", err)
+	}
+	hash := full.(*tg.MessagesFavedStickers).Hash
+	again, err := r.onMessagesGetFavedStickers(ctx, hash)
+	if err != nil {
+		t.Fatalf("get faved again: %v", err)
+	}
+	if _, ok := again.(*tg.MessagesFavedStickersNotModified); !ok {
+		t.Fatalf("re-get with hash = %T, want NotModified", again)
+	}
+
+	// unfave 101 → 只剩 102。
+	if ok, err := r.onMessagesFaveSticker(ctx, &tg.MessagesFaveStickerRequest{ID: inputDoc(101, 11), Unfave: true}); err != nil || !ok {
+		t.Fatalf("unfave 101 = ok %v err %v", ok, err)
+	}
+	if faved := favedStickerIDs(t, r, ctx, 0); len(faved) != 1 || faved[0] != 102 {
+		t.Fatalf("faved after unfave = %v, want [102]", faved)
+	}
+}
+
+// TestRecentStickersRoundTrip 验证 saveRecentSticker/getRecentStickers + dates + clear。
+func TestRecentStickersRoundTrip(t *testing.T) {
+	r := stickerCollectionRouter(t)
+	ctx := WithUserID(context.Background(), 1000000001)
+
+	if ok, err := r.onMessagesSaveRecentSticker(ctx, &tg.MessagesSaveRecentStickerRequest{ID: inputDoc(101, 11)}); err != nil || !ok {
+		t.Fatalf("save recent = ok %v err %v", ok, err)
+	}
+	out, err := r.onMessagesGetRecentStickers(ctx, &tg.MessagesGetRecentStickersRequest{})
+	if err != nil {
+		t.Fatalf("get recent: %v", err)
+	}
+	recent := out.(*tg.MessagesRecentStickers)
+	if len(recent.Stickers) != 1 {
+		t.Fatalf("recent stickers = %d, want 1", len(recent.Stickers))
+	}
+	if len(recent.Dates) != 1 || recent.Dates[0] == 0 {
+		t.Fatalf("recent dates = %v, want one non-zero date", recent.Dates)
+	}
+
+	// clear → 空。
+	if ok, err := r.onMessagesClearRecentStickers(ctx, &tg.MessagesClearRecentStickersRequest{}); err != nil || !ok {
+		t.Fatalf("clear recent = ok %v err %v", ok, err)
+	}
+	out, _ = r.onMessagesGetRecentStickers(ctx, &tg.MessagesGetRecentStickersRequest{})
+	if got := out.(*tg.MessagesRecentStickers); len(got.Stickers) != 0 {
+		t.Fatalf("recent after clear = %d, want 0", len(got.Stickers))
+	}
+}
+
+// TestSavedGifsRoundTrip 验证 saveGif/getSavedGifs + 非 GIF 拒绝。
+func TestSavedGifsRoundTrip(t *testing.T) {
+	r := stickerCollectionRouter(t)
+	ctx := WithUserID(context.Background(), 1000000001)
+
+	// 非 GIF（贴纸）拒绝。
+	if ok, err := r.onMessagesSaveGif(ctx, &tg.MessagesSaveGifRequest{ID: inputDoc(101, 11)}); ok || !tgerr.Is(err, "STICKER_DOCUMENT_INVALID") {
+		t.Fatalf("save non-gif = ok %v err %v, want STICKER_DOCUMENT_INVALID", ok, err)
+	}
+	if ok, err := r.onMessagesSaveGif(ctx, &tg.MessagesSaveGifRequest{ID: inputDoc(201, 21)}); err != nil || !ok {
+		t.Fatalf("save gif = ok %v err %v", ok, err)
+	}
+	out, err := r.onMessagesGetSavedGifs(ctx, 0)
+	if err != nil {
+		t.Fatalf("get saved gifs: %v", err)
+	}
+	if got := out.(*tg.MessagesSavedGifs); len(got.Gifs) != 1 {
+		t.Fatalf("saved gifs = %d, want 1", len(got.Gifs))
+	}
+}
+
+func favedStickerIDs(t *testing.T, r *Router, ctx context.Context, hash int64) []int64 {
+	t.Helper()
+	out, err := r.onMessagesGetFavedStickers(ctx, hash)
+	if err != nil {
+		t.Fatalf("get faved: %v", err)
+	}
+	faved, ok := out.(*tg.MessagesFavedStickers)
+	if !ok {
+		t.Fatalf("get faved = %T, want *tg.MessagesFavedStickers", out)
+	}
+	ids := make([]int64, 0, len(faved.Stickers))
+	for _, d := range faved.Stickers {
+		if doc, ok := d.(*tg.Document); ok {
+			ids = append(ids, doc.ID)
+		}
+	}
+	return ids
+}