176 lines
7.2 KiB
Go
176 lines
7.2 KiB
Go
package rpc
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/gotd/td/clock"
|
||
"github.com/gotd/td/tg"
|
||
"go.uber.org/zap/zaptest"
|
||
appchannels "telesrv/internal/app/channels"
|
||
appusers "telesrv/internal/app/users"
|
||
"telesrv/internal/domain"
|
||
"telesrv/internal/store/memory"
|
||
"testing"
|
||
)
|
||
|
||
func TestLegacyChannelSettingsRPC(t *testing.T) {
|
||
ctx := context.Background()
|
||
userStore := memory.NewUserStore()
|
||
owner, _ := userStore.Create(ctx, domain.User{AccessHash: 81, Phone: "15550002181", FirstName: "Owner"})
|
||
friend, _ := userStore.Create(ctx, domain.User{AccessHash: 82, Phone: "15550002182", FirstName: "Friend"})
|
||
channelStore := memory.NewChannelStore()
|
||
r := New(Config{}, Deps{
|
||
Users: appusers.NewService(userStore),
|
||
Channels: appchannels.NewService(channelStore),
|
||
}, zaptest.NewLogger(t), clock.System)
|
||
created, err := r.onMessagesCreateChat(WithUserID(ctx, owner.ID), &tg.MessagesCreateChatRequest{
|
||
Users: []tg.InputUserClass{&tg.InputUser{UserID: friend.ID, AccessHash: friend.AccessHash}},
|
||
Title: "Legacy Settings Group",
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("create chat: %v", err)
|
||
}
|
||
channel := created.Updates.(*tg.Updates).Chats[0].(*tg.Channel)
|
||
peer := &tg.InputPeerChannel{ChannelID: channel.ID, AccessHash: channel.AccessHash}
|
||
|
||
themeUpdates, err := r.onMessagesSetChatTheme(WithUserID(ctx, owner.ID), &tg.MessagesSetChatThemeRequest{Peer: peer})
|
||
if err != nil {
|
||
t.Fatalf("set chat theme channel peer: %v", err)
|
||
}
|
||
if len(themeUpdates.(*tg.Updates).Chats) != 1 {
|
||
t.Fatalf("set chat theme updates = %+v, want channel context", themeUpdates)
|
||
}
|
||
privateTheme, err := r.onMessagesSetChatTheme(WithUserID(ctx, owner.ID), &tg.MessagesSetChatThemeRequest{
|
||
Peer: &tg.InputPeerUser{UserID: friend.ID, AccessHash: friend.AccessHash},
|
||
Theme: &tg.InputChatThemeEmpty{},
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("set chat theme private peer: %v", err)
|
||
}
|
||
if len(privateTheme.(*tg.Updates).Updates) != 0 {
|
||
t.Fatalf("private set chat theme updates = %+v, want empty compat ack", privateTheme)
|
||
}
|
||
|
||
reactionUpdates, err := r.onMessagesSetChatAvailableReactions(WithUserID(ctx, owner.ID), &tg.MessagesSetChatAvailableReactionsRequest{
|
||
Peer: peer,
|
||
AvailableReactions: &tg.ChatReactionsSome{Reactions: []tg.ReactionClass{&tg.ReactionEmoji{Emoticon: "\U0001f44d"}}},
|
||
ReactionsLimit: 8,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("set available reactions: %v", err)
|
||
}
|
||
if len(reactionUpdates.(*tg.Updates).Chats) != 1 {
|
||
t.Fatalf("set reactions updates = %+v, want channel state update", reactionUpdates)
|
||
}
|
||
full, err := r.onChannelsGetFullChannel(WithUserID(ctx, owner.ID), &tg.InputChannel{ChannelID: channel.ID, AccessHash: channel.AccessHash})
|
||
if err != nil {
|
||
t.Fatalf("get full channel after reactions: %v", err)
|
||
}
|
||
fullChannel := full.FullChat.(*tg.ChannelFull)
|
||
reactions, ok := fullChannel.GetAvailableReactions()
|
||
if !ok {
|
||
t.Fatalf("full channel reactions missing after set")
|
||
}
|
||
some, ok := reactions.(*tg.ChatReactionsSome)
|
||
if !ok || len(some.Reactions) != 1 {
|
||
t.Fatalf("full channel reactions = %#v, want one explicit reaction", reactions)
|
||
}
|
||
if fullChannel.ReactionsLimit != 8 {
|
||
t.Fatalf("full channel reactions limit = %d, want 8", fullChannel.ReactionsLimit)
|
||
}
|
||
if _, err := r.onMessagesSetChatAvailableReactions(WithUserID(ctx, owner.ID), &tg.MessagesSetChatAvailableReactionsRequest{
|
||
Peer: peer,
|
||
AvailableReactions: &tg.ChatReactionsSome{Reactions: make([]tg.ReactionClass, domain.MaxChannelReactionTypes+1)},
|
||
}); err == nil {
|
||
t.Fatalf("set too many reactions err = nil, want limit error")
|
||
}
|
||
|
||
noForwards, err := r.onMessagesToggleNoForwards(WithUserID(ctx, owner.ID), &tg.MessagesToggleNoForwardsRequest{
|
||
Peer: peer,
|
||
Enabled: true,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("toggle noforwards: %v", err)
|
||
}
|
||
if got := noForwards.(*tg.Updates).Chats[0].(*tg.Channel); !got.Noforwards {
|
||
t.Fatalf("noforwards channel = %+v, want enabled", got)
|
||
}
|
||
sent, err := r.onMessagesSendMessage(WithUserID(ctx, owner.ID), &tg.MessagesSendMessageRequest{
|
||
Peer: peer,
|
||
Message: "protected content",
|
||
RandomID: 8181,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("send protected message: %v", err)
|
||
}
|
||
msg := sent.(*tg.Updates).Updates[1].(*tg.UpdateNewChannelMessage).Message.(*tg.Message)
|
||
if !msg.Noforwards {
|
||
t.Fatalf("protected channel message = %+v, want noforwards inherited", msg)
|
||
}
|
||
}
|
||
|
||
// TestBroadcastChannelAcceptsFullReactionCatalog 复现并锁定真机 bug：广播频道开启
|
||
// reactions 时，DrKLO 把「启用全部标准 reaction」发成显式 chatReactionsSome 列表
|
||
// （megagroup 走 chatReactionsAll），列表长度等于 getAvailableReactions 目录大小
|
||
// （当前 ~74）。此前 MaxChannelReactionItems=64 把它误判成 LIMIT_INVALID。修复后
|
||
// 任何不超过 MaxChannelReactionTypes 的列表都必须被接受。
|
||
func TestBroadcastChannelAcceptsFullReactionCatalog(t *testing.T) {
|
||
ctx := context.Background()
|
||
userStore := memory.NewUserStore()
|
||
owner, _ := userStore.Create(ctx, domain.User{AccessHash: 91, Phone: "15550002191", FirstName: "Owner"})
|
||
channelStore := memory.NewChannelStore()
|
||
r := New(Config{}, Deps{
|
||
Users: appusers.NewService(userStore),
|
||
Channels: appchannels.NewService(channelStore),
|
||
}, zaptest.NewLogger(t), clock.System)
|
||
|
||
created, err := r.onChannelsCreateChannel(WithUserID(ctx, owner.ID), &tg.ChannelsCreateChannelRequest{
|
||
Title: "Broadcast Reactions",
|
||
Broadcast: true,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("create broadcast channel: %v", err)
|
||
}
|
||
channel := created.(*tg.Updates).Chats[0].(*tg.Channel)
|
||
if channel.Megagroup {
|
||
t.Fatalf("created channel = %+v, want broadcast (not megagroup)", channel)
|
||
}
|
||
peer := &tg.InputPeerChannel{ChannelID: channel.ID, AccessHash: channel.AccessHash}
|
||
|
||
// 用一个明显超过旧上限(64)的目录大小，证明修复生效。每个 emoticon 取互不相同
|
||
// 的合法短串即可（验证只关心非空且 rune 数 <= MaxChannelReactionEmoticonLength）。
|
||
const catalogSize = 74
|
||
if catalogSize <= 64 || catalogSize > domain.MaxChannelReactionTypes {
|
||
t.Fatalf("test catalog size %d must exceed the old 64 cap and stay within %d",
|
||
catalogSize, domain.MaxChannelReactionTypes)
|
||
}
|
||
reactions := make([]tg.ReactionClass, 0, catalogSize)
|
||
for i := 0; i < catalogSize; i++ {
|
||
reactions = append(reactions, &tg.ReactionEmoji{Emoticon: fmt.Sprintf("r%02d", i)})
|
||
}
|
||
updates, err := r.onMessagesSetChatAvailableReactions(WithUserID(ctx, owner.ID), &tg.MessagesSetChatAvailableReactionsRequest{
|
||
Peer: peer,
|
||
AvailableReactions: &tg.ChatReactionsSome{Reactions: reactions},
|
||
ReactionsLimit: 11,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("set full-catalog reactions on broadcast channel: %v", err)
|
||
}
|
||
if len(updates.(*tg.Updates).Chats) != 1 {
|
||
t.Fatalf("set reactions updates = %+v, want channel state update", updates)
|
||
}
|
||
|
||
full, err := r.onChannelsGetFullChannel(WithUserID(ctx, owner.ID), &tg.InputChannel{ChannelID: channel.ID, AccessHash: channel.AccessHash})
|
||
if err != nil {
|
||
t.Fatalf("get full channel after reactions: %v", err)
|
||
}
|
||
fullChannel := full.FullChat.(*tg.ChannelFull)
|
||
stored, ok := fullChannel.GetAvailableReactions()
|
||
if !ok {
|
||
t.Fatalf("full channel reactions missing after set")
|
||
}
|
||
some, ok := stored.(*tg.ChatReactionsSome)
|
||
if !ok || len(some.Reactions) != catalogSize {
|
||
t.Fatalf("full channel reactions = %#v, want %d explicit reactions", stored, catalogSize)
|
||
}
|
||
}
|