129 lines
4.2 KiB
Go
129 lines
4.2 KiB
Go
package rpc
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"testing"
|
||
|
||
"github.com/iamxvbaba/td/clock"
|
||
"github.com/iamxvbaba/td/tg"
|
||
"go.uber.org/zap/zaptest"
|
||
|
||
aiapp "telesrv/internal/app/ai"
|
||
"telesrv/internal/domain"
|
||
"telesrv/internal/store/memory"
|
||
)
|
||
|
||
// TestWebPagePreviewMedia 验证 getWebPagePreview 的 media 决策：done 卡片→messageMediaWebPage，
|
||
// 无链接/解析失败/非 done 一律 messageMediaEmpty（绝不返回 pending、绝不报错）。
|
||
func TestWebPagePreviewMedia(t *testing.T) {
|
||
ctx := context.Background()
|
||
urlEntities := []tg.MessageEntityClass{&tg.MessageEntityURL{Offset: 0, Length: 9}} // "https://x"
|
||
|
||
doneCard := domain.MessageWebPage{
|
||
State: domain.MessageWebPageStateDone,
|
||
ID: 42,
|
||
URL: "https://x",
|
||
DisplayURL: "x",
|
||
Title: "Hello",
|
||
Photo: &domain.Photo{
|
||
ID: 7, AccessHash: 8, DCID: 2,
|
||
Sizes: []domain.PhotoSize{{Kind: domain.PhotoSizeKindDefault, Type: "x", W: 320, H: 200, Size: 1024}},
|
||
},
|
||
}
|
||
|
||
files := &fakeFiles{resolveWebPageFn: func(string) (domain.MessageWebPage, error) { return doneCard, nil }}
|
||
r := &Router{deps: Deps{Files: files}}
|
||
|
||
t.Run("done-card", func(t *testing.T) {
|
||
media := r.webPagePreviewMedia(ctx, "https://x", urlEntities)
|
||
wrap, ok := media.(*tg.MessageMediaWebPage)
|
||
if !ok {
|
||
t.Fatalf("media = %T, want *tg.MessageMediaWebPage", media)
|
||
}
|
||
page, ok := wrap.Webpage.(*tg.WebPage)
|
||
if !ok {
|
||
t.Fatalf("webpage = %T, want *tg.WebPage", wrap.Webpage)
|
||
}
|
||
if v, _ := page.GetTitle(); v != "Hello" {
|
||
t.Fatalf("title = %q, want Hello", v)
|
||
}
|
||
})
|
||
|
||
t.Run("no-url", func(t *testing.T) {
|
||
if media := r.webPagePreviewMedia(ctx, "plain text", nil); !isEmptyMedia(media) {
|
||
t.Fatalf("media = %T, want empty", media)
|
||
}
|
||
})
|
||
|
||
t.Run("resolve-error-degrades", func(t *testing.T) {
|
||
files.resolveWebPageFn = func(string) (domain.MessageWebPage, error) {
|
||
return domain.MessageWebPage{}, errors.New("boom")
|
||
}
|
||
if media := r.webPagePreviewMedia(ctx, "https://x", urlEntities); !isEmptyMedia(media) {
|
||
t.Fatalf("media = %T, want empty on error", media)
|
||
}
|
||
})
|
||
|
||
t.Run("non-done-state-degrades", func(t *testing.T) {
|
||
files.resolveWebPageFn = func(string) (domain.MessageWebPage, error) {
|
||
return domain.MessageWebPage{State: domain.MessageWebPageStateEmpty, ID: 1, URL: "https://x"}, nil
|
||
}
|
||
if media := r.webPagePreviewMedia(ctx, "https://x", urlEntities); !isEmptyMedia(media) {
|
||
t.Fatalf("media = %T, want empty for empty-state webpage", media)
|
||
}
|
||
})
|
||
}
|
||
|
||
func TestAIComposeStyleWebPagePreview(t *testing.T) {
|
||
const userID int64 = 1001
|
||
ctx := WithUserID(context.Background(), userID)
|
||
aiSvc := aiapp.NewService(memory.NewAIComposeStore())
|
||
tone, err := aiSvc.CreateTone(ctx, domain.AIComposeToneInput{
|
||
UserID: userID,
|
||
EmojiID: 12345,
|
||
Title: "Sharp",
|
||
Prompt: "Make it crisp.",
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("CreateTone: %v", err)
|
||
}
|
||
r := New(Config{}, Deps{AICompose: aiSvc}, zaptest.NewLogger(t), clock.System)
|
||
link := "https://t.me/addstyle/" + tone.Slug
|
||
|
||
media := r.webPagePreviewMedia(ctx, "try "+link, nil)
|
||
wrap, ok := media.(*tg.MessageMediaWebPage)
|
||
if !ok {
|
||
t.Fatalf("media = %T, want *tg.MessageMediaWebPage", media)
|
||
}
|
||
page, ok := wrap.Webpage.(*tg.WebPage)
|
||
if !ok {
|
||
t.Fatalf("webpage = %T, want *tg.WebPage", wrap.Webpage)
|
||
}
|
||
if typ, ok := page.GetType(); !ok || typ != "telegram_aicomposetone" {
|
||
t.Fatalf("type = %q ok=%v, want telegram_aicomposetone", typ, ok)
|
||
}
|
||
if title, ok := page.GetTitle(); !ok || title != "Sharp" {
|
||
t.Fatalf("title = %q ok=%v, want Sharp", title, ok)
|
||
}
|
||
attrs, ok := page.GetAttributes()
|
||
if !ok || len(attrs) != 1 {
|
||
t.Fatalf("attributes = %#v ok=%v, want one", attrs, ok)
|
||
}
|
||
attr, ok := attrs[0].(*tg.WebPageAttributeAiComposeTone)
|
||
if !ok || attr.EmojiID != 12345 {
|
||
t.Fatalf("attribute = %#v, want ai compose tone emoji", attrs[0])
|
||
}
|
||
|
||
got := r.webPageForURL(ctx, "https://t.me/addstyle?slug="+tone.Slug, 0)
|
||
if page, ok := got.Webpage.(*tg.WebPage); !ok {
|
||
t.Fatalf("getWebPage webpage = %T, want *tg.WebPage", got.Webpage)
|
||
} else if typ, ok := page.GetType(); !ok || typ != "telegram_aicomposetone" {
|
||
t.Fatalf("getWebPage type = %q ok=%v", typ, ok)
|
||
}
|
||
}
|
||
|
||
func isEmptyMedia(m tg.MessageMediaClass) bool {
|
||
_, ok := m.(*tg.MessageMediaEmpty)
|
||
return ok
|
||
}
|