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) } }