112 lines
3.7 KiB
Go
112 lines
3.7 KiB
Go
package rpc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/iamxvbaba/td/clock"
|
|
"github.com/iamxvbaba/td/tg"
|
|
"go.uber.org/zap/zaptest"
|
|
|
|
appstories "telesrv/internal/app/stories"
|
|
appusers "telesrv/internal/app/users"
|
|
"telesrv/internal/domain"
|
|
"telesrv/internal/store/memory"
|
|
)
|
|
|
|
func TestPhotosGetUserPhotosPreservesMaxIDRefreshOffset(t *testing.T) {
|
|
ctx := context.Background()
|
|
userStore := memory.NewUserStore()
|
|
user, err := userStore.Create(ctx, domain.User{AccessHash: 21, Phone: "15550002101", FirstName: "Photo"})
|
|
if err != nil {
|
|
t.Fatalf("create user: %v", err)
|
|
}
|
|
files := &fakeFiles{
|
|
profilePhotos: []domain.Photo{{ID: 9100000000000000301, AccessHash: 9, DCID: 2}},
|
|
profilePhotosTotal: 1,
|
|
}
|
|
r := New(Config{}, Deps{
|
|
Users: appusers.NewService(userStore),
|
|
Files: files,
|
|
}, zaptest.NewLogger(t), clock.System)
|
|
|
|
got, err := r.onPhotosGetUserPhotos(WithUserID(ctx, user.ID), &tg.PhotosGetUserPhotosRequest{
|
|
UserID: &tg.InputUser{UserID: user.ID, AccessHash: user.AccessHash},
|
|
Offset: -1,
|
|
MaxID: 9100000000000000301,
|
|
Limit: 1,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("get user photos: %v", err)
|
|
}
|
|
if _, ok := got.(*tg.PhotosPhotos); !ok {
|
|
t.Fatalf("get user photos = %T, want *tg.PhotosPhotos", got)
|
|
}
|
|
if files.lastProfileOffset != -1 || files.lastProfileMaxID != 9100000000000000301 || files.lastProfileLimit != 1 {
|
|
t.Fatalf("profile photo args offset=%d max_id=%d limit=%d, want -1/exact/1", files.lastProfileOffset, files.lastProfileMaxID, files.lastProfileLimit)
|
|
}
|
|
}
|
|
|
|
func TestPhotosGetUserPhotosProjectsStoryPeerFlags(t *testing.T) {
|
|
ctx := context.Background()
|
|
userStore := memory.NewUserStore()
|
|
target, err := userStore.Create(ctx, domain.User{AccessHash: 22, Phone: "15550002102", FirstName: "StoryTarget"})
|
|
if err != nil {
|
|
t.Fatalf("create target: %v", err)
|
|
}
|
|
viewer, err := userStore.Create(ctx, domain.User{AccessHash: 23, Phone: "15550002103", FirstName: "Viewer"})
|
|
if err != nil {
|
|
t.Fatalf("create viewer: %v", err)
|
|
}
|
|
storyStore := memory.NewStoryStore()
|
|
if _, err := storyStore.UpsertStory(ctx, domain.UpsertStoryRequest{Story: domain.Story{
|
|
Owner: domain.Peer{Type: domain.PeerTypeUser, ID: target.ID},
|
|
ID: 4,
|
|
Date: 1700000000,
|
|
ExpireDate: 1700003600,
|
|
Public: true,
|
|
}}); err != nil {
|
|
t.Fatalf("upsert story: %v", err)
|
|
}
|
|
files := &fakeFiles{profilePhotosTotal: 0}
|
|
r := New(Config{}, Deps{
|
|
Users: appusers.NewService(userStore),
|
|
Stories: appstories.NewService(storyStore),
|
|
Files: files,
|
|
}, zaptest.NewLogger(t), fixedClock{now: time.Unix(1700000100, 0)})
|
|
reqCtx := WithUserID(ctx, viewer.ID)
|
|
if ok, err := r.onStoriesTogglePeerStoriesHidden(reqCtx, &tg.StoriesTogglePeerStoriesHiddenRequest{
|
|
Peer: &tg.InputPeerUser{UserID: target.ID, AccessHash: target.AccessHash},
|
|
Hidden: true,
|
|
}); err != nil || !ok {
|
|
t.Fatalf("hide target stories = %v, %v", ok, err)
|
|
}
|
|
|
|
got, err := r.onPhotosGetUserPhotos(reqCtx, &tg.PhotosGetUserPhotosRequest{
|
|
UserID: &tg.InputUser{UserID: target.ID, AccessHash: target.AccessHash},
|
|
Limit: 1,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("get user photos: %v", err)
|
|
}
|
|
photos, ok := got.(*tg.PhotosPhotos)
|
|
if !ok {
|
|
t.Fatalf("get user photos = %T, want *tg.PhotosPhotos", got)
|
|
}
|
|
if len(photos.Users) != 1 {
|
|
t.Fatalf("photos users = %d, want target user", len(photos.Users))
|
|
}
|
|
user, ok := photos.Users[0].(*tg.User)
|
|
if !ok {
|
|
t.Fatalf("photos user = %T, want *tg.User", photos.Users[0])
|
|
}
|
|
recent, ok := user.GetStoriesMaxID()
|
|
maxID, hasMaxID := recent.GetMaxID()
|
|
if !ok || !hasMaxID || maxID != 4 {
|
|
t.Fatalf("photos user stories_max_id = %+v ok=%v hasMaxID=%v, want max_id 4", recent, ok, hasMaxID)
|
|
}
|
|
if !user.StoriesHidden {
|
|
t.Fatalf("photos user stories_hidden = false, want true")
|
|
}
|
|
}
|