117 lines
4 KiB
Go
117 lines
4 KiB
Go
package files
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"telesrv/internal/domain"
|
||
)
|
||
|
||
func defaultStatusesTestService(t *testing.T) (*Service, *fakeMediaStore) {
|
||
t.Helper()
|
||
media := newFakeMediaStore()
|
||
return NewService(media, nil, 2), media
|
||
}
|
||
|
||
func putAnimatedEmojiSet(t *testing.T, media *fakeMediaStore, packs []domain.StickerPack) {
|
||
t.Helper()
|
||
ctx := context.Background()
|
||
var ids []int64
|
||
for _, p := range packs {
|
||
ids = append(ids, p.DocumentIDs...)
|
||
}
|
||
for _, id := range ids {
|
||
if err := media.PutDocument(ctx, domain.Document{ID: id, AccessHash: id, MimeType: "application/x-tgsticker"}); err != nil {
|
||
t.Fatalf("put document %d: %v", id, err)
|
||
}
|
||
}
|
||
if err := media.PutStickerSet(ctx, domain.StickerSet{
|
||
ID: 1,
|
||
ShortName: "AnimatedEmojies",
|
||
Kind: domain.StickerSetKindSystem,
|
||
SystemKey: "animated_emoji",
|
||
Animated: true,
|
||
DocumentIDs: ids,
|
||
Packs: packs,
|
||
}); err != nil {
|
||
t.Fatalf("put animated_emoji set: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestEnsureDefaultEmojiStatusSet(t *testing.T) {
|
||
ctx := context.Background()
|
||
svc, media := defaultStatusesTestService(t)
|
||
putAnimatedEmojiSet(t, media, []domain.StickerPack{
|
||
// seed 导出常见为裸码点（无 FE0F），精选清单两种形态都必须匹配。
|
||
{Emoticon: "❤", DocumentIDs: []int64{101}},
|
||
{Emoticon: "👍", DocumentIDs: []int64{102}},
|
||
{Emoticon: "☕️", DocumentIDs: []int64{103}}, // 带 FE0F 的反向形态
|
||
{Emoticon: "🥔", DocumentIDs: []int64{999}}, // 不在精选清单，必须被排除
|
||
})
|
||
|
||
count, created, err := svc.EnsureDefaultEmojiStatusSet(ctx)
|
||
if err != nil {
|
||
t.Fatalf("ensure: %v", err)
|
||
}
|
||
if !created || count != 3 {
|
||
t.Fatalf("ensure = (count=%d, created=%v), want (3, true)", count, created)
|
||
}
|
||
set, found, err := media.GetStickerSetBySystemKey(ctx, domain.StickerSetSystemKeyEmojiDefaultStatuses)
|
||
if err != nil || !found {
|
||
t.Fatalf("synthesized set not found: found=%v err=%v", found, err)
|
||
}
|
||
if set.Kind != domain.StickerSetKindSystem || !set.Emojis || set.Count != 3 || set.Hash == 0 {
|
||
t.Fatalf("set meta = %+v, want system kind, emojis, count 3, non-zero hash", set)
|
||
}
|
||
got := map[int64]bool{}
|
||
for _, id := range set.DocumentIDs {
|
||
got[id] = true
|
||
}
|
||
if !got[101] || !got[102] || !got[103] || got[999] {
|
||
t.Fatalf("document ids = %v, want 101/102/103 without 999", set.DocumentIDs)
|
||
}
|
||
// 精选顺序：☕(=103) 在 ❤(=101) 之前、❤ 在 👍(=102) 之前（按清单序而非 pack 序）。
|
||
index := map[int64]int{}
|
||
for i, id := range set.DocumentIDs {
|
||
index[id] = i
|
||
}
|
||
if !(index[103] < index[101] && index[101] < index[102]) {
|
||
t.Fatalf("document order = %v, want curated order ☕<❤<👍", set.DocumentIDs)
|
||
}
|
||
|
||
// 幂等：第二次调用不得重建。
|
||
count2, created2, err := svc.EnsureDefaultEmojiStatusSet(ctx)
|
||
if err != nil {
|
||
t.Fatalf("ensure again: %v", err)
|
||
}
|
||
if created2 || count2 != 3 {
|
||
t.Fatalf("ensure again = (count=%d, created=%v), want (3, false)", count2, created2)
|
||
}
|
||
|
||
// ResolveStickerSet（inputStickerSetEmojiDefaultStatuses 的服务路径）能解析。
|
||
resolved, docs, found, err := svc.ResolveStickerSet(ctx, domain.StickerSetRef{
|
||
Kind: domain.StickerSetRefBySystem,
|
||
SystemKey: domain.StickerSetSystemKeyEmojiDefaultStatuses,
|
||
})
|
||
if err != nil || !found {
|
||
t.Fatalf("resolve: found=%v err=%v", found, err)
|
||
}
|
||
if resolved.ID != set.ID || len(docs) != 3 {
|
||
t.Fatalf("resolve = set %d with %d docs, want %d with 3", resolved.ID, len(docs), set.ID)
|
||
}
|
||
}
|
||
|
||
func TestEnsureDefaultEmojiStatusSetWithoutSeed(t *testing.T) {
|
||
ctx := context.Background()
|
||
svc, media := defaultStatusesTestService(t)
|
||
count, created, err := svc.EnsureDefaultEmojiStatusSet(ctx)
|
||
if err != nil {
|
||
t.Fatalf("ensure without seed: %v", err)
|
||
}
|
||
if created || count != 0 {
|
||
t.Fatalf("ensure without seed = (count=%d, created=%v), want (0, false)", count, created)
|
||
}
|
||
if _, found, _ := media.GetStickerSetBySystemKey(ctx, domain.StickerSetSystemKeyEmojiDefaultStatuses); found {
|
||
t.Fatal("set must not be created without animated_emoji seed")
|
||
}
|
||
}
|