merged from gramsrv upstream

internal/store/memory/channel_stats_test.go

@@ -0,0 +1,133 @@
+package memory
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"telesrv/internal/domain"
+)
+
+func TestChannelStatsUseDurableFactsAndPagePublicForwards(t *testing.T) {
+	ctx := context.Background()
+	store := NewChannelStore()
+	const owner, viewer int64 = 1, 2
+	period := domain.StatsPeriod{MinDate: 1_700_006_400, MaxDate: 1_700_611_200}
+
+	source, err := store.CreateChannel(ctx, domain.CreateChannelRequest{
+		CreatorUserID: owner,
+		Title:         "stats source",
+		Broadcast:     true,
+		MemberUserIDs: []int64{viewer},
+		Date:          period.MinDate - 100,
+	})
+	if err != nil {
+		t.Fatalf("create source: %v", err)
+	}
+	previous, err := store.SendChannelMessage(ctx, domain.SendChannelMessageRequest{
+		UserID: owner, ChannelID: source.Channel.ID, RandomID: 100, Message: "previous", Date: period.MinDate - 10,
+	})
+	if err != nil {
+		t.Fatalf("send previous: %v", err)
+	}
+	_ = previous
+	post, err := store.SendChannelMessage(ctx, domain.SendChannelMessageRequest{
+		UserID: owner, ChannelID: source.Channel.ID, RandomID: 101, Message: "current post", Date: period.MinDate + 10,
+	})
+	if err != nil {
+		t.Fatalf("send current: %v", err)
+	}
+	if _, err := store.GetChannelMessageViews(ctx, domain.ChannelMessageViewsRequest{
+		UserID: viewer, ChannelID: source.Channel.ID, IDs: []int{post.Message.ID}, Increment: true, Date: period.MinDate + 20,
+	}); err != nil {
+		t.Fatalf("increment view: %v", err)
+	}
+	reaction := domain.MessageReaction{Type: domain.MessageReactionEmoji, Emoticon: "👍"}
+	if _, err := store.SetChannelMessageReactions(ctx, domain.SetChannelMessageReactionsRequest{
+		UserID: viewer, ChannelID: source.Channel.ID, MessageID: post.Message.ID, Reactions: []domain.MessageReaction{reaction}, Date: period.MinDate + 30,
+	}); err != nil {
+		t.Fatalf("react: %v", err)
+	}
+
+	publicCreated, err := store.CreateChannel(ctx, domain.CreateChannelRequest{
+		CreatorUserID: owner, Title: "public destination", Broadcast: true, Date: period.MinDate + 40,
+	})
+	if err != nil {
+		t.Fatalf("create public destination: %v", err)
+	}
+	publicChannel, err := store.UpdateUsername(ctx, domain.UpdateChannelUsernameRequest{
+		UserID: owner, ChannelID: publicCreated.Channel.ID, Username: "stats_forward_memory",
+	})
+	if err != nil {
+		t.Fatalf("make public destination: %v", err)
+	}
+	privateChannel, err := store.CreateChannel(ctx, domain.CreateChannelRequest{
+		CreatorUserID: owner, Title: "private destination", Broadcast: true, Date: period.MinDate + 40,
+	})
+	if err != nil {
+		t.Fatalf("create private destination: %v", err)
+	}
+	forward := &domain.MessageForward{
+		From: domain.Peer{Type: domain.PeerTypeChannel, ID: source.Channel.ID}, Date: post.Message.Date, ChannelPost: post.Message.ID,
+	}
+	for i, date := range []int{period.MinDate + 50, period.MinDate + 60} {
+		if _, err := store.SendChannelMessage(ctx, domain.SendChannelMessageRequest{
+			UserID: owner, ChannelID: publicChannel.ID, RandomID: int64(200 + i), Message: "public forward", Forward: forward, Date: date,
+		}); err != nil {
+			t.Fatalf("send public forward %d: %v", i, err)
+		}
+	}
+	if _, err := store.SendChannelMessage(ctx, domain.SendChannelMessageRequest{
+		UserID: owner, ChannelID: privateChannel.Channel.ID, RandomID: 300, Message: "private forward", Forward: forward, Date: period.MinDate + 70,
+	}); err != nil {
+		t.Fatalf("send private forward: %v", err)
+	}
+
+	stats, err := store.GetChannelStats(ctx, domain.ChannelStatsRequest{
+		ViewerUserID: owner, ChannelID: source.Channel.ID, Period: period,
+	})
+	if err != nil {
+		t.Fatalf("get stats: %v", err)
+	}
+	if stats.Members.Current != 2 || stats.Messages.Current != 1 || stats.Messages.Previous != 1 ||
+		stats.Viewers.Current != 1 || stats.Posters.Current != 1 || stats.ViewsPerPost.Current != 1 ||
+		stats.SharesPerPost.Current != 2 || stats.ReactionsPerPost.Current != 1 {
+		t.Fatalf("stats = %+v, want durable current/previous aggregates", stats)
+	}
+	if len(stats.Days) == 0 || len(stats.Days[0].ByReaction) != 1 || stats.Days[0].Shares != 2 {
+		t.Fatalf("stats days = %+v, want view/share/reaction buckets", stats.Days)
+	}
+	messageStats, err := store.GetChannelMessageStats(ctx, domain.ChannelMessageStatsRequest{
+		ViewerUserID: owner, ChannelID: source.Channel.ID, MessageID: post.Message.ID, Period: period,
+	})
+	if err != nil {
+		t.Fatalf("get message stats: %v", err)
+	}
+	if len(messageStats.Days) == 0 || messageStats.Days[0].Views != 1 || messageStats.Days[0].Reactions != 1 {
+		t.Fatalf("message stats days = %+v, want one view and reaction", messageStats.Days)
+	}
+
+	first, err := store.ListChannelMessagePublicForwards(ctx, domain.ChannelMessagePublicForwardListRequest{
+		ViewerUserID: owner, ChannelID: source.Channel.ID, MessageID: post.Message.ID, Limit: 1,
+	})
+	if err != nil {
+		t.Fatalf("list first forward page: %v", err)
+	}
+	if first.Count != 2 || len(first.Messages) != 1 || first.NextOffset == "" || first.Messages[0].ChannelID != publicChannel.ID {
+		t.Fatalf("first forward page = %+v, want one of two public forwards", first)
+	}
+	second, err := store.ListChannelMessagePublicForwards(ctx, domain.ChannelMessagePublicForwardListRequest{
+		ViewerUserID: owner, ChannelID: source.Channel.ID, MessageID: post.Message.ID, Offset: first.NextOffset, Limit: 1,
+	})
+	if err != nil {
+		t.Fatalf("list second forward page: %v", err)
+	}
+	if second.Count != 2 || len(second.Messages) != 1 || second.Messages[0].ID == first.Messages[0].ID || second.NextOffset != "" {
+		t.Fatalf("second forward page = %+v, want remaining public forward", second)
+	}
+	if _, err := store.ListChannelMessagePublicForwards(ctx, domain.ChannelMessagePublicForwardListRequest{
+		ViewerUserID: owner, ChannelID: source.Channel.ID, MessageID: post.Message.ID, Offset: "bad", Limit: 1,
+	}); !errors.Is(err, domain.ErrStatsOffsetInvalid) {
+		t.Fatalf("invalid cursor err = %v, want ErrStatsOffsetInvalid", err)
+	}
+}