package worker import ( "encoding/json" "testing" "git.vakhrushev.me/av/jellybit/internal/naming" "git.vakhrushev.me/av/jellybit/internal/recognize" "git.vakhrushev.me/av/jellybit/internal/store" ) func parsedContext(t *testing.T, f naming.Fields) string { t.Helper() b, err := json.Marshal(f) if err != nil { t.Fatalf("marshal fields: %v", err) } return string(b) } func episode(season int) recognize.PlanFile { s := season return recognize.PlanFile{Role: recognize.RoleEpisode, Season: &s} } // effectiveDisplayName: слоистое разрешение полей ярлыка (план → контекст) и // сводка сезонов из плана/контекста. func TestEffectiveDisplayName(t *testing.T) { cases := []struct { name string parsed naming.Fields plan recognize.Plan want string }{ { name: "матч даёт режиссёра и год", plan: recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Director: "Дени Вильнёв", Year: 2024}, want: "Дюна (Дени Вильнёв, 2024)", }, { name: "режиссёр из контекста переживает распознавание без матча", parsed: naming.Fields{Type: "movie", Title: "Дюна", Director: "Дени Вильнёв", Year: 2021}, plan: recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Year: 2024}, // без director want: "Дюна (Дени Вильнёв, 2024)", // год из плана, режиссёр из контекста }, { name: "режиссёр матча бьёт контекстного", parsed: naming.Fields{Type: "movie", Title: "Дюна", Director: "Кто-то из контекста"}, plan: recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Director: "Дени Вильнёв", Year: 2024}, want: "Дюна (Дени Вильнёв, 2024)", }, { name: "год из контекста, когда плана нет", parsed: naming.Fields{Type: "movie", Title: "Брат", Year: 1997}, plan: recognize.Plan{Type: recognize.MediaMovie, Title: "Брат"}, // year 0 want: "Брат (1997)", }, { name: "сериал: сводка сезонов из плана", plan: recognize.Plan{Type: recognize.MediaSeries, Title: "Фарго", Files: []recognize.PlanFile{episode(1), episode(2), episode(3)}}, want: "Фарго. Сезоны 1–3", }, { name: "сериал: сезон из контекста, когда в плане нет эпизодов", parsed: naming.Fields{Type: "series", Title: "Сёгун", Season: ptrInt(2)}, plan: recognize.Plan{Type: recognize.MediaSeries, Title: "Сёгун", Year: 2024}, want: "Сёгун (2024). Сезон 2", }, } for _, c := range cases { t.Run(c.name, func(t *testing.T) { d := store.Download{} if c.parsed.Title != "" { d.ParsedContext = parsedContext(t, c.parsed) } if got := effectiveDisplayName(d, c.plan); got != c.want { t.Errorf("effectiveDisplayName = %q, want %q", got, c.want) } }) } } // Битый parsed_context не валит рендер (best-effort): используется только план. func TestEffectiveDisplayNameBrokenParsedContext(t *testing.T) { d := store.Download{ParsedContext: "{не json"} plan := recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Year: 2024} if got := effectiveDisplayName(d, plan); got != "Дюна (2024)" { t.Errorf("effectiveDisplayName = %q, want «Дюна (2024)»", got) } } func ptrInt(v int) *int { return &v }