MeBox/internal/service/introdb_test.go

package service

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"go.uber.org/zap"
)

// 这两段响应是从 api.theintrodb.org/v3/media 实测抓下来的原文，
// 用来锁住 null 语义：start_ms: null = 从片头开始，end_ms: null = 一直到片尾。
const (
	introDBTVPayload    = `{"tmdb_id":1396,"type":"tv","season":1,"episode":1,"intro":[{"start_ms":228664,"end_ms":246143}],"credits":[{"start_ms":3431000,"end_ms":null}]}`
	introDBMoviePayload = `{"tmdb_id":27205,"type":"movie","intro":[{"start_ms":null,"end_ms":38000}]}`
)

func TestParseIntroDBResponseResolvesNullBounds(t *testing.T) {
	spans, err := parseIntroDBResponse([]byte(introDBTVPayload))
	if err != nil {
		t.Fatalf("parse: %v", err)
	}
	if len(spans) != 2 {
		t.Fatalf("spans = %d, want 2 (%#v)", len(spans), spans)
	}
	if spans[0].Kind != "intro" || spans[0].StartMs != 228_664 || spans[0].EndMs != 246_143 {
		t.Fatalf("intro span = %#v", spans[0])
	}
	// end_ms: null 表示一直到片尾，落成 0 由客户端结合时长补齐。
	if spans[1].Kind != "credits" || spans[1].StartMs != 3_431_000 || spans[1].EndMs != 0 {
		t.Fatalf("credits span = %#v", spans[1])
	}

	movie, err := parseIntroDBResponse([]byte(introDBMoviePayload))
	if err != nil {
		t.Fatalf("parse movie: %v", err)
	}
	if len(movie) != 1 {
		t.Fatalf("movie spans = %d, want 1", len(movie))
	}
	// start_ms: null = 从片头开始。
	if movie[0].StartMs != 0 || movie[0].EndMs != 38_000 {
		t.Fatalf("movie intro span = %#v", movie[0])
	}
}

func TestParseIntroDBResponseDropsEmptyRanges(t *testing.T) {
	body := `{"tmdb_id":1,"type":"movie",
		"intro":[{"start_ms":5000,"end_ms":5000},{"start_ms":9000,"end_ms":8000},{"start_ms":1000,"end_ms":2000}],
		"recap":[],"credits":[],"preview":[]}`
	spans, err := parseIntroDBResponse([]byte(body))
	if err != nil {
		t.Fatalf("parse: %v", err)
	}
	// 只有 end > start 的区间是可跳过的；end == 0（到片尾）是合法值，此处不涉及。
	if len(spans) != 1 || spans[0].StartMs != 1_000 || spans[0].EndMs != 2_000 {
		t.Fatalf("spans = %#v, want only the 1000-2000 range", spans)
	}
}

func TestParseIntroDBResponseOrdersByType(t *testing.T) {
	body := `{"tmdb_id":1,"type":"tv","credits":[{"start_ms":900,"end_ms":1000}],
		"intro":[{"start_ms":100,"end_ms":200}],"recap":[{"start_ms":50,"end_ms":60}]}`
	spans, err := parseIntroDBResponse([]byte(body))
	if err != nil {
		t.Fatalf("parse: %v", err)
	}
	want := []string{"intro", "recap", "credits"}
	if len(spans) != len(want) {
		t.Fatalf("spans = %#v, want %d", spans, len(want))
	}
	for i, kind := range want {
		if spans[i].Kind != kind {
			t.Fatalf("span[%d].kind = %q, want %q", i, spans[i].Kind, kind)
		}
	}
}

func TestIntroDBMediaURLOnlyAddsSeasonEpisodeForTV(t *testing.T) {
	svc := NewIntroDBService(zap.NewNop())
	if got, want := svc.mediaURL(1396, 1, 1),
		"https://api.theintrodb.org/v3/media?tmdb_id=1396&season=1&episode=1"; got != want {
		t.Fatalf("tv url = %q, want %q", got, want)
	}
	// 电影（season/episode 为 0）不能带季集参数，否则会被当成剧集查不到。
	if got, want := svc.mediaURL(27205, 0, 0),
		"https://api.theintrodb.org/v3/media?tmdb_id=27205"; got != want {
		t.Fatalf("movie url = %q, want %q", got, want)
	}
}

func TestIntroDBFetchTreatsNotFoundAsNoData(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		w.WriteHeader(http.StatusNotFound)
	}))
	defer server.Close()

	svc := NewIntroDBService(zap.NewNop()).SetBaseURL(server.URL)
	spans, err := svc.Fetch(t.Context(), 999_999, 1, 1)
	if err != nil {
		t.Fatalf("404 must not be an error, got %v", err)
	}
	if len(spans) != 0 {
		t.Fatalf("spans = %#v, want none", spans)
	}
}

func TestIntroDBFetchReportsUnexpectedStatus(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		w.WriteHeader(http.StatusInternalServerError)
	}))
	defer server.Close()

	svc := NewIntroDBService(zap.NewNop()).SetBaseURL(server.URL)
	if _, err := svc.Fetch(t.Context(), 1, 0, 0); err == nil {
		t.Fatal("500 should surface as an error so the caller can keep its cache")
	}
}

func TestIntroDBFetchSkipsRequestWithoutTMDbID(t *testing.T) {
	calls := 0
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		calls++
		_, _ = w.Write([]byte(introDBMoviePayload))
	}))
	defer server.Close()

	svc := NewIntroDBService(zap.NewNop()).SetBaseURL(server.URL)
	spans, err := svc.Fetch(t.Context(), 0, 0, 0)
	if err != nil {
		t.Fatalf("fetch: %v", err)
	}
	if len(spans) != 0 || calls != 0 {
		t.Fatalf("spans = %#v calls = %d, want no request without a tmdb id", spans, calls)
	}
}

func TestIntroDBFetchParsesBody(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if got := r.URL.Query().Get("tmdb_id"); got != "1396" {
			t.Errorf("tmdb_id = %q, want 1396", got)
		}
		if got := r.URL.Query().Get("season"); got != "1" {
			t.Errorf("season = %q, want 1", got)
		}
		if got := r.URL.Query().Get("episode"); got != "1" {
			t.Errorf("episode = %q, want 1", got)
		}
		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(introDBTVPayload))
	}))
	defer server.Close()

	svc := NewIntroDBService(zap.NewNop()).SetBaseURL(server.URL)
	spans, err := svc.Fetch(t.Context(), 1396, 1, 1)
	if err != nil {
		t.Fatalf("fetch: %v", err)
	}
	if len(spans) != 2 || spans[0].Kind != "intro" {
		t.Fatalf("spans = %#v", spans)
	}
}