astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

merged from gramsrv upstream

Browse source
This commit is contained in:
onysd 2026-09-01 12:06:31 +03:00
parent 79c64ee916
commit 21a0856587
651 changed files with 54774 additions and 4590 deletions
Download patch file Download diff file Expand all files Collapse all files

46
internal/rpc/messages_send_rpc_test.go
View file

@ -17,16 +17,20 @@ func TestMessagesSendMessageReturnsUpdateAndRecordsOwnerContext(t *testing.T) {
sender := domain.User{ID: 1000000001, AccessHash: 11, FirstName: "Sender"}
recipient := domain.User{ID: 1000000002, AccessHash: 22, FirstName: "Recipient"}
messages := &captureMessages{}
dialogs := &captureDialogs{}
users := &countingMapUsersService{mapUsersService: mapUsersService{users: map[int64]domain.User{sender.ID: sender, recipient.ID: recipient}}}
metrics := &captureRPCMetrics{}
r := New(Config{}, Deps{
Messages: messages,
Users: mapUsersService{users: map[int64]domain.User{sender.ID: sender, recipient.ID: recipient}},
Dialogs: dialogs,
Users: users,
Metrics: metrics,
}, zaptest.NewLogger(t), clock.System)
req := &tg.MessagesSendMessageRequest{
Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash},
Message: "hello",
RandomID: 123456,
Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash},
Message: "hello",
RandomID: 123456,
ClearDraft: true,
Entities: []tg.MessageEntityClass{
&tg.MessageEntityBold{Offset: 0, Length: 5},
&tg.MessageEntityFormattedDate{Offset: 6, Length: 8, Date: 1773436800, ShortDate: true, ShortTime: true},
@ -82,6 +86,40 @@ func TestMessagesSendMessageReturnsUpdateAndRecordsOwnerContext(t *testing.T) {
if metrics.messageSend != 1 || metrics.messageSendErr != nil {
t.Fatalf("metrics send=%d err=%v, want one successful send", metrics.messageSend, metrics.messageSendErr)
}
if users.byIDsCalls != 1 || users.byIDCalls != 0 || users.selfCalls != 0 {
t.Fatalf("send user lookups byIDs/byID/self = %d/%d/%d, want one shared batch projection", users.byIDsCalls, users.byIDCalls, users.selfCalls)
}
}
func TestUsersForMessageUpdateUsesOneBatchLookup(t *testing.T) {
const ownerID int64 = 1000000001
const peerID int64 = 1000000002
users := &countingMapUsersService{mapUsersService: mapUsersService{users: map[int64]domain.User{
ownerID: {ID: ownerID, FirstName: "Owner"},
peerID: {ID: peerID, FirstName: "Peer"},
}}}
r := New(Config{}, Deps{Users: users}, zaptest.NewLogger(t), clock.System)
got := r.usersForMessageUpdate(context.Background(), ownerID, domain.Message{
OwnerUserID: ownerID,
From: domain.Peer{Type: domain.PeerTypeUser, ID: ownerID},
Peer: domain.Peer{Type: domain.PeerTypeUser, ID: peerID},
})
if users.byIDsCalls != 1 || users.selfCalls != 0 || users.byIDCalls != 0 {
t.Fatalf("user lookups byIDs/self/byID = %d/%d/%d, want 1/0/0", users.byIDsCalls, users.selfCalls, users.byIDCalls)
}
if len(got) != 2 {
t.Fatalf("users = %+v, want owner and peer", got)
}
owner, ok := got[0].(*tg.User)
if !ok || owner.ID != ownerID || !owner.Self {
t.Fatalf("first user = %+v, want self owner %d", got[0], ownerID)
}
peer, ok := got[1].(*tg.User)
if !ok || peer.ID != peerID || peer.Self {
t.Fatalf("second user = %+v, want non-self peer %d", got[1], peerID)
}
}
func TestMessagesSendMessageRateLimitReturnsFloodWait(t *testing.T) {