Something went wrong. Try again.
Something went wrong. Try again.
2.1 kB · 54 lines
Go
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455package statedb
import ( "context" "path" "testing" "time"
"github.com/stretchr/testify/require" "stream.place/streamplace/pkg/config")
// Objects recorded under several livestream records come back in one// recording order; an object still uploading is left out.func TestListS3SegmentsForLivestreams(t *testing.T) { ctx := context.Background() state, err := MakeDB(ctx, &config.CLI{DBURL: ":memory:"}, nil, nil) require.NoError(t, err) t0 := time.Date(2026, 9, 16, 7, 12, 34, 0, time.UTC) first, second := "at://did:plc:x/place.stream.livestream/first", "at://did:plc:x/place.stream.livestream/second" rec := func(uri, key string, at time.Time, complete bool) { id, err := RecordStartHelper(ctx, state, uri, key, at) require.NoError(t, err) if complete { require.NoError(t, state.RecordComplete(ctx, id, 1, 100, nil)) } } rec(first, "07-12-34-1.m4s", t0, true) rec(first, "07-22-47-2.m4s", t0.Add(10*time.Minute), true) rec(second, "07-25-15-3.m4s", t0.Add(13*time.Minute), true) rec(second, "07-35-23-4.m4s", t0.Add(23*time.Minute), true) rec(second, "07-45-34-5.m4s", t0.Add(33*time.Minute), false) // still uploading rec("at://did:plc:x/place.stream.livestream/other", "x.m4s", t0.Add(5*time.Minute), true)
segs, err := state.ListS3SegmentsForLivestreams(ctx, []string{second, first}) // caller order irrelevant require.NoError(t, err) keys := make([]string, len(segs)) for i, s := range segs { keys[i] = path.Base(s.Key) } require.Equal(t, []string{"07-12-34-1.m4s", "07-22-47-2.m4s", "07-25-15-3.m4s", "07-35-23-4.m4s"}, keys)
one, err := state.ListS3SegmentsForLivestream(ctx, first) require.NoError(t, err) require.Len(t, one, 2) none, err := state.ListS3SegmentsForLivestreams(ctx, nil) require.NoError(t, err) require.Empty(t, none)}
// RecordStartHelper records an object start for tests.func RecordStartHelper(ctx context.Context, state *StatefulDB, uri, key string, at time.Time) (string, error) { return state.RecordStart(ctx, "did:plc:x", "bucket", "live-rec/did:plc:x/"+key, uri, at)}