package app
import (
"context"
"encoding/base64"
"errors"
"fmt"
"net/http"
"strings"
"sync/atomic"
"testing"
"time"
)
func parityRankingHTML(board rankingBoard, jsonLD bool) string {
header := ``
if jsonLD {
return header + ``
}
return header + `- 1
合成榜单剧
12.3万热度
`
}
func TestParityHongguoRankingTextFallbackAndIdentity(t *testing.T) {
board, _ := findRankingBoard("hongguo-hot")
for _, jsonLD := range []bool{false, true} {
body := parityRankingHTML(board, jsonLD)
page, err := parseHongguoRanking(body, board, 1)
if err != nil || len(page.Items) != 1 || page.Items[0].Drama.ID != "hongguo:700001" || page.Items[0].Rank != 1 || page.HasMore {
t.Fatalf("text fallback failed (JSON-LD=%v): %+v %v", jsonLD, page, err)
}
if _, err := parseHongguoRanking(strings.ReplaceAll(body, board.path, "hot-ai-drama"), board, 1); err == nil {
t.Fatal("a different ranking board was accepted")
}
if _, err := parseHongguoRanking(body, board, 2); err == nil {
t.Fatal("a different page was accepted")
}
if _, err := parseHongguoRanking(strings.ReplaceAll(body, "700001", "invalid"), board, 1); err == nil {
t.Fatal("invalid drama identity was accepted")
}
}
}
func TestParityHongguoRankingRetryIsBoundedAndBypassesCache(t *testing.T) {
board, _ := findRankingBoard("hongguo-hot")
var calls atomic.Int32
d := rankingTestDownloader(t, func(request *http.Request) (*http.Response, error) {
count := calls.Add(1)
if request.URL.RawQuery != "" {
t.Error("first page is not canonical")
}
if count > 1 && (request.Header.Get("Cache-Control") != "no-cache" || request.Header.Get("Pragma") != "no-cache") {
t.Error("retry reused the incomplete cached page")
}
return rankingHTTPResponse(request, 200, ``), nil
})
d.cfg.Retries = 10
_, err := d.fetchHongguoRankingPage(context.Background(), board, 1)
if !errors.Is(err, errHongguoRankingIncomplete) || calls.Load() != 3 {
t.Fatalf("retry budget changed: %d %v", calls.Load(), err)
}
}
func TestParityHongguoSearchDeadlineKeepsSharedPartialResult(t *testing.T) {
d := rankingTestDownloader(t, func(request *http.Request) (*http.Response, error) {
return nil, fmt.Errorf("unexpected network request: %s", request.URL.Path)
})
pending := &hongguoSearchCall{done: make(chan struct{}), changed: make(chan struct{}), snapshot: hongguoSearchEntry{Dramas: []Drama{{ID: "hongguo:700001", Title: "合成结果"}}, Total: 1}}
ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(-time.Second))
defer cancel()
result, err := d.waitHongguoSearch(ctx, pending, nil)
if err != nil || len(result.Dramas) != 1 || !result.Limited || result.Warning == "" {
t.Fatalf("deadline discarded partial result: %+v %v", result, err)
}
canceled, stop := context.WithCancel(context.Background())
stop()
if _, err := d.waitHongguoSearch(canceled, pending, nil); !errors.Is(err, context.Canceled) {
t.Fatal("explicit cancellation became a successful search")
}
}
func TestParityHongguoSearchSortKeepsFamiliesAndTokenMatches(t *testing.T) {
if hongguoTitleSearchRank("合成 第五季 故事", "合成 故事") >= hongguoTitleSearchRank("无关内容", "合成 故事") {
t.Fatal("separate title terms did not outrank unrelated results")
}
rows := []Drama{{Title: "合成故事第十季"}, {Title: "合成番外第二季"}, {Title: "合成故事第二季"}, {Title: "合成故事第一季"}}
sortHongguoSearchDramas(rows, "合成")
want := []string{"合成故事第一季", "合成番外第二季", "合成故事第二季", "合成故事第十季"}
for index, drama := range rows {
if drama.Title != want[index] {
t.Fatalf("cross-family order changed: %+v", rows)
}
}
}
func TestParityHongguoMediaRetainsBackupsWithoutInventingPrimary(t *testing.T) {
primary := "https://media.example.test/main.mp4?token=a%2Fb"
backup := "https://backup.example.test/backup.mp4"
model := map[string]any{"video_list": []any{map[string]any{
"main_url": "invalid", "backup_url": base64.StdEncoding.EncodeToString([]byte(primary)), "url_list": []any{primary, backup},
"video_meta": map[string]any{"codec_type": "h264", "vheight": "720"},
}}}
media, err := selectHongguoAppMedia(model)
if err != nil || media.URL != primary || len(media.Variants) != 2 || media.Variants[1].URL != backup || media.Quality != 720 {
t.Fatalf("backup media fields were lost: %+v %v", media, err)
}
}