chore: refresh gramsrv public release
This commit is contained in:
A
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
264
internal/rpc/account_themes_rpc_test.go
Normal file
264
internal/rpc/account_themes_rpc_test.go
Normal file
|
|
@ -0,0 +1,264 @@
|
|||
package rpc
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/gotd/td/bin"
|
||||
"github.com/gotd/td/clock"
|
||||
"github.com/gotd/td/tg"
|
||||
"github.com/gotd/td/tgerr"
|
||||
"go.uber.org/zap/zaptest"
|
||||
|
||||
themesapp "telesrv/internal/app/themes"
|
||||
"telesrv/internal/domain"
|
||||
"telesrv/internal/store/memory"
|
||||
)
|
||||
|
||||
func newThemeRouter(t *testing.T, files *fakeFiles) *Router {
|
||||
t.Helper()
|
||||
return New(Config{}, Deps{
|
||||
Files: files,
|
||||
Themes: themesapp.NewService(memory.NewThemeStore()),
|
||||
}, zaptest.NewLogger(t), clock.System)
|
||||
}
|
||||
|
||||
func mustEncodeTheme(t *testing.T, th *tg.Theme) {
|
||||
t.Helper()
|
||||
var b bin.Buffer
|
||||
if err := th.Encode(&b); err != nil {
|
||||
t.Fatalf("theme encode failed (likely nil base_theme): %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAccountUploadThemeReturnsDocument 验证 uploadTheme 把上传文件落成可下载 Document 返回,
|
||||
// 且校验 mime 前缀。
|
||||
func TestAccountUploadThemeReturnsDocument(t *testing.T) {
|
||||
ctx := WithUserID(context.Background(), 1000001)
|
||||
r := newThemeRouter(t, &fakeFiles{})
|
||||
|
||||
req := &tg.AccountUploadThemeRequest{FileName: "theme.attheme", MimeType: "application/x-tgtheme-android"}
|
||||
req.File = &tg.InputFile{ID: 42, Parts: 1, Name: "theme.attheme"}
|
||||
got, err := r.onAccountUploadTheme(ctx, req)
|
||||
if err != nil {
|
||||
t.Fatalf("uploadTheme err = %v", err)
|
||||
}
|
||||
doc, ok := got.(*tg.Document)
|
||||
if !ok {
|
||||
t.Fatalf("uploadTheme = %T, want *tg.Document", got)
|
||||
}
|
||||
if doc.ID == 0 {
|
||||
t.Fatalf("uploaded document id = 0, want non-zero downloadable id")
|
||||
}
|
||||
|
||||
// 非主题 mime 前缀拒绝。
|
||||
bad := &tg.AccountUploadThemeRequest{FileName: "x", MimeType: "image/png"}
|
||||
bad.File = &tg.InputFile{ID: 7, Parts: 1, Name: "x"}
|
||||
if _, err := r.onAccountUploadTheme(ctx, bad); !tgerr.Is(err, "THEME_MIME_INVALID") {
|
||||
t.Fatalf("bad mime err = %v, want THEME_MIME_INVALID", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAccountCreateThemeFullFlow 验证 createTheme 自动分配 slug、设 creator=true、回填 document,
|
||||
// getTheme 按 slug 取回带 document;saveTheme/installTheme 返回 true。
|
||||
func TestAccountCreateThemeFullFlow(t *testing.T) {
|
||||
const userID = 1000002
|
||||
ctx := WithUserID(context.Background(), userID)
|
||||
files := &fakeFiles{docs: map[int64]domain.Document{
|
||||
555: {ID: 555, AccessHash: 5, DCID: 2, MimeType: "application/x-tgtheme-android", Size: 1234},
|
||||
}}
|
||||
r := newThemeRouter(t, files)
|
||||
|
||||
create := &tg.AccountCreateThemeRequest{Slug: "", Title: "My Theme"}
|
||||
create.SetDocument(&tg.InputDocument{ID: 555, AccessHash: 5})
|
||||
th, err := r.onAccountCreateTheme(ctx, create)
|
||||
if err != nil {
|
||||
t.Fatalf("createTheme err = %v", err)
|
||||
}
|
||||
if th.ID == 0 || th.AccessHash == 0 {
|
||||
t.Fatalf("created theme id/access_hash = %d/%d, want non-zero", th.ID, th.AccessHash)
|
||||
}
|
||||
if !th.GetCreator() {
|
||||
t.Fatalf("created theme creator = false, want true (so client routes re-upload to updateTheme)")
|
||||
}
|
||||
if th.Slug == "" {
|
||||
t.Fatalf("created theme slug empty, want auto-assigned slug")
|
||||
}
|
||||
if doc, ok := th.GetDocument(); !ok {
|
||||
t.Fatalf("created theme has no document, want document (deep-link addtheme needs it)")
|
||||
} else if d, ok := doc.(*tg.Document); !ok || d.ID != 555 {
|
||||
t.Fatalf("created theme document = %#v, want *tg.Document id 555", doc)
|
||||
}
|
||||
mustEncodeTheme(t, th)
|
||||
|
||||
// getTheme by slug 取回,仍带 document。
|
||||
get := &tg.AccountGetThemeRequest{Format: "android"}
|
||||
get.Theme = &tg.InputThemeSlug{Slug: th.Slug}
|
||||
got, err := r.onAccountGetTheme(ctx, get)
|
||||
if err != nil {
|
||||
t.Fatalf("getTheme by slug err = %v", err)
|
||||
}
|
||||
if got.ID != th.ID {
|
||||
t.Fatalf("getTheme id = %d, want %d", got.ID, th.ID)
|
||||
}
|
||||
if _, ok := got.GetDocument(); !ok {
|
||||
t.Fatalf("getTheme by slug missing document → client would show ThemeNotSupported")
|
||||
}
|
||||
mustEncodeTheme(t, got)
|
||||
|
||||
// saveTheme + installTheme by id+access_hash。
|
||||
save := &tg.AccountSaveThemeRequest{Theme: &tg.InputTheme{ID: th.ID, AccessHash: th.AccessHash}, Unsave: false}
|
||||
if ok, err := r.onAccountSaveTheme(ctx, save); err != nil || !ok {
|
||||
t.Fatalf("saveTheme = %v/%v, want true/nil", ok, err)
|
||||
}
|
||||
inst := &tg.AccountInstallThemeRequest{}
|
||||
inst.SetTheme(&tg.InputTheme{ID: th.ID, AccessHash: th.AccessHash})
|
||||
inst.SetDark(true)
|
||||
if ok, err := r.onAccountInstallTheme(ctx, inst); err != nil || !ok {
|
||||
t.Fatalf("installTheme = %v/%v, want true/nil", ok, err)
|
||||
}
|
||||
|
||||
// 未知主题引用 → THEME_INVALID。
|
||||
bad := &tg.AccountGetThemeRequest{Format: "android", Theme: &tg.InputThemeSlug{Slug: "does-not-exist"}}
|
||||
if _, err := r.onAccountGetTheme(ctx, bad); !tgerr.Is(err, "THEME_INVALID") {
|
||||
t.Fatalf("getTheme unknown slug err = %v, want THEME_INVALID", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAccountGetThemesIncludesUserThemes 验证 getThemes 跨设备同步:返回内置默认主题
|
||||
// (is_default=true,emoji 预览条用)+ 当前用户创建的自定义主题(is_default=false,creator=true);
|
||||
// hash 稳定→NotModified,集合变化→重取。
|
||||
func TestAccountGetThemesIncludesUserThemes(t *testing.T) {
|
||||
const userID = 1000004
|
||||
ctx := WithUserID(context.Background(), userID)
|
||||
files := &fakeFiles{docs: map[int64]domain.Document{
|
||||
321: {ID: 321, AccessHash: 3, DCID: 2, MimeType: "application/x-tgtheme-android", Size: 100},
|
||||
}}
|
||||
r := newThemeRouter(t, files)
|
||||
|
||||
// 初始:仅默认主题。
|
||||
first, err := r.onAccountGetThemes(ctx, &tg.AccountGetThemesRequest{Format: "android", Hash: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getThemes initial err = %v", err)
|
||||
}
|
||||
base, ok := first.(*tg.AccountThemes)
|
||||
if !ok {
|
||||
t.Fatalf("getThemes initial = %T, want *tg.AccountThemes", first)
|
||||
}
|
||||
defaultCount := len(base.Themes)
|
||||
if defaultCount == 0 {
|
||||
t.Fatalf("getThemes returned no default themes")
|
||||
}
|
||||
// 同 hash 回传 → NotModified。
|
||||
if again, _ := r.onAccountGetThemes(ctx, &tg.AccountGetThemesRequest{Hash: base.Hash}); func() bool {
|
||||
_, ok := again.(*tg.AccountThemesNotModified)
|
||||
return !ok
|
||||
}() {
|
||||
t.Fatalf("getThemes with matching hash should be NotModified")
|
||||
}
|
||||
|
||||
// 创建一个自定义主题。
|
||||
create := &tg.AccountCreateThemeRequest{Title: "Synced"}
|
||||
create.SetDocument(&tg.InputDocument{ID: 321, AccessHash: 3})
|
||||
created, err := r.onAccountCreateTheme(ctx, create)
|
||||
if err != nil {
|
||||
t.Fatalf("createTheme err = %v", err)
|
||||
}
|
||||
|
||||
// 现在 getThemes 必须包含该自定义主题,且 hash 变化。
|
||||
second, err := r.onAccountGetThemes(ctx, &tg.AccountGetThemesRequest{Hash: base.Hash})
|
||||
if err != nil {
|
||||
t.Fatalf("getThemes after create err = %v", err)
|
||||
}
|
||||
merged, ok := second.(*tg.AccountThemes)
|
||||
if !ok {
|
||||
t.Fatalf("getThemes after create = %T, want *tg.AccountThemes (hash changed)", second)
|
||||
}
|
||||
if len(merged.Themes) != defaultCount+1 {
|
||||
t.Fatalf("getThemes themes = %d, want %d (defaults + 1 custom)", len(merged.Themes), defaultCount+1)
|
||||
}
|
||||
var foundCustom bool
|
||||
for _, th := range merged.Themes {
|
||||
if th.ID == created.ID {
|
||||
foundCustom = true
|
||||
if th.GetDefault() {
|
||||
t.Fatalf("custom theme is_default = true, want false (must not pollute emoji strip)")
|
||||
}
|
||||
if !th.GetCreator() {
|
||||
t.Fatalf("custom theme creator = false, want true for owner")
|
||||
}
|
||||
}
|
||||
}
|
||||
if !foundCustom {
|
||||
t.Fatalf("getThemes did not include the user's created theme id %d", created.ID)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAccountGetThemesTDesktopExcludesDocumentlessDefaults 验证:对 format="tdesktop"
|
||||
// 不下发无 document 的合成 emoji 默认主题(否则 TDesktop 云主题网格点击会弹
|
||||
// lng_theme_no_desktop),而 format="android" 仍下发它们供 emoji 预览条使用。
|
||||
func TestAccountGetThemesTDesktopExcludesDocumentlessDefaults(t *testing.T) {
|
||||
const userID = 1000007
|
||||
ctx := WithUserID(context.Background(), userID)
|
||||
r := newThemeRouter(t, &fakeFiles{})
|
||||
|
||||
android, err := r.onAccountGetThemes(ctx, &tg.AccountGetThemesRequest{Format: "android", Hash: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getThemes android err = %v", err)
|
||||
}
|
||||
a, ok := android.(*tg.AccountThemes)
|
||||
if !ok || len(a.Themes) == 0 {
|
||||
t.Fatalf("getThemes android = %#v, want non-empty default themes", android)
|
||||
}
|
||||
|
||||
desktop, err := r.onAccountGetThemes(ctx, &tg.AccountGetThemesRequest{Format: "tdesktop", Hash: 0})
|
||||
if err != nil {
|
||||
t.Fatalf("getThemes tdesktop err = %v", err)
|
||||
}
|
||||
switch d := desktop.(type) {
|
||||
case *tg.AccountThemes:
|
||||
for _, th := range d.Themes {
|
||||
if _, hasDoc := th.GetDocument(); !hasDoc {
|
||||
t.Fatalf("tdesktop getThemes returned documentless theme id=%d emoticon=%q, want only document-backed cloud themes", th.ID, func() string { e, _ := th.GetEmoticon(); return e }())
|
||||
}
|
||||
}
|
||||
case *tg.AccountThemesNotModified:
|
||||
// 空集合(全新账号无自建主题)对应的稳定 hash 与 0 不同,首个 hash=0 不会命中,
|
||||
// 故这里若返回 NotModified 说明哈希实现异常。
|
||||
t.Fatalf("tdesktop getThemes with hash=0 = NotModified, want themes")
|
||||
default:
|
||||
t.Fatalf("tdesktop getThemes = %T, want *tg.AccountThemes", desktop)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAccountCreateThemeAccentSettingsEncode 验证带 settings 的 accent 主题往返且 base_theme 非空可编码。
|
||||
func TestAccountCreateThemeAccentSettingsEncode(t *testing.T) {
|
||||
ctx := WithUserID(context.Background(), 1000003)
|
||||
r := newThemeRouter(t, &fakeFiles{})
|
||||
|
||||
settings := tg.InputThemeSettings{BaseTheme: &tg.BaseThemeDay{}, AccentColor: 0x3997d3}
|
||||
settings.SetMessageColors([]int{0xd4f1ff, 0xb9e4ff})
|
||||
wp := tg.WallPaperSettings{}
|
||||
wp.SetBackgroundColor(0xd4f1ff)
|
||||
wp.SetSecondBackgroundColor(0xb9e4ff)
|
||||
settings.SetWallpaperSettings(wp)
|
||||
settings.SetWallpaper(&tg.InputWallPaperNoFile{})
|
||||
|
||||
create := &tg.AccountCreateThemeRequest{Title: "Accent"}
|
||||
create.SetSettings([]tg.InputThemeSettings{settings})
|
||||
th, err := r.onAccountCreateTheme(ctx, create)
|
||||
if err != nil {
|
||||
t.Fatalf("createTheme accent err = %v", err)
|
||||
}
|
||||
out, ok := th.GetSettings()
|
||||
if !ok || len(out) != 1 {
|
||||
t.Fatalf("created accent theme settings = %#v ok=%v, want 1", out, ok)
|
||||
}
|
||||
if _, isDay := out[0].BaseTheme.(*tg.BaseThemeDay); !isDay {
|
||||
t.Fatalf("settings[0].base_theme = %T, want BaseThemeDay", out[0].BaseTheme)
|
||||
}
|
||||
if out[0].AccentColor != 0x3997d3 {
|
||||
t.Fatalf("accent color = %#x, want 0x3997d3", out[0].AccentColor)
|
||||
}
|
||||
mustEncodeTheme(t, th)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue