Big changes
This commit is contained in:
ponzischeme89
@@ -7,11 +7,14 @@ import (
|
||||
"io"
|
||||
"log/slog"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/ponzischeme89/memby/server/internal/emby"
|
||||
"github.com/ponzischeme89/memby/server/internal/tracearr"
|
||||
)
|
||||
|
||||
// fakeSource records the queries the engine makes and replays canned answers.
|
||||
@@ -22,6 +25,8 @@ type fakeSource struct {
|
||||
similar map[string][]json.RawMessage
|
||||
itemsErr error
|
||||
similarErr error
|
||||
nextUp []json.RawMessage
|
||||
nextUpErr error
|
||||
|
||||
genreQueries []string
|
||||
similarSeeds []string
|
||||
@@ -31,6 +36,36 @@ type fakeCuratedLibrary struct {
|
||||
byGenre map[string][]json.RawMessage
|
||||
}
|
||||
|
||||
type fakeForYouLibrary struct {
|
||||
items []json.RawMessage
|
||||
}
|
||||
|
||||
func (f *fakeForYouLibrary) AllRecommendationCandidates(
|
||||
_ context.Context,
|
||||
) ([]json.RawMessage, error) {
|
||||
return f.items, nil
|
||||
}
|
||||
|
||||
func (f *fakeForYouLibrary) LibraryCandidates(
|
||||
_ context.Context,
|
||||
_ []string,
|
||||
_ int,
|
||||
) ([]json.RawMessage, error) {
|
||||
return f.items, nil
|
||||
}
|
||||
|
||||
type fakeTracearr struct {
|
||||
sessions []tracearr.Session
|
||||
}
|
||||
|
||||
func (f fakeTracearr) History(
|
||||
_ context.Context,
|
||||
_ string,
|
||||
_ int,
|
||||
) ([]tracearr.Session, error) {
|
||||
return f.sessions, nil
|
||||
}
|
||||
|
||||
func (f *fakeCuratedLibrary) LibraryCandidates(
|
||||
_ context.Context,
|
||||
_ []string,
|
||||
@@ -62,7 +97,11 @@ func (f *fakeSource) Items(_ context.Context, _ emby.Credentials, params url.Val
|
||||
f.genreQueries = append(f.genreQueries, genres)
|
||||
}
|
||||
key := params.Get("Filters")
|
||||
return &emby.ItemsResult{Items: f.itemsByFilter[key]}, nil
|
||||
items := f.itemsByFilter[key]
|
||||
if limit, err := strconv.Atoi(params.Get("Limit")); err == nil && limit > 0 && len(items) > limit {
|
||||
items = items[:limit]
|
||||
}
|
||||
return &emby.ItemsResult{Items: items}, nil
|
||||
}
|
||||
|
||||
func (f *fakeSource) Similar(_ context.Context, _ emby.Credentials, itemID string, _ url.Values) (*emby.ItemsResult, error) {
|
||||
@@ -75,6 +114,69 @@ func (f *fakeSource) Similar(_ context.Context, _ emby.Credentials, itemID strin
|
||||
return &emby.ItemsResult{Items: f.similar[itemID]}, nil
|
||||
}
|
||||
|
||||
func (f *fakeSource) NextUp(
|
||||
_ context.Context,
|
||||
_ emby.Credentials,
|
||||
_ url.Values,
|
||||
) (*emby.ItemsResult, error) {
|
||||
if f.nextUpErr != nil {
|
||||
return nil, f.nextUpErr
|
||||
}
|
||||
return &emby.ItemsResult{Items: f.nextUp}, nil
|
||||
}
|
||||
|
||||
func TestAbandonedShowsRequireAnEmbyNextUpAndRespectSeasonProgress(t *testing.T) {
|
||||
now := time.Date(2026, time.July, 29, 12, 0, 0, 0, time.UTC)
|
||||
catalogue := Decode([]json.RawMessage{
|
||||
json.RawMessage(`{"Id":"early","Name":"Early Show","Type":"Series"}`),
|
||||
json.RawMessage(`{"Id":"deep","Name":"Deep Show","Type":"Series"}`),
|
||||
json.RawMessage(`{"Id":"complete","Name":"Complete Show","Type":"Series"}`),
|
||||
json.RawMessage(`{"Id":"recent","Name":"Recent Show","Type":"Series"}`),
|
||||
})
|
||||
session := func(show string, season, episode, daysAgo int) tracearr.Session {
|
||||
value := tracearr.Session{
|
||||
ID: "session-" + show, MediaType: "episode", ShowTitle: show,
|
||||
SeasonNumber: intPointer(season), EpisodeNumber: intPointer(episode),
|
||||
Watched: true, StoppedAt: now.AddDate(0, 0, -daysAgo).Format(time.RFC3339Nano),
|
||||
}
|
||||
return value
|
||||
}
|
||||
sessions := []tracearr.Session{
|
||||
session("Early Show", 1, 2, 40),
|
||||
session("Deep Show", 2, 8, 60),
|
||||
session("Complete Show", 4, 10, 50),
|
||||
session("Recent Show", 1, 3, 5),
|
||||
}
|
||||
nextUp := Decode([]json.RawMessage{
|
||||
json.RawMessage(`{"Id":"early-next","Type":"Episode","SeriesId":"early","ParentIndexNumber":1,"IndexNumber":3,"RunTimeTicks":18000000000}`),
|
||||
json.RawMessage(`{"Id":"deep-next","Type":"Episode","SeriesId":"deep","ParentIndexNumber":3,"IndexNumber":1,"RunTimeTicks":36000000000}`),
|
||||
json.RawMessage(`{"Id":"recent-next","Type":"Episode","SeriesId":"recent","ParentIndexNumber":1,"IndexNumber":4}`),
|
||||
})
|
||||
|
||||
got := abandonedShowCandidates(
|
||||
newCatalogueIndex(catalogue),
|
||||
sessions,
|
||||
nextUp,
|
||||
compatibilityProfile{directCodecs: map[string]int{}, transcodeCodecs: map[string]int{}},
|
||||
now,
|
||||
)
|
||||
if len(got) != 2 {
|
||||
t.Fatalf("pickup candidates = %+v, want only two abandoned unfinished shows", got)
|
||||
}
|
||||
if got[0].ItemID != "deep" || got[1].ItemID != "early" {
|
||||
t.Fatalf("pickup order = %q, %q; later-season commitment should lead", got[0].ItemID, got[1].ItemID)
|
||||
}
|
||||
if got[0].RecommendationReason != "You made it through season 2 · season 3 is waiting" {
|
||||
t.Fatalf("later-season reason = %q", got[0].RecommendationReason)
|
||||
}
|
||||
if got[1].RecommendationReason != "You left this in season 1 · pick it up again" {
|
||||
t.Fatalf("season-one reason = %q", got[1].RecommendationReason)
|
||||
}
|
||||
if got[0].RuntimeMinutes != 60 || got[1].RuntimeMinutes != 30 {
|
||||
t.Fatalf("next-episode runtimes = %d, %d", got[0].RuntimeMinutes, got[1].RuntimeMinutes)
|
||||
}
|
||||
}
|
||||
|
||||
func raw(id, name, itemType string, genres ...string) json.RawMessage {
|
||||
quoted := make([]string, 0, len(genres))
|
||||
for _, g := range genres {
|
||||
@@ -128,6 +230,117 @@ func TestBuildRowsProducesSimilarAndHistoryRows(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildForYouFiltersTimeAndAddsExplanation(t *testing.T) {
|
||||
source := &fakeSource{
|
||||
itemsByFilter: map[string][]json.RawMessage{
|
||||
"IsResumable": {},
|
||||
"IsPlayed": {
|
||||
json.RawMessage(`{"Id":"seen","Name":"Arrival","Type":"Movie","Genres":["Science Fiction"],"RunTimeTicks":69600000000}`),
|
||||
},
|
||||
"IsFavorite": {},
|
||||
},
|
||||
}
|
||||
engine := testEngine(source)
|
||||
engine.Library = &fakeForYouLibrary{items: []json.RawMessage{
|
||||
json.RawMessage(`{"Id":"short","Name":"Moon","Type":"Movie","Genres":["Science Fiction"],"RunTimeTicks":54000000000,"MediaStreams":[{"Type":"Video","Codec":"h264"}]}`),
|
||||
json.RawMessage(`{"Id":"long","Name":"Dune","Type":"Movie","Genres":["Science Fiction"],"RunTimeTicks":93000000000,"MediaStreams":[{"Type":"Video","Codec":"hevc"}]}`),
|
||||
}}
|
||||
session := tracearr.Session{
|
||||
MediaType: "movie",
|
||||
MediaTitle: "Arrival",
|
||||
Watched: true,
|
||||
Platform: "Android TV",
|
||||
SourceVideoCodec: "h264",
|
||||
VideoDecision: "directplay",
|
||||
}
|
||||
engine.Tracearr = fakeTracearr{sessions: []tracearr.Session{session}}
|
||||
engine.MinRowItems = 1
|
||||
|
||||
rows, err := engine.BuildForYou(
|
||||
context.Background(),
|
||||
emby.Credentials{UserID: "u1"},
|
||||
"Matt",
|
||||
ForYouOptions{AvailableMinutes: 100},
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(rows) != 1 || len(rows[0].Items) != 1 {
|
||||
t.Fatalf("rows = %+v", rows)
|
||||
}
|
||||
if !strings.Contains(string(rows[0].Items[0]), `"MembyRecommendationReason"`) ||
|
||||
!strings.Contains(string(rows[0].Items[0]), `100-minute window`) {
|
||||
t.Fatalf("explanation missing: %s", rows[0].Items[0])
|
||||
}
|
||||
if strings.Contains(string(rows[0].Items[0]), `"Id":"long"`) {
|
||||
t.Fatalf("over-budget item was retained: %s", rows[0].Items[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrepareForYouKeepsAnOverProvisionedPoolAndSpecificEvidence(t *testing.T) {
|
||||
source := &fakeSource{itemsByFilter: map[string][]json.RawMessage{
|
||||
"IsResumable": {},
|
||||
"IsPlayed": {
|
||||
json.RawMessage(`{"Id":"watched","Name":"Arrival","Type":"Movie","ProductionYear":2016,"Genres":["Science Fiction"],"UserData":{"Played":true}}`),
|
||||
},
|
||||
"IsFavorite": {},
|
||||
}}
|
||||
catalogue := []json.RawMessage{
|
||||
json.RawMessage(`{"Id":"arrival","Name":"Arrival","Type":"Movie","ProductionYear":2016,"Genres":["Science Fiction"],"RunTimeTicks":69600000000}`),
|
||||
}
|
||||
for i := 0; i < 30; i++ {
|
||||
catalogue = append(catalogue, json.RawMessage(
|
||||
`{"Id":"candidate-`+strconv.Itoa(i)+`","Name":"Candidate `+strconv.Itoa(i)+
|
||||
`","Type":"Movie","Genres":["Science Fiction"],"RunTimeTicks":54000000000}`,
|
||||
))
|
||||
}
|
||||
engine := testEngine(source)
|
||||
engine.Library = &fakeForYouLibrary{items: catalogue}
|
||||
session := tracearr.Session{
|
||||
ID: "trace-1", ServerID: "server-1", MediaTitle: "Arrival",
|
||||
MediaType: "movie", Year: intPointer(2016), Watched: true,
|
||||
StartedAt: "2026-07-20T08:00:00Z",
|
||||
}
|
||||
session.User.ID = "trace-user"
|
||||
session.User.Username = "Matt"
|
||||
|
||||
result, err := engine.PrepareForYou(
|
||||
context.Background(), emby.Credentials{UserID: "emby-user"}, "Matt",
|
||||
[]tracearr.Session{session},
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatalf("PrepareForYou: %v", err)
|
||||
}
|
||||
if len(result.Candidates) < 30 {
|
||||
t.Fatalf("prepared pool was prematurely row-sized: %d candidates", len(result.Candidates))
|
||||
}
|
||||
foundSpecific := false
|
||||
for _, candidate := range result.Candidates {
|
||||
if strings.Contains(candidate.RecommendationReason, "Because you finished Arrival") {
|
||||
foundSpecific = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !foundSpecific {
|
||||
t.Fatal("expected a candidate explanation grounded in the completed Tracearr title")
|
||||
}
|
||||
if result.Profile.TracearrUserID != "trace-user" || len(result.Mappings) != 1 {
|
||||
t.Fatalf("profile/mapping = %+v / %+v", result.Profile, result.Mappings)
|
||||
}
|
||||
}
|
||||
|
||||
func intPointer(value int) *int { return &value }
|
||||
|
||||
func TestStableEvidenceIndexDistributesCandidatesAcrossCompletedTitles(t *testing.T) {
|
||||
seen := map[int]bool{}
|
||||
for i := 0; i < 20; i++ {
|
||||
seen[stableEvidenceIndex("candidate-"+strconv.Itoa(i), 3)] = true
|
||||
}
|
||||
if len(seen) != 3 {
|
||||
t.Fatalf("evidence indices = %+v, want all three sources represented", seen)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildRowsQueriesTheProfilesTopGenres(t *testing.T) {
|
||||
source := &fakeSource{
|
||||
itemsByFilter: map[string][]json.RawMessage{
|
||||
@@ -179,6 +392,123 @@ func TestBuildRowsExcludesAlreadyWatchedFromSimilarRow(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildRowsExcludesPlayedTitlesBeyondTheOldHistoryWindow(t *testing.T) {
|
||||
played := make([]json.RawMessage, 0, 61)
|
||||
for i := 0; i < 60; i++ {
|
||||
played = append(played, raw("history-"+strconv.Itoa(i), "History", "Movie", "Drama"))
|
||||
}
|
||||
played = append(played, raw("old-watched", "Old Watched", "Movie", "Drama"))
|
||||
source := &fakeSource{
|
||||
itemsByFilter: map[string][]json.RawMessage{"IsPlayed": played},
|
||||
similar: map[string][]json.RawMessage{},
|
||||
}
|
||||
engine := testEngine(source)
|
||||
engine.Library = &fakeForYouLibrary{items: []json.RawMessage{
|
||||
raw("old-watched", "Old Watched", "Movie", "Drama"),
|
||||
raw("new-pick", "New Pick", "Movie", "Drama"),
|
||||
}}
|
||||
engine.MinRowItems = 1
|
||||
|
||||
rows, err := engine.BuildRows(context.Background(), emby.Credentials{UserID: "u1"})
|
||||
if err != nil {
|
||||
t.Fatalf("BuildRows: %v", err)
|
||||
}
|
||||
for _, row := range rows {
|
||||
for _, candidate := range row.Items {
|
||||
if strings.Contains(string(candidate), `"Id":"old-watched"`) {
|
||||
t.Fatalf("row %q retained a title older than the previous 60-item exclusion window", row.ID)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildForYouExcludesTracearrCompletedTitleMissingFromEmbyHistory(t *testing.T) {
|
||||
source := &fakeSource{itemsByFilter: map[string][]json.RawMessage{
|
||||
"IsPlayed": {raw("taste", "Taste", "Movie", "Science Fiction")},
|
||||
}}
|
||||
engine := testEngine(source)
|
||||
engine.Library = &fakeForYouLibrary{items: []json.RawMessage{
|
||||
json.RawMessage(`{"Id":"watched-copy","Name":"Arrival","Type":"Movie","ProductionYear":2016,"Genres":["Science Fiction"]}`),
|
||||
raw("unseen", "Moon", "Movie", "Science Fiction"),
|
||||
}}
|
||||
year := 2016
|
||||
session := tracearr.Session{
|
||||
MediaType: "movie", MediaTitle: "Arrival", Year: &year, Watched: true,
|
||||
}
|
||||
engine.Tracearr = fakeTracearr{sessions: []tracearr.Session{session}}
|
||||
engine.MinRowItems = 1
|
||||
|
||||
rows, err := engine.BuildForYou(
|
||||
context.Background(), emby.Credentials{UserID: "u1"}, "FamilyTV", ForYouOptions{},
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatalf("BuildForYou: %v", err)
|
||||
}
|
||||
if len(rows) != 1 || len(rows[0].Items) != 1 ||
|
||||
!strings.Contains(string(rows[0].Items[0]), `"Id":"unseen"`) {
|
||||
t.Fatalf("Tracearr-completed title was not excluded: %+v", rows)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPreparedExplanationsLimitOneSourceAndMixReasonKinds(t *testing.T) {
|
||||
profile := Profile{
|
||||
GenreWeights: map[string]float64{"Drama": 2, "Science Fiction": 1},
|
||||
Seen: map[string]bool{}, SeenTitles: map[string]bool{},
|
||||
}
|
||||
evidence := map[string][]PreparedEvidence{
|
||||
"drama": {{
|
||||
ItemID: "arrival", Title: "Arrival",
|
||||
Genres: []string{"Drama", "Science Fiction"},
|
||||
}},
|
||||
}
|
||||
counts := map[string]int{}
|
||||
kinds := map[string]int{}
|
||||
for i := 0; i < 20; i++ {
|
||||
candidate := item(
|
||||
"candidate-"+strconv.Itoa(i), "Candidate", "Movie",
|
||||
[]string{"Drama", "Science Fiction"}, 7,
|
||||
)
|
||||
_, _, kind, _, _ := explainPreparedRecommendation(
|
||||
profile, candidate, compatibilityProfile{}, false, evidence, counts,
|
||||
)
|
||||
kinds[kind]++
|
||||
}
|
||||
if kinds["completed-title"] == 0 || kinds["completed-title"] > 4 {
|
||||
t.Fatalf("completed-title reasons = %d, want 1..4", kinds["completed-title"])
|
||||
}
|
||||
if kinds["genre"] == 0 {
|
||||
t.Fatalf("reason kinds were not mixed: %+v", kinds)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPreparedExplanationRejectsOneBroadGenreAsSpecificEvidence(t *testing.T) {
|
||||
profile := Profile{
|
||||
GenreWeights: map[string]float64{"Drama": 2},
|
||||
Seen: map[string]bool{}, SeenTitles: map[string]bool{},
|
||||
}
|
||||
evidence := map[string][]PreparedEvidence{
|
||||
"drama": {{ItemID: "source", Title: "Source", Genres: []string{"Drama"}}},
|
||||
}
|
||||
_, _, kind, _, _ := explainPreparedRecommendation(
|
||||
profile, item("candidate", "Candidate", "Movie", []string{"Drama"}, 7),
|
||||
compatibilityProfile{}, false, evidence, map[string]int{},
|
||||
)
|
||||
if kind != "genre" {
|
||||
t.Fatalf("one broad shared genre produced %q, want genre", kind)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRecommendationSessionsDiscardPrerolls(t *testing.T) {
|
||||
sessions := []tracearr.Session{
|
||||
{MediaTitle: "PreRoll_Swirls"},
|
||||
{MediaTitle: "A Real Film"},
|
||||
}
|
||||
got := recommendationSessions(sessions)
|
||||
if len(got) != 1 || got[0].MediaTitle != "A Real Film" {
|
||||
t.Fatalf("recommendation sessions = %+v", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildRowsDropsRowsShorterThanTheMinimum(t *testing.T) {
|
||||
source := &fakeSource{
|
||||
itemsByFilter: map[string][]json.RawMessage{
|
||||
@@ -256,6 +586,130 @@ func TestCuratedShowRowsAndItemsAreOrderedByViewingAffinity(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestDefaultCuratedRowsIncludePersonalizedShowGenres(t *testing.T) {
|
||||
engine := NewEngine(&fakeSource{}, slog.New(slog.NewTextHandler(io.Discard, nil)))
|
||||
got := map[string]string{}
|
||||
for _, row := range engine.CuratedRows {
|
||||
got[row.ID] = row.Title
|
||||
}
|
||||
|
||||
for id, title := range map[string]string{
|
||||
"curated:comedy-shows": "Comedy Shows",
|
||||
"curated:drama-shows": "Drama Shows",
|
||||
"curated:horror-shows": "Horror Shows",
|
||||
} {
|
||||
if got[id] != title {
|
||||
t.Fatalf("%s title = %q, want %q", id, got[id], title)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDefaultCuratedRowsIncludeMovieGenresAndStudioFamilies(t *testing.T) {
|
||||
engine := NewEngine(&fakeSource{}, slog.New(slog.NewTextHandler(io.Discard, nil)))
|
||||
got := map[string]CuratedRow{}
|
||||
for _, row := range engine.CuratedRows {
|
||||
got[row.ID] = row
|
||||
}
|
||||
|
||||
for _, id := range []string{
|
||||
"curated:movies:genre:science-fiction",
|
||||
"curated:movies:genre:animation",
|
||||
"curated:movies:studio:pixar",
|
||||
"curated:movies:studio:disney",
|
||||
} {
|
||||
row, ok := got[id]
|
||||
if !ok {
|
||||
t.Fatalf("missing default movie shelf %q", id)
|
||||
}
|
||||
if row.Kind != "movies" || !row.RequireAffinity {
|
||||
t.Fatalf("movie shelf %q = %+v", id, row)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestMovieShelvesRequireAffinityAndRankAStudioBeforeItsGenre(t *testing.T) {
|
||||
source := &fakeSource{itemsByFilter: map[string][]json.RawMessage{
|
||||
"IsPlayed": {
|
||||
json.RawMessage(`{"Id":"watched","Name":"Toy Story","Type":"Movie","Genres":["Comedy"],"Studios":[{"Name":"Pixar Animation Studios"}]}`),
|
||||
},
|
||||
}}
|
||||
engine := testEngine(source)
|
||||
engine.Library = &fakeCuratedLibrary{byGenre: map[string][]json.RawMessage{
|
||||
"Comedy": {
|
||||
raw("comedy-1", "Comedy One", "Movie", "Comedy"),
|
||||
raw("comedy-2", "Comedy Two", "Movie", "Comedy"),
|
||||
},
|
||||
"Drama": {
|
||||
raw("drama-1", "Drama One", "Movie", "Drama"),
|
||||
raw("drama-2", "Drama Two", "Movie", "Drama"),
|
||||
},
|
||||
"studio:Pixar": {
|
||||
raw("pixar-1", "Pixar One", "Movie", "Animation"),
|
||||
raw("pixar-2", "Pixar Two", "Movie", "Animation"),
|
||||
},
|
||||
}}
|
||||
engine.CuratedRows = []CuratedRow{
|
||||
movieGenreRow("comedy", "Comedy"),
|
||||
movieGenreRow("drama", "Drama"),
|
||||
movieStudioRow("pixar", "Pixar", "Pixar", "Pixar Animation Studios"),
|
||||
}
|
||||
|
||||
rows, err := engine.BuildRows(context.Background(), emby.Credentials{UserID: "u1"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
curated := make([]Row, 0, len(rows))
|
||||
for _, row := range rows {
|
||||
if strings.HasPrefix(row.ID, "curated:movies:") {
|
||||
curated = append(curated, row)
|
||||
}
|
||||
}
|
||||
if len(curated) != 2 {
|
||||
t.Fatalf("movie shelves = %+v, want Pixar and Comedy only", rowTitles(curated))
|
||||
}
|
||||
if curated[0].ID != "curated:movies:studio:pixar" ||
|
||||
curated[1].ID != "curated:movies:genre:comedy" {
|
||||
t.Fatalf("movie shelf order = %+v", rowTitles(curated))
|
||||
}
|
||||
}
|
||||
|
||||
func TestCuratedRowsDoNotRepeatCardsAcrossGenres(t *testing.T) {
|
||||
source := &fakeSource{itemsByFilter: map[string][]json.RawMessage{
|
||||
"IsPlayed": {raw("history", "Funny", "Episode", "Comedy")},
|
||||
}}
|
||||
engine := testEngine(source)
|
||||
engine.MinRowItems = 1
|
||||
engine.RowSize = 3
|
||||
engine.Library = &fakeCuratedLibrary{byGenre: map[string][]json.RawMessage{
|
||||
"Comedy": {
|
||||
raw("shared", "Shared Show", "Series", "Comedy", "Drama"),
|
||||
raw("comedy", "Comedy Only", "Series", "Comedy"),
|
||||
},
|
||||
"Drama": {
|
||||
raw("shared", "Shared Show", "Series", "Comedy", "Drama"),
|
||||
raw("drama", "Drama Only", "Series", "Drama"),
|
||||
},
|
||||
}}
|
||||
engine.CuratedRows = []CuratedRow{
|
||||
{ID: "comedy", Title: "Comedy Shows", Kind: "shows", ItemTypes: []string{"Series"}, Genres: []string{"Comedy"}},
|
||||
{ID: "drama", Title: "Drama Shows", Kind: "shows", ItemTypes: []string{"Series"}, Genres: []string{"Drama"}},
|
||||
}
|
||||
|
||||
rows, err := engine.BuildRows(context.Background(), emby.Credentials{UserID: "u1"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
ids := map[string]int{}
|
||||
for _, row := range rows {
|
||||
for _, item := range Decode(row.Items) {
|
||||
ids[item.ID]++
|
||||
}
|
||||
}
|
||||
if ids["shared"] != 1 {
|
||||
t.Fatalf("shared card appeared %d times across curated rows", ids["shared"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestCuratedRowsFallBackToRatingForANewUser(t *testing.T) {
|
||||
source := &fakeSource{itemsByFilter: map[string][]json.RawMessage{}}
|
||||
engine := testEngine(source)
|
||||
|
||||
Reference in New Issue
Block a user