QuickandGitHub
cac435c4b9
* Refine playback session handling and API responses * fix(watchstate): harden completed-history visibility
127 lines
4.0 KiB
Go
127 lines
4.0 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Silo-Server/silo-server/internal/models"
|
|
"github.com/Silo-Server/silo-server/internal/userstore"
|
|
)
|
|
|
|
func TestResolveItemUserStatesIncludesCompletedEbookReaderProgress(t *testing.T) {
|
|
ctx := context.Background()
|
|
store := newProfileTestStore(t)
|
|
items := []*models.MediaItem{
|
|
{ContentID: "ebook-complete", Type: "ebook", Title: "Complete Ebook"},
|
|
{ContentID: "ebook-progress", Type: "ebook", Title: "Partial Ebook"},
|
|
}
|
|
|
|
states, err := resolveItemUserStatesWithOptions(ctx, store, "profile-1", nil, items, itemUserStateOptions{
|
|
UserID: 42,
|
|
EbookProgressStore: &fakeEbookReaderProgressLister{
|
|
progress: map[string]EbookReaderProgress{
|
|
"ebook-complete": {
|
|
UserID: 42,
|
|
ProfileID: "profile-1",
|
|
ContentID: "ebook-complete",
|
|
Progress: 0.95,
|
|
UpdatedAt: time.Date(2026, 6, 4, 10, 0, 0, 0, time.UTC),
|
|
},
|
|
"ebook-progress": {
|
|
UserID: 42,
|
|
ProfileID: "profile-1",
|
|
ContentID: "ebook-progress",
|
|
Progress: 0.45,
|
|
UpdatedAt: time.Date(2026, 6, 4, 9, 0, 0, 0, time.UTC),
|
|
},
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("resolveItemUserStatesWithOptions: %v", err)
|
|
}
|
|
|
|
if states["ebook-complete"] == nil || !states["ebook-complete"].Played {
|
|
t.Fatalf("completed ebook state = %+v, want played", states["ebook-complete"])
|
|
}
|
|
if states["ebook-progress"] == nil || states["ebook-progress"].Played {
|
|
t.Fatalf("partial ebook state = %+v, want not played", states["ebook-progress"])
|
|
}
|
|
}
|
|
|
|
func TestResolveItemUserStatesExcludesHiddenEbookReaderProgress(t *testing.T) {
|
|
ctx := context.Background()
|
|
store := newProfileTestStore(t)
|
|
updatedAt := time.Date(2026, 6, 4, 10, 0, 0, 0, time.UTC)
|
|
items := []*models.MediaItem{
|
|
{ContentID: "ebook-hidden", Type: "ebook", Title: "Hidden Ebook"},
|
|
{ContentID: "ebook-reread", Type: "ebook", Title: "Reread Ebook"},
|
|
}
|
|
|
|
states, err := resolveItemUserStatesWithOptions(ctx, store, "profile-1", nil, items, itemUserStateOptions{
|
|
UserID: 42,
|
|
EbookProgressStore: &fakeEbookReaderProgressLister{
|
|
progress: map[string]EbookReaderProgress{
|
|
"ebook-hidden": {
|
|
UserID: 42,
|
|
ProfileID: "profile-1",
|
|
ContentID: "ebook-hidden",
|
|
Progress: 0.95,
|
|
UpdatedAt: updatedAt,
|
|
},
|
|
"ebook-reread": {
|
|
UserID: 42,
|
|
ProfileID: "profile-1",
|
|
ContentID: "ebook-reread",
|
|
Progress: 0.95,
|
|
UpdatedAt: updatedAt.Add(2 * time.Hour),
|
|
},
|
|
},
|
|
hiddenBefore: map[string]time.Time{
|
|
"ebook-hidden": updatedAt.Add(time.Hour),
|
|
"ebook-reread": updatedAt.Add(time.Hour),
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("resolveItemUserStatesWithOptions: %v", err)
|
|
}
|
|
|
|
if states["ebook-hidden"] == nil || states["ebook-hidden"].Played {
|
|
t.Fatalf("hidden ebook state = %+v, want not played", states["ebook-hidden"])
|
|
}
|
|
if states["ebook-reread"] == nil || !states["ebook-reread"].Played {
|
|
t.Fatalf("ebook updated after hidden_before = %+v, want played again", states["ebook-reread"])
|
|
}
|
|
}
|
|
|
|
func TestResolveItemUserStatesIncludesCompletedHistory(t *testing.T) {
|
|
ctx := context.Background()
|
|
store := newProfileTestStore(t)
|
|
addCompletedHistoryForUserDataTest(t, store, "movie-history-only")
|
|
items := []*models.MediaItem{
|
|
{ContentID: "movie-history-only", Type: "movie", Title: "Imported Movie"},
|
|
}
|
|
|
|
states, err := resolveItemUserStates(ctx, store, "profile-1", nil, items)
|
|
if err != nil {
|
|
t.Fatalf("resolveItemUserStates: %v", err)
|
|
}
|
|
if states["movie-history-only"] == nil || !states["movie-history-only"].Played {
|
|
t.Fatalf("history-only movie state = %+v, want played", states["movie-history-only"])
|
|
}
|
|
}
|
|
|
|
func TestAllEpisodesCompletedIncludesCompletedHistory(t *testing.T) {
|
|
episodes := []*models.Episode{{ContentID: "episode-progress"}, {ContentID: "episode-history"}}
|
|
progress := map[string]userstore.WatchProgress{
|
|
"episode-progress": {MediaItemID: "episode-progress", Completed: true},
|
|
"episode-history": {MediaItemID: "episode-history", Completed: true},
|
|
}
|
|
|
|
if !allEpisodesCompleted(episodes, progress) {
|
|
t.Fatal("allEpisodesCompleted = false, want completed from progress plus history")
|
|
}
|
|
}
|