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

70
internal/rpc/story_peer_projection_rpc_test.go
View file

@ -14,16 +14,19 @@ import (
appstories "telesrv/internal/app/stories"
appusers "telesrv/internal/app/users"
"telesrv/internal/domain"
"telesrv/internal/store"
"telesrv/internal/store/memory"
)
type countingStoriesService struct {
StoriesService
maxIDCalls int
hiddenCalls int
projectionCalls int
pinnedAvailCalls int
pinnedStoriesCalls int
maxIDCalls int
hiddenCalls int
projectionCalls int
activeCandidateCalls int
hiddenSnapshotCalls int
pinnedAvailCalls int
pinnedStoriesCalls int
}
type blockingProjectionStoriesService struct {
@ -69,6 +72,16 @@ func (s *countingStoriesService) GetPeerStoryProjections(ctx context.Context, vi
return s.StoriesService.GetPeerStoryProjections(ctx, viewerUserID, peers, now)
}
func (s *countingStoriesService) ActiveStoryPeerExpirations(ctx context.Context, peers []domain.Peer, now int) (map[domain.Peer]int, error) {
s.activeCandidateCalls++
return s.StoriesService.(storySparseProjectionProvider).ActiveStoryPeerExpirations(ctx, peers, now)
}
func (s *countingStoriesService) ListHiddenStoryPeers(ctx context.Context, viewerUserID int64) ([]domain.Peer, error) {
s.hiddenSnapshotCalls++
return s.StoriesService.(storySparseProjectionProvider).ListHiddenStoryPeers(ctx, viewerUserID)
}
func (s *countingStoriesService) HasPinnedStories(ctx context.Context, viewerUserID int64, peer domain.Peer, now int) (bool, error) {
s.pinnedAvailCalls++
return s.StoriesService.HasPinnedStories(ctx, viewerUserID, peer, now)
@ -130,6 +143,43 @@ func TestUsersProjectStoriesMaxID(t *testing.T) {
}
}
func TestStorySparseProjectionAvoidsViewerPeerNegativeMatrix(t *testing.T) {
ctx := context.Background()
const viewerID int64 = 94001
active := domain.Peer{Type: domain.PeerTypeUser, ID: 94002}
inactiveHidden := domain.Peer{Type: domain.PeerTypeChannel, ID: 94003}
storyStore := memory.NewStoryStore()
if _, err := storyStore.UpsertStory(ctx, domain.UpsertStoryRequest{Story: domain.Story{
Owner: active, ID: 7, Date: 100, ExpireDate: 200, Public: true,
}}); err != nil {
t.Fatalf("upsert active story: %v", err)
}
if err := storyStore.SetPeerHidden(ctx, viewerID, inactiveHidden, true); err != nil {
t.Fatalf("hide inactive peer: %v", err)
}
service := &countingStoriesService{StoriesService: appstories.NewService(storyStore)}
versions := &fakeRPCReadModelVersions{hashes: map[store.ReadModelKey]int64{
storyPeerVersionKey(active): 401,
storyPeerVersionKey(inactiveHidden): 402,
storyHiddenListVersionKey(viewerID): 403,
}}
r := New(Config{}, Deps{Stories: service, ReadModelVersions: versions}, zaptest.NewLogger(t), fixedClock{now: time.Unix(150, 0)})
for i := 0; i < 2; i++ {
recent, hidden := r.storyProjectionMaps(ctx, viewerID, []domain.Peer{active, inactiveHidden})
if story, ok := recent[active]; !ok || story.MaxID != 7 {
t.Fatalf("recent(%d) = %+v, want active max id 7", i, recent)
}
if _, ok := recent[inactiveHidden]; ok || !hidden[inactiveHidden] {
t.Fatalf("inactive hidden projection(%d) recent=%+v hidden=%+v", i, recent, hidden)
}
}
if service.activeCandidateCalls != 1 || service.hiddenSnapshotCalls != 1 || service.projectionCalls != 1 {
t.Fatalf("sparse loads active=%d hidden=%d viewerProjection=%d, want 1/1/1",
service.activeCandidateCalls, service.hiddenSnapshotCalls, service.projectionCalls)
}
}
func TestUsersProjectStoriesHiddenWithoutActiveStory(t *testing.T) {
ctx := context.Background()
userStore := memory.NewUserStore()
@ -1275,7 +1325,7 @@ func TestBuildOutboxStoryUpdatesHydratesCompanionPeersWithStoriesMaxID(t *testin
Usernames: registry,
}, zaptest.NewLogger(t), fixedClock{now: time.Unix(1700000300, 0)})
updates := r.BuildOutboxUpdates(ctx, []OutboxUpdateRequest{{
updates, err := r.BuildOutboxUpdates(ctx, []OutboxUpdateRequest{{
TargetUserID: viewer.ID,
Event: domain.UpdateEvent{
UserID: viewer.ID,
@ -1287,6 +1337,9 @@ func TestBuildOutboxStoryUpdatesHydratesCompanionPeersWithStoriesMaxID(t *testin
Story: story,
},
}})
if err != nil {
t.Fatalf("BuildOutboxUpdates: %v", err)
}
if len(updates) != 1 || updates[0] == nil {
t.Fatalf("updates = %+v, want one story update", updates)
}
@ -1333,7 +1386,7 @@ func TestBuildOutboxNewStoryReactionHydratesReactorUser(t *testing.T) {
Stories: appstories.NewService(storyStore),
}, zaptest.NewLogger(t), fixedClock{now: time.Unix(1700000310, 0)})
updates := r.BuildOutboxUpdates(ctx, []OutboxUpdateRequest{{
updates, err := r.BuildOutboxUpdates(ctx, []OutboxUpdateRequest{{
TargetUserID: owner.ID,
Event: domain.UpdateEvent{
UserID: owner.ID,
@ -1347,6 +1400,9 @@ func TestBuildOutboxNewStoryReactionHydratesReactorUser(t *testing.T) {
Reaction: &domain.MessageReaction{Type: domain.MessageReactionEmoji, Emoticon: "🔥"},
},
}})
if err != nil {
t.Fatalf("BuildOutboxUpdates: %v", err)
}
if len(updates) != 1 || updates[0] == nil {
t.Fatalf("updates = %+v, want one new story reaction update", updates)
}