Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
6.0 kB · 177 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178package model
import ( "context" "testing"
"github.com/bluesky-social/indigo/atproto/syntax" "github.com/stretchr/testify/require" "stream.place/streamplace/pkg/comatproto"
"stream.place/streamplace/pkg/placestream")
func parseURI(t *testing.T, s string) syntax.ATURI { t.Helper() u, err := syntax.ParseATURI(s) require.NoError(t, err) return u}
// putViewCount writes one place.stream.media.viewCount record under// the given reporter repo. Tracks slice is what shows up in the// summary's bytes / durationMs totals. rkey is a per-call discriminator// — same reporter + same rkey upserts in place.func putViewCount(t *testing.T, m Model, reporter, rkey, videoURI string, count int64, tracks []placestream.MediaViewCount_TrackUsage) { t.Helper() rec := placestream.MediaViewCount{ LexiconTypeID: "place.stream.media.viewCount", Video: videoURI, Count: count, WindowStart: "2026-05-17T12:00:00Z", WindowEnd: "2026-05-17T12:05:00Z", IndexedAt: "2026-05-17T12:15:00Z", Tracks: tracks, } aturi := parseURI(t, "at://"+reporter+"/place.stream.media.viewCount/"+rkey) require.NoError(t, m.UpsertMediaViewCount(context.Background(), rec, aturi))}
func TestGetVideoView_NoViewCounts(t *testing.T) { m, err := MakeDB(":memory:") require.NoError(t, err) ctx := context.Background()
const ( owner = "did:plc:alice" videoURI = "at://did:plc:alice/place.stream.video/v1" ) video := placestream.Video{ LexiconTypeID: "place.stream.video", Title: "hello", Source: placestream.Video_Source{ MediaDefs_SourceTracks: &placestream.MediaDefs_SourceTracks{ LexiconTypeID: "place.stream.media.defs#sourceTracks", Tracks: []comatproto.RepoStrongRef{}, }, }, } require.NoError(t, m.UpsertVideo(ctx, video, parseURI(t, videoURI)))
view, err := m.GetVideoView(ctx, videoURI) require.NoError(t, err) require.NotNil(t, view) require.Equal(t, videoURI, view.Uri) require.Equal(t, owner, view.Author.Did) require.NotNil(t, view.ViewCounts, "summary is always present; zero-valued when no records") require.Equal(t, int64(0), view.ViewCounts.Count) require.Equal(t, int64(0), view.ViewCounts.Bytes) require.Equal(t, int64(0), view.ViewCounts.DurationMs) require.Equal(t, int64(0), view.ViewCounts.Reporters)}
func TestGetVideoView_SumsAcrossReporters(t *testing.T) { m, err := MakeDB(":memory:") require.NoError(t, err) ctx := context.Background()
const videoURI = "at://did:plc:alice/place.stream.video/v1" video := placestream.Video{ LexiconTypeID: "place.stream.video", Title: "popular", Source: placestream.Video_Source{ MediaDefs_SourceTracks: &placestream.MediaDefs_SourceTracks{ LexiconTypeID: "place.stream.media.defs#sourceTracks", }, }, } require.NoError(t, m.UpsertVideo(ctx, video, parseURI(t, videoURI)))
trackRef := comatproto.RepoStrongRef{ LexiconTypeID: "com.atproto.repo.strongRef", Uri: "at://did:plc:alice/place.stream.media.track/t1", Cid: "bafytrack", } // Three reporters, one record each. putViewCount(t, m, "did:plc:node1", "w1", videoURI, 5, []placestream.MediaViewCount_TrackUsage{ {Track: trackRef, Bytes: 100, DurationMs: 1000}, }) putViewCount(t, m, "did:plc:node2", "w1", videoURI, 7, []placestream.MediaViewCount_TrackUsage{ {Track: trackRef, Bytes: 200, DurationMs: 2000}, }) putViewCount(t, m, "did:plc:node3", "w1", videoURI, 3, []placestream.MediaViewCount_TrackUsage{ {Track: trackRef, Bytes: 50, DurationMs: 500}, {Track: trackRef, Bytes: 25, DurationMs: 250}, })
view, err := m.GetVideoView(ctx, videoURI) require.NoError(t, err) require.NotNil(t, view) require.NotNil(t, view.ViewCounts) require.Equal(t, int64(15), view.ViewCounts.Count, "5+7+3") require.Equal(t, int64(375), view.ViewCounts.Bytes, "100+200+(50+25)") require.Equal(t, int64(3750), view.ViewCounts.DurationMs, "1000+2000+(500+250)") require.Equal(t, int64(3), view.ViewCounts.Reporters)}
func TestGetVideoView_NotFound(t *testing.T) { m, err := MakeDB(":memory:") require.NoError(t, err)
view, err := m.GetVideoView(context.Background(), "at://did:plc:nobody/place.stream.video/v404") require.NoError(t, err) require.Nil(t, view, "missing video returns (nil, nil) so callers can surface a 404 without sniffing errors")}
func TestUpsertMediaViewCountIdempotent(t *testing.T) { m, err := MakeDB(":memory:") require.NoError(t, err) ctx := context.Background()
const ( reporter = "did:plc:node1" videoURI = "at://did:plc:alice/place.stream.video/v1" ) rec := placestream.MediaViewCount{ LexiconTypeID: "place.stream.media.viewCount", Video: videoURI, Count: 5, WindowStart: "2026-05-17T12:00:00Z", WindowEnd: "2026-05-17T12:05:00Z", IndexedAt: "2026-05-17T12:15:00Z", } aturi := parseURI(t, "at://"+reporter+"/place.stream.media.viewCount/win-1")
require.NoError(t, m.UpsertMediaViewCount(ctx, rec, aturi)) // Second call with the same URI replaces, doesn't append. rec.Count = 8 require.NoError(t, m.UpsertMediaViewCount(ctx, rec, aturi))
got, err := m.GetMediaViewCountByURI(ctx, aturi.String()) require.NoError(t, err) require.NotNil(t, got) require.Equal(t, int64(8), got.Count, "upsert replaces by URI")}
func TestDeleteMediaViewCount(t *testing.T) { m, err := MakeDB(":memory:") require.NoError(t, err) ctx := context.Background()
const videoURI = "at://did:plc:alice/place.stream.video/v1" rec := placestream.MediaViewCount{ LexiconTypeID: "place.stream.media.viewCount", Video: videoURI, Count: 5, WindowStart: "2026-05-17T12:00:00Z", WindowEnd: "2026-05-17T12:05:00Z", IndexedAt: "2026-05-17T12:15:00Z", } aturi := parseURI(t, "at://did:plc:node1/place.stream.media.viewCount/win-1") require.NoError(t, m.UpsertMediaViewCount(ctx, rec, aturi))
require.NoError(t, m.DeleteMediaViewCount(ctx, aturi.String())) got, err := m.GetMediaViewCountByURI(ctx, aturi.String()) require.NoError(t, err) require.Nil(t, got)}