117 lines
4.9 KiB
Go
117 lines
4.9 KiB
Go
package help
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
// TestAppConfigPremiumKeys 断言 premium 相关 key 完整下发且 hash 已递增：
|
|
// premium_purchase_blocked 必须显式为 false——客户端把 star gift「Send a Gift」入口与
|
|
// premiumCanBuy()=!premium_purchase_blocked 耦合，置 true 会同时隐藏送礼入口；
|
|
// reactions_user_max_premium 必须与服务端 enforcement 档位一致。
|
|
func TestAppConfigPremiumKeys(t *testing.T) {
|
|
cfg, notModified, err := (*Service)(nil).GetAppConfig(context.Background(), 0)
|
|
if err != nil || notModified {
|
|
t.Fatalf("GetAppConfig = notModified %v err %v", notModified, err)
|
|
}
|
|
if cfg.Hash != defaultAppConfigHash || cfg.Hash < 10 {
|
|
t.Fatalf("hash = %d, want defaultAppConfigHash(≥10)", cfg.Hash)
|
|
}
|
|
var decoded map[string]any
|
|
if err := json.Unmarshal(cfg.JSON, &decoded); err != nil {
|
|
t.Fatalf("app config json invalid: %v", err)
|
|
}
|
|
if blocked, ok := decoded["premium_purchase_blocked"].(bool); !ok || blocked {
|
|
t.Fatalf("premium_purchase_blocked = %v, want false (star gift 送礼入口耦合此 flag)", decoded["premium_purchase_blocked"])
|
|
}
|
|
// DrKLO 缺省 stargiftsBlocked=true 会隐藏 star gift 送礼网格，必须显式下发 false。
|
|
if blocked, ok := decoded["stargifts_blocked"].(bool); !ok || blocked {
|
|
t.Fatalf("stargifts_blocked = %v, want false (DrKLO GiftSheet 据此隐藏礼物网格)", decoded["stargifts_blocked"])
|
|
}
|
|
if posting, ok := decoded["rich_message_posting"].(string); !ok || posting != "enabled" {
|
|
t.Fatalf("rich_message_posting = %v, want enabled (TDesktop 富文本编辑入口默认打开)", decoded["rich_message_posting"])
|
|
}
|
|
wantNumbers := map[string]float64{
|
|
"reactions_user_max_default": 1,
|
|
"reactions_user_max_premium": 3,
|
|
"boosts_channel_level_max": 100,
|
|
"about_length_limit_default": 70,
|
|
"about_length_limit_premium": 140,
|
|
"dialogs_pinned_limit_default": 5,
|
|
"dialogs_pinned_limit_premium": 10,
|
|
"dialogs_folder_pinned_limit_default": 100,
|
|
"dialogs_folder_pinned_limit_premium": 200,
|
|
"saved_dialogs_pinned_limit_default": 5,
|
|
"saved_dialogs_pinned_limit_premium": 100,
|
|
"caption_length_limit_default": 1024,
|
|
"caption_length_limit_premium": 4096,
|
|
"channels_limit_default": 500,
|
|
"channels_limit_premium": 1000,
|
|
"dialog_filters_limit_default": 10,
|
|
"dialog_filters_limit_premium": 20,
|
|
"upload_max_fileparts_default": 4000,
|
|
"upload_max_fileparts_premium": 8000,
|
|
"aicompose_tone_examples_num": 3,
|
|
"aicompose_tone_title_length_max": 12,
|
|
"aicompose_tone_prompt_length_max": 1024,
|
|
"aicompose_tone_saved_limit_default": 5,
|
|
"aicompose_tone_saved_limit_premium": 20,
|
|
"stories_stealth_future_period": 1500,
|
|
"stories_stealth_past_period": 300,
|
|
"stories_stealth_cooldown_period": 10800,
|
|
}
|
|
for key, want := range wantNumbers {
|
|
got, ok := decoded[key].(float64)
|
|
if !ok || got != want {
|
|
t.Errorf("appConfig[%q] = %v, want %v", key, decoded[key], want)
|
|
}
|
|
}
|
|
// 未实现功能族的 key 不得下发（诱导客户端进入未实现路径）。
|
|
for _, forbidden := range []string{"chatlists_joined_limit_default", "stories_sent_weekly_limit_default", "premium_bot_username", "premium_invoice_slug"} {
|
|
if _, ok := decoded[forbidden]; ok {
|
|
t.Errorf("appConfig 不应包含 %q", forbidden)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAppConfigOmitsMapboxTokenByDefault(t *testing.T) {
|
|
cfg, notModified, err := (*Service)(nil).GetAppConfig(context.Background(), 0)
|
|
if err != nil || notModified {
|
|
t.Fatalf("GetAppConfig = notModified %v err %v", notModified, err)
|
|
}
|
|
var decoded map[string]any
|
|
if err := json.Unmarshal(cfg.JSON, &decoded); err != nil {
|
|
t.Fatalf("app config json invalid: %v", err)
|
|
}
|
|
if _, ok := decoded["tdesktop_config_map"]; ok {
|
|
t.Fatal("tdesktop_config_map present without configured Mapbox token")
|
|
}
|
|
}
|
|
|
|
func TestAppConfigUsesConfiguredMapboxTokenAndHash(t *testing.T) {
|
|
svc := NewService(nil, nil, WithMapboxToken("pk.test-token"))
|
|
cfg, notModified, err := svc.GetAppConfig(context.Background(), 0)
|
|
if err != nil || notModified {
|
|
t.Fatalf("GetAppConfig = notModified %v err %v", notModified, err)
|
|
}
|
|
if cfg.Hash == defaultAppConfigHash {
|
|
t.Fatalf("hash = %d, want token-specific hash", cfg.Hash)
|
|
}
|
|
if _, notModified, err := svc.GetAppConfig(context.Background(), cfg.Hash); err != nil || !notModified {
|
|
t.Fatalf("GetAppConfig(hash) = notModified %v err %v, want notModified", notModified, err)
|
|
}
|
|
var decoded map[string]any
|
|
if err := json.Unmarshal(cfg.JSON, &decoded); err != nil {
|
|
t.Fatalf("app config json invalid: %v", err)
|
|
}
|
|
configMap, ok := decoded["tdesktop_config_map"].(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("tdesktop_config_map = %T, want object", decoded["tdesktop_config_map"])
|
|
}
|
|
for _, key := range []string{"maps", "geo", "bmaps", "bgeo"} {
|
|
if got := configMap[key]; got != "pk.test-token" {
|
|
t.Fatalf("tdesktop_config_map[%q] = %v, want token", key, got)
|
|
}
|
|
}
|
|
}
|