262 lines
10 KiB
Go
262 lines
10 KiB
Go
package rpc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gotd/td/bin"
|
|
"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"
|
|
)
|
|
|
|
// TestMonoforumSavedDialogsAndHistory 验证频道私信(monoforum)读侧 RPC:管理员经
|
|
// getSavedDialogs(parent_peer=monoforum) 看订阅者子会话列表、经 getSavedHistory 看某订阅者历史
|
|
// (消息带 saved_peer_id);非管理员被拒。
|
|
func TestMonoforumSavedDialogsAndHistory(t *testing.T) {
|
|
ctx := context.Background()
|
|
userStore := memory.NewUserStore()
|
|
owner, err := userStore.Create(ctx, domain.User{AccessHash: 11, Phone: "15550002001", FirstName: "Owner"})
|
|
if err != nil {
|
|
t.Fatalf("create owner: %v", err)
|
|
}
|
|
sub, err := userStore.Create(ctx, domain.User{AccessHash: 12, Phone: "15550002002", FirstName: "Sub"})
|
|
if err != nil {
|
|
t.Fatalf("create sub: %v", err)
|
|
}
|
|
|
|
channelStore := memory.NewChannelStore()
|
|
channelSvc := appchannels.NewService(channelStore)
|
|
r := New(Config{}, Deps{
|
|
Users: appusers.NewService(userStore),
|
|
Channels: channelSvc,
|
|
}, zaptest.NewLogger(t), clock.System)
|
|
|
|
created, err := channelSvc.CreateChannel(ctx, owner.ID, domain.CreateChannelRequest{Title: "DM Broadcast", Broadcast: true, Date: 1000})
|
|
if err != nil {
|
|
t.Fatalf("create channel: %v", err)
|
|
}
|
|
enabled, err := channelStore.SetPaidMessagesPrice(ctx, owner.ID, created.Channel.ID, 0, true)
|
|
if err != nil {
|
|
t.Fatalf("enable DM: %v", err)
|
|
}
|
|
monoID := enabled.Channel.LinkedMonoforumID
|
|
if monoID == 0 {
|
|
t.Fatalf("no monoforum created")
|
|
}
|
|
|
|
subPeer := domain.Peer{Type: domain.PeerTypeUser, ID: sub.ID}
|
|
if _, err := channelStore.SendMonoforumMessage(ctx, domain.SendMonoforumMessageRequest{
|
|
MonoforumID: monoID, SenderUserID: sub.ID, SavedPeer: subPeer, RandomID: 1, Message: "hello channel", Date: 1100,
|
|
}); err != nil {
|
|
t.Fatalf("seed monoforum message: %v", err)
|
|
}
|
|
|
|
mono, err := channelStore.GetChannelByID(ctx, monoID)
|
|
if err != nil {
|
|
t.Fatalf("get monoforum: %v", err)
|
|
}
|
|
monoInput := &tg.InputPeerChannel{ChannelID: monoID, AccessHash: mono.AccessHash}
|
|
|
|
// TDesktop 点 Direct Messages 入口会先按 monoforum peer 拉普通 channel history。
|
|
// 主历史只应返回 monoforum 自身的 service messages,不能混入 saved_peer 子会话消息。
|
|
var raw bin.Buffer
|
|
if err := (&tg.MessagesGetHistoryRequest{Peer: monoInput, Limit: 20}).Encode(&raw); err != nil {
|
|
t.Fatalf("encode getHistory(monoforum): %v", err)
|
|
}
|
|
mainEnc, err := r.Dispatch(WithUserID(ctx, owner.ID), [8]byte{}, 0, &raw)
|
|
if err != nil {
|
|
t.Fatalf("dispatch getHistory(monoforum): %v", err)
|
|
}
|
|
mainBox, ok := mainEnc.(*tg.MessagesMessagesBox)
|
|
if !ok {
|
|
t.Fatalf("getHistory(monoforum) = %T, want MessagesMessagesBox", mainEnc)
|
|
}
|
|
mainHistory, ok := mainBox.Messages.(*tg.MessagesChannelMessages)
|
|
if !ok {
|
|
t.Fatalf("getHistory(monoforum) boxed = %T, want MessagesChannelMessages", mainBox.Messages)
|
|
}
|
|
if len(mainHistory.Messages) != 1 {
|
|
t.Fatalf("main monoforum history = %d msgs, want only the creation service", len(mainHistory.Messages))
|
|
}
|
|
service, ok := mainHistory.Messages[0].(*tg.MessageService)
|
|
if !ok {
|
|
t.Fatalf("main monoforum message = %T, want MessageService", mainHistory.Messages[0])
|
|
}
|
|
// monoforum 的首条服务消息是创建消息;TDesktop 对 monoforum 渲染为 "Direct messages were
|
|
// enabled in this channel."(lng_action_created_monoforum)。paid_messages_price 只进母广播频道。
|
|
if _, ok := service.Action.(*tg.MessageActionChannelCreate); !ok {
|
|
t.Fatalf("main monoforum action = %T %+v, want MessageActionChannelCreate", service.Action, service.Action)
|
|
}
|
|
seenChats := map[int64]bool{}
|
|
for _, chat := range mainHistory.Chats {
|
|
if ch, ok := chat.(*tg.Channel); ok {
|
|
seenChats[ch.ID] = true
|
|
}
|
|
}
|
|
if !seenChats[monoID] || !seenChats[created.Channel.ID] {
|
|
t.Fatalf("main monoforum chats = %+v, want monoforum %d and parent %d", seenChats, monoID, created.Channel.ID)
|
|
}
|
|
var deniedRaw bin.Buffer
|
|
if err := (&tg.MessagesGetHistoryRequest{Peer: monoInput, Limit: 20}).Encode(&deniedRaw); err != nil {
|
|
t.Fatalf("encode non-admin getHistory(monoforum): %v", err)
|
|
}
|
|
if _, err := r.Dispatch(WithUserID(ctx, sub.ID), [8]byte{}, 0, &deniedRaw); err == nil {
|
|
t.Fatalf("non-admin getHistory(monoforum) = nil err, want denied")
|
|
}
|
|
|
|
// 管理员看私信列表。
|
|
dreq := &tg.MessagesGetSavedDialogsRequest{}
|
|
dreq.SetParentPeer(monoInput)
|
|
dres, err := r.onMessagesGetSavedDialogs(WithUserID(ctx, owner.ID), dreq)
|
|
if err != nil {
|
|
t.Fatalf("getSavedDialogs(monoforum): %v", err)
|
|
}
|
|
sd, ok := dres.(*tg.MessagesSavedDialogs)
|
|
if !ok {
|
|
t.Fatalf("getSavedDialogs = %T, want *tg.MessagesSavedDialogs", dres)
|
|
}
|
|
if len(sd.Dialogs) != 1 {
|
|
t.Fatalf("dialogs = %d, want 1 subscriber sublist", len(sd.Dialogs))
|
|
}
|
|
md, ok := sd.Dialogs[0].(*tg.MonoForumDialog)
|
|
if !ok {
|
|
t.Fatalf("dialog = %T, want *tg.MonoForumDialog", sd.Dialogs[0])
|
|
}
|
|
if pu, ok := md.Peer.(*tg.PeerUser); !ok || pu.UserID != sub.ID {
|
|
t.Fatalf("dialog peer = %#v, want PeerUser %d", md.Peer, sub.ID)
|
|
}
|
|
foundSub := false
|
|
for _, u := range sd.Users {
|
|
if usr, ok := u.(*tg.User); ok && usr.ID == sub.ID {
|
|
foundSub = true
|
|
}
|
|
}
|
|
if !foundSub {
|
|
t.Fatalf("getSavedDialogs users missing subscriber %d", sub.ID)
|
|
}
|
|
|
|
// 管理员看某订阅者会话历史。
|
|
hreq := &tg.MessagesGetSavedHistoryRequest{Peer: &tg.InputPeerUser{UserID: sub.ID}}
|
|
hreq.SetParentPeer(monoInput)
|
|
hres, err := r.onMessagesGetSavedHistory(WithUserID(ctx, owner.ID), hreq)
|
|
if err != nil {
|
|
t.Fatalf("getSavedHistory(monoforum): %v", err)
|
|
}
|
|
var gotMsgs []tg.MessageClass
|
|
switch m := hres.(type) {
|
|
case *tg.MessagesMessages:
|
|
gotMsgs = m.Messages
|
|
case *tg.MessagesMessagesSlice:
|
|
gotMsgs = m.Messages
|
|
case *tg.MessagesChannelMessages:
|
|
gotMsgs = m.Messages
|
|
default:
|
|
t.Fatalf("getSavedHistory = %T, want messages", hres)
|
|
}
|
|
if len(gotMsgs) != 1 {
|
|
t.Fatalf("history = %d msgs, want 1", len(gotMsgs))
|
|
}
|
|
msg, ok := gotMsgs[0].(*tg.Message)
|
|
if !ok {
|
|
t.Fatalf("msg = %T, want *tg.Message", gotMsgs[0])
|
|
}
|
|
if msg.Message != "hello channel" {
|
|
t.Fatalf("body = %q, want 'hello channel'", msg.Message)
|
|
}
|
|
sp, ok := msg.GetSavedPeerID()
|
|
if !ok {
|
|
t.Fatalf("message missing saved_peer_id (client can't group into subscriber sublist)")
|
|
}
|
|
if pu, ok := sp.(*tg.PeerUser); !ok || pu.UserID != sub.ID {
|
|
t.Fatalf("saved_peer_id = %#v, want sub %d", sp, sub.ID)
|
|
}
|
|
|
|
// 非管理员(订阅者本人)经管理员入口看列表被拒。
|
|
if _, err := r.onMessagesGetSavedDialogs(WithUserID(ctx, sub.ID), dreq); err == nil {
|
|
t.Fatalf("non-admin getSavedDialogs(monoforum) = nil err, want denied")
|
|
}
|
|
}
|
|
|
|
// TestMonoforumSendMessageWritePath 验证写侧:订阅者经 sendMessage(peer=monoforum,
|
|
// reply_to=InputReplyToMonoForum{自己}) 发私信;管理员回复到目标订阅者;订阅者不能写他人子会话;
|
|
// 普通发送(无 monoforum_peer_id)不受影响。
|
|
func TestMonoforumSendMessageWritePath(t *testing.T) {
|
|
ctx := context.Background()
|
|
userStore := memory.NewUserStore()
|
|
owner, err := userStore.Create(ctx, domain.User{AccessHash: 21, Phone: "15550003001", FirstName: "Owner"})
|
|
if err != nil {
|
|
t.Fatalf("create owner: %v", err)
|
|
}
|
|
sub, err := userStore.Create(ctx, domain.User{AccessHash: 22, Phone: "15550003002", FirstName: "Sub"})
|
|
if err != nil {
|
|
t.Fatalf("create sub: %v", err)
|
|
}
|
|
|
|
channelStore := memory.NewChannelStore()
|
|
channelSvc := appchannels.NewService(channelStore)
|
|
r := New(Config{}, Deps{
|
|
Users: appusers.NewService(userStore),
|
|
Channels: channelSvc,
|
|
}, zaptest.NewLogger(t), clock.System)
|
|
|
|
created, err := channelSvc.CreateChannel(ctx, owner.ID, domain.CreateChannelRequest{Title: "DM Broadcast", Broadcast: true, Date: 1000})
|
|
if err != nil {
|
|
t.Fatalf("create channel: %v", err)
|
|
}
|
|
enabled, err := channelStore.SetPaidMessagesPrice(ctx, owner.ID, created.Channel.ID, 0, true)
|
|
if err != nil {
|
|
t.Fatalf("enable DM: %v", err)
|
|
}
|
|
monoID := enabled.Channel.LinkedMonoforumID
|
|
monoInput := &tg.InputPeerChannel{ChannelID: monoID}
|
|
|
|
// 订阅者发私信到自己的子会话。
|
|
subReq := &tg.MessagesSendMessageRequest{Peer: monoInput, Message: "hi from sub", RandomID: 555}
|
|
subReq.SetReplyTo(&tg.InputReplyToMonoForum{MonoforumPeerID: &tg.InputPeerUser{UserID: sub.ID}})
|
|
subUpd, err := r.onMessagesSendMessage(WithUserID(ctx, sub.ID), subReq)
|
|
if err != nil {
|
|
t.Fatalf("subscriber sendMessage(monoforum): %v", err)
|
|
}
|
|
if _, ok := subUpd.(*tg.Updates); !ok {
|
|
t.Fatalf("subscriber send updates = %T, want *tg.Updates", subUpd)
|
|
}
|
|
|
|
// 管理员回复到该订阅者的子会话。
|
|
adminReq := &tg.MessagesSendMessageRequest{Peer: monoInput, Message: "admin reply", RandomID: 556}
|
|
adminReq.SetReplyTo(&tg.InputReplyToMonoForum{MonoforumPeerID: &tg.InputPeerUser{UserID: sub.ID}})
|
|
if _, err := r.onMessagesSendMessage(WithUserID(ctx, owner.ID), adminReq); err != nil {
|
|
t.Fatalf("admin reply sendMessage(monoforum): %v", err)
|
|
}
|
|
|
|
// 订阅者不能写他人(owner)的子会话。
|
|
sneaky := &tg.MessagesSendMessageRequest{Peer: monoInput, Message: "sneaky", RandomID: 557}
|
|
sneaky.SetReplyTo(&tg.InputReplyToMonoForum{MonoforumPeerID: &tg.InputPeerUser{UserID: owner.ID}})
|
|
if _, err := r.onMessagesSendMessage(WithUserID(ctx, sub.ID), sneaky); err == nil {
|
|
t.Fatalf("subscriber writing another's sublist = nil err, want REPLY_TO_MONOFORUM_PEER_INVALID")
|
|
}
|
|
|
|
// 经管理员读历史:子会话含两条(订阅者发 + 管理员回复),倒序。
|
|
hreq := &tg.MessagesGetSavedHistoryRequest{Peer: &tg.InputPeerUser{UserID: sub.ID}}
|
|
hreq.SetParentPeer(monoInput)
|
|
hres, err := r.onMessagesGetSavedHistory(WithUserID(ctx, owner.ID), hreq)
|
|
if err != nil {
|
|
t.Fatalf("getSavedHistory after sends: %v", err)
|
|
}
|
|
slice, ok := hres.(*tg.MessagesMessagesSlice)
|
|
if !ok {
|
|
t.Fatalf("getSavedHistory = %T, want *tg.MessagesMessagesSlice", hres)
|
|
}
|
|
if len(slice.Messages) != 2 {
|
|
t.Fatalf("history = %d msgs, want 2 (sub + admin)", len(slice.Messages))
|
|
}
|
|
top, ok := slice.Messages[0].(*tg.Message)
|
|
if !ok || top.Message != "admin reply" {
|
|
t.Fatalf("history[0] = %#v, want newest 'admin reply'", slice.Messages[0])
|
|
}
|
|
}
|