owpengram-server/internal/rpc/messages_stickers_personal_rpc_test.go

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
}