58 lines
2.1 KiB
Go
58 lines
2.1 KiB
Go
package rpc
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/iamxvbaba/td/clock"
|
||
"github.com/iamxvbaba/td/tg"
|
||
"go.uber.org/zap/zaptest"
|
||
|
||
"telesrv/internal/domain"
|
||
)
|
||
|
||
func TestUsersGetUsersUsesSingleBatchLookupAndPreservesInputOrder(t *testing.T) {
|
||
const (
|
||
currentID = int64(1000000001)
|
||
peerID = int64(1000000002)
|
||
)
|
||
users := &countingMapUsersService{mapUsersService: mapUsersService{users: map[int64]domain.User{
|
||
currentID: {ID: currentID, AccessHash: 101, FirstName: "Alice"},
|
||
peerID: {ID: peerID, AccessHash: 202, FirstName: "Bob"},
|
||
}}}
|
||
r := New(Config{}, Deps{Users: users}, zaptest.NewLogger(t), clock.System)
|
||
|
||
got, err := r.onUsersGetUsers(WithUserID(context.Background(), currentID), []tg.InputUserClass{
|
||
&tg.InputUserSelf{},
|
||
&tg.InputUser{UserID: peerID, AccessHash: 202},
|
||
&tg.InputUser{UserID: peerID, AccessHash: 999},
|
||
&tg.InputUser{UserID: peerID},
|
||
&tg.InputUser{UserID: currentID, AccessHash: 101},
|
||
&tg.InputUserSelf{},
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("get users: %v", err)
|
||
}
|
||
if users.selfCalls != 1 || users.byIDsCalls != 1 || users.byIDCalls != 0 {
|
||
t.Fatalf("lookups self=%d byIDs=%d byID=%d, want one Self, one ByIDs and no ByID", users.selfCalls, users.byIDsCalls, users.byIDCalls)
|
||
}
|
||
if len(users.lastByIDs) != 2 || users.lastByIDs[0] != peerID || users.lastByIDs[1] != currentID {
|
||
t.Fatalf("ByIDs ids = %+v, want [%d %d]", users.lastByIDs, peerID, currentID)
|
||
}
|
||
if len(got) != 5 {
|
||
t.Fatalf("users = %+v, want bad access_hash skipped and duplicate inputs preserved", got)
|
||
}
|
||
// 显式自己 ID（InputUser{UserID: currentID}，index 3）也必须带 self 标志——
|
||
// 否则 self=false 的自己 user 会污染 DrKLO 账号缓存（Saved Messages 变身）。
|
||
wantIDs := []int64{currentID, peerID, peerID, currentID, currentID}
|
||
wantSelf := []bool{true, false, false, true, true}
|
||
for i, item := range got {
|
||
user, ok := item.(*tg.User)
|
||
if !ok {
|
||
t.Fatalf("user[%d] = %T, want *tg.User", i, item)
|
||
}
|
||
if user.ID != wantIDs[i] || user.Self != wantSelf[i] {
|
||
t.Fatalf("user[%d] = id %d self %v, want id %d self %v", i, user.ID, user.Self, wantIDs[i], wantSelf[i])
|
||
}
|
||
}
|
||
}
|