167 lines
6.9 KiB
Go
167 lines
6.9 KiB
Go
package rpc
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/iamxvbaba/td/clock"
|
|
"github.com/iamxvbaba/td/tg"
|
|
"go.uber.org/zap/zaptest"
|
|
|
|
appstories "telesrv/internal/app/stories"
|
|
"telesrv/internal/compat/tdesktop"
|
|
"telesrv/internal/domain"
|
|
"telesrv/internal/store/memory"
|
|
)
|
|
|
|
func TestDeletedUserTLProjectionContainsOnlyTombstoneIdentity(t *testing.T) {
|
|
u := domain.User{
|
|
ID: 42, AccessHash: 99, Phone: "secret", FirstName: "Alice", LastName: "Private",
|
|
Username: "released", About: "hidden", Verified: true, PremiumUntil: 2_000_000_000,
|
|
PhotoID: 123, Deleted: true, DeletedAt: 1_800_000_000,
|
|
}
|
|
got := tgUser(u)
|
|
if got.ID != u.ID || !got.Deleted {
|
|
t.Fatalf("deleted user = %+v", got)
|
|
}
|
|
if got.AccessHash != 0 || got.Phone != "" || got.FirstName != "" || got.LastName != "" || got.Username != "" || got.Verified || got.Premium || got.Photo != nil || got.Status != nil || len(got.Usernames) != 0 {
|
|
t.Fatalf("deleted user leaked profile state: %+v", got)
|
|
}
|
|
self := tgSelfUser(u)
|
|
if !self.Deleted || self.Self || self.ID != u.ID {
|
|
t.Fatalf("deleted self projection = %+v", self)
|
|
}
|
|
}
|
|
|
|
func TestHistoryHydrationReplacesStaleUserWithDeletedTombstone(t *testing.T) {
|
|
viewer := domain.User{ID: 7, FirstName: "Viewer"}
|
|
deleted := domain.User{ID: 42, AccessHash: 99, Deleted: true, DeletedAt: 1_800_000_000}
|
|
r := New(Config{}, Deps{Users: mapUsersService{users: map[int64]domain.User{
|
|
viewer.ID: viewer, deleted.ID: deleted,
|
|
}}}, zaptest.NewLogger(t), clock.System)
|
|
|
|
list := r.enrichMessageList(context.Background(), viewer.ID, domain.MessageList{
|
|
Messages: []domain.Message{{
|
|
OwnerUserID: viewer.ID,
|
|
Peer: domain.Peer{Type: domain.PeerTypeUser, ID: deleted.ID},
|
|
From: domain.Peer{Type: domain.PeerTypeUser, ID: deleted.ID},
|
|
Body: "retained history",
|
|
}},
|
|
// Simulate an old denormalized message query row. The authoritative
|
|
// Users.ByIDs hydration must replace it, not keep an empty active user.
|
|
Users: []domain.User{{ID: deleted.ID, Phone: "stale", FirstName: "Stale"}},
|
|
})
|
|
if len(list.Users) != 1 || !list.Users[0].Deleted || list.Users[0].Phone != "" || list.Users[0].FirstName != "" {
|
|
t.Fatalf("history users = %+v, want authoritative tombstone", list.Users)
|
|
}
|
|
if got := tgUser(list.Users[0]); !got.Deleted || got.ID != deleted.ID {
|
|
t.Fatalf("history TL user = %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestDeletedUserReadModelOverlaysRemainMinimal(t *testing.T) {
|
|
ctx := context.Background()
|
|
viewerID := int64(7)
|
|
deletedID := int64(42)
|
|
peer := domain.Peer{Type: domain.PeerTypeUser, ID: deletedID}
|
|
now := int(time.Now().Unix())
|
|
|
|
storyStore := memory.NewStoryStore()
|
|
if _, err := storyStore.UpsertStory(ctx, domain.UpsertStoryRequest{Story: domain.Story{
|
|
Owner: peer, ID: 1, Date: now, ExpireDate: now + 3600, Public: true,
|
|
}}); err != nil {
|
|
t.Fatalf("upsert retained story: %v", err)
|
|
}
|
|
stories := &countingStoriesService{StoriesService: appstories.NewService(storyStore)}
|
|
usernames := newFakeUsernameRegistry()
|
|
usernames.byPeer[peer] = []domain.Username{{Username: "retained_collectible", Active: true, CollectibleID: 9}}
|
|
verifications := newFakeBotVerifications()
|
|
verifications.marks[peer] = domain.CustomVerification{
|
|
VerifierBotID: 9001, Peer: peer, IconDocumentID: 9002, Description: "retained mark",
|
|
}
|
|
r := New(Config{}, Deps{
|
|
Stories: stories, Usernames: usernames, BotVerifications: verifications,
|
|
}, zaptest.NewLogger(t), clock.System)
|
|
|
|
users := []tg.UserClass{tgUser(domain.User{ID: deletedID, Deleted: true, DeletedAt: int64(now)})}
|
|
r.applyPeerReadModels(ctx, viewerID, users, nil)
|
|
u := users[0].(*tg.User)
|
|
_, usernamesSet := u.GetUsernames()
|
|
_, storiesSet := u.GetStoriesMaxID()
|
|
_, verificationSet := u.GetBotVerificationIcon()
|
|
if !u.Deleted || u.ID != deletedID || usernamesSet || storiesSet || u.GetStoriesHidden() || verificationSet {
|
|
t.Fatalf("deleted user gained retained read-model overlays: %+v", u)
|
|
}
|
|
if stories.projectionCalls != 0 || usernames.peerCalls != 0 || usernames.batchCalls != 0 || verifications.peerCalls != 0 || verifications.batchCalls != 0 {
|
|
t.Fatalf("deleted user triggered overlay reads: stories=%d usernames=(%d,%d) verifications=(%d,%d)",
|
|
stories.projectionCalls, usernames.peerCalls, usernames.batchCalls, verifications.peerCalls, verifications.batchCalls)
|
|
}
|
|
}
|
|
|
|
func TestGetFullDeletedUserReturnsOnlyTombstone(t *testing.T) {
|
|
ctx := context.Background()
|
|
viewer := domain.User{ID: 7, FirstName: "Viewer"}
|
|
deleted := domain.User{
|
|
ID: 42, AccessHash: 99, Phone: "secret", FirstName: "Alice", LastName: "Private",
|
|
Username: "released", About: "retained about", PhotoID: 123, Deleted: true,
|
|
}
|
|
peer := domain.Peer{Type: domain.PeerTypeUser, ID: deleted.ID}
|
|
now := int(time.Now().Unix())
|
|
|
|
storyStore := memory.NewStoryStore()
|
|
if _, err := storyStore.UpsertStory(ctx, domain.UpsertStoryRequest{Story: domain.Story{
|
|
Owner: peer, ID: 1, Date: now, ExpireDate: now + 3600, Public: true,
|
|
}}); err != nil {
|
|
t.Fatalf("upsert retained story: %v", err)
|
|
}
|
|
stories := &countingStoriesService{StoriesService: appstories.NewService(storyStore)}
|
|
usernames := newFakeUsernameRegistry()
|
|
usernames.byPeer[peer] = []domain.Username{{Username: "retained_collectible", Active: true, CollectibleID: 9}}
|
|
verifications := newFakeBotVerifications()
|
|
verifications.marks[peer] = domain.CustomVerification{
|
|
VerifierBotID: 9001, Peer: peer, IconDocumentID: 9002, Description: "retained mark",
|
|
}
|
|
r := New(Config{}, Deps{
|
|
Users: mapUsersService{users: map[int64]domain.User{
|
|
viewer.ID: viewer,
|
|
deleted.ID: deleted,
|
|
}},
|
|
Stories: stories, Usernames: usernames, BotVerifications: verifications,
|
|
}, zaptest.NewLogger(t), clock.System)
|
|
|
|
got, err := r.onUsersGetFullUser(WithUserID(ctx, viewer.ID), &tg.InputUser{
|
|
UserID: deleted.ID, AccessHash: deleted.AccessHash,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("get full deleted user: %v", err)
|
|
}
|
|
wantFull := tg.UserFull{
|
|
ID: deleted.ID,
|
|
Settings: tg.PeerSettings{},
|
|
NotifySettings: *tdesktop.NotifySettings(),
|
|
}
|
|
if !reflect.DeepEqual(got.FullUser, wantFull) {
|
|
t.Fatalf("full deleted user = %+v, want minimal %+v", got.FullUser, wantFull)
|
|
}
|
|
if len(got.Users) != 1 {
|
|
t.Fatalf("users = %+v, want one tombstone", got.Users)
|
|
}
|
|
u, ok := got.Users[0].(*tg.User)
|
|
if !ok || !reflect.DeepEqual(u, &tg.User{ID: deleted.ID, Deleted: true}) {
|
|
t.Fatalf("deleted user envelope = %+v", got.Users[0])
|
|
}
|
|
if len(got.Chats) != 0 {
|
|
t.Fatalf("deleted user chats = %+v, want none", got.Chats)
|
|
}
|
|
if stories.projectionCalls != 0 || usernames.peerCalls != 0 || usernames.batchCalls != 0 || verifications.peerCalls != 0 || verifications.batchCalls != 0 {
|
|
t.Fatalf("deleted getFullUser triggered retained read-model reads: stories=%d usernames=(%d,%d) verifications=(%d,%d)",
|
|
stories.projectionCalls, usernames.peerCalls, usernames.batchCalls, verifications.peerCalls, verifications.batchCalls)
|
|
}
|
|
wire := &tg.UsersUserFull{}
|
|
tlRoundTrip(t, got, wire)
|
|
if !reflect.DeepEqual(wire.FullUser, wantFull) || len(wire.Users) != 1 {
|
|
t.Fatalf("wire deleted user full = %+v", wire)
|
|
}
|
|
}
|