jellybit/internal/naming/effective_test.go

package naming

import (
	"encoding/json"
	"testing"

	"git.vakhrushev.me/av/jellybit/internal/recognize"
)

func mustJSON(t *testing.T, f 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}
}

// EffectiveFields: слоистое разрешение полей ярлыка (план → контекст) и сводка
// сезонов из плана/контекста. Проверяем и итоговый Label (совпадение с прежним
// display_name), и отдельно очищенное поле Director (его печатает шаблон).
func TestEffectiveFields(t *testing.T) {
	cases := []struct {
		name    string
		parsed  Fields // пустой Title → parsed_context не задаётся
		plan    recognize.Plan
		want    string // ожидаемый Label()
		wantDir string // ожидаемый Director
	}{
		{
			name:    "матч даёт режиссёра и год",
			plan:    recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Director: "Дени Вильнёв", Year: 2024},
			want:    "Дюна (Дени Вильнёв, 2024)",
			wantDir: "Дени Вильнёв",
		},
		{
			name:    "режиссёр из контекста переживает распознавание без матча",
			parsed:  Fields{Type: "movie", Title: "Дюна", Director: "Дени Вильнёв", Year: 2021},
			plan:    recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Year: 2024}, // без director
			want:    "Дюна (Дени Вильнёв, 2024)",                                           // год из плана, режиссёр из контекста
			wantDir: "Дени Вильнёв",
		},
		{
			name:    "режиссёр матча бьёт контекстного",
			parsed:  Fields{Type: "movie", Title: "Дюна", Director: "Кто-то из контекста"},
			plan:    recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Director: "Дени Вильнёв", Year: 2024},
			want:    "Дюна (Дени Вильнёв, 2024)",
			wantDir: "Дени Вильнёв",
		},
		{
			name:    "год из контекста, когда плана нет",
			parsed:  Fields{Type: "movie", Title: "Брат", Year: 1997},
			plan:    recognize.Plan{Type: recognize.MediaMovie, Title: "Брат"}, // year 0, без режиссёра
			want:    "Брат (1997)",
			wantDir: "",
		},
		{
			name: "сериал: сводка сезонов из плана",
			plan: recognize.Plan{Type: recognize.MediaSeries, Title: "Фарго",
				Files: []recognize.PlanFile{episode(1), episode(2), episode(3)}},
			want:    "Фарго. Сезоны 1–3",
			wantDir: "",
		},
		{
			name:    "сериал: сезон из контекста, когда в плане нет эпизодов",
			parsed:  Fields{Type: "series", Title: "Сёгун", Season: ptr(2)},
			plan:    recognize.Plan{Type: recognize.MediaSeries, Title: "Сёгун", Year: 2024},
			want:    "Сёгун (2024). Сезон 2",
			wantDir: "",
		},
		{
			name:    "режиссёр из управляющих символов и лишних пробелов очищается",
			plan:    recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Director: "  Дени\tВильнёв\n ", Year: 2024},
			want:    "Дюна (Дени Вильнёв, 2024)",
			wantDir: "Дени Вильнёв",
		},
		{
			name:    "режиссёр только из управляющих символов даёт пустое поле и выпадает из ярлыка",
			plan:    recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Director: "\x01\x02", Year: 2024},
			want:    "Дюна (2024)",
			wantDir: "",
		},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			parsed := ""
			if c.parsed.Title != "" {
				parsed = mustJSON(t, c.parsed)
			}
			ef := EffectiveFields(parsed, c.plan)
			if got := ef.Label(); got != c.want {
				t.Errorf("Label() = %q, want %q", got, c.want)
			}
			if ef.Director != c.wantDir {
				t.Errorf("Director = %q, want %q", ef.Director, c.wantDir)
			}
		})
	}
}

// Битый parsed_context не валит разрешение (best-effort): используется только план.
func TestEffectiveFieldsBrokenParsedContext(t *testing.T) {
	plan := recognize.Plan{Type: recognize.MediaMovie, Title: "Дюна", Year: 2024}
	if got := EffectiveFields("{не json", plan).Label(); got != "Дюна (2024)" {
		t.Errorf("Label() = %q, want «Дюна (2024)»", got)
	}
}