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

20
internal/rpc/rpc_testkit_users_test.go
View file

@ -15,10 +15,11 @@ type mapUsersService struct {
type countingMapUsersService struct {
mapUsersService
selfCalls int
byIDCalls int
byIDsCalls int
lastByIDs []int64
selfCalls int
byIDCalls int
byIDsCalls int
lastByIDs []int64
byIDsBatches [][]int64
}
func (s staticUsersService) Self(context.Context, int64) (domain.User, error) {
@ -35,6 +36,11 @@ func (s staticUsersService) ByID(_ context.Context, _, userID int64) (domain.Use
return domain.User{}, false, nil
}
func (s staticUsersService) BotStatus(_ context.Context, userID int64) (bool, bool, error) {
u, found, err := s.ByID(context.Background(), userID, userID)
return u.Bot, found, err
}
func (s staticUsersService) ByIDs(_ context.Context, _ int64, userIDs []int64) ([]domain.User, error) {
out := make([]domain.User, 0, len(userIDs))
seen := map[int64]struct{}{}
@ -61,6 +67,11 @@ func (s mapUsersService) ByID(_ context.Context, _, userID int64) (domain.User,
return u, ok, nil
}
func (s mapUsersService) BotStatus(_ context.Context, userID int64) (bool, bool, error) {
u, found := s.users[userID]
return u.Bot, found, nil
}
func (s mapUsersService) ByIDs(_ context.Context, _ int64, userIDs []int64) ([]domain.User, error) {
out := make([]domain.User, 0, len(userIDs))
seen := map[int64]struct{}{}
@ -89,6 +100,7 @@ func (s *countingMapUsersService) Self(ctx context.Context, userID int64) (domai
func (s *countingMapUsersService) ByIDs(ctx context.Context, currentUserID int64, userIDs []int64) ([]domain.User, error) {
s.byIDsCalls++
s.lastByIDs = append([]int64(nil), userIDs...)
s.byIDsBatches = append(s.byIDsBatches, append([]int64(nil), userIDs...))
return s.mapUsersService.ByIDs(ctx, currentUserID, userIDs)
}