package watchorder import ( "context" "errors" "io" "net/http" "strings" "testing" "time" ) func mockResponse(status int, headers map[string]string, body string) *http.Response { h := make(http.Header, len(headers)) for k, v := range headers { h.Set(k, v) } return &http.Response{ StatusCode: status, Header: h, Body: io.NopCloser(strings.NewReader(body)), } } func testHTMLWithMetadata() string { return `

TV Movie

Naruto Movie 1 Naruto the Movie 1

` } func testHTMLEmptyRows() string { return `

TV Movie

` } func TestFetchWatchOrder_OutputShape(t *testing.T) { client := &http.Client{ Timeout: time.Second, Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { if request.URL.RawQuery == "/tools/watch_order/id/442" { return mockResponse(http.StatusOK, map[string]string{"Content-Type": "text/html; charset=utf-8"}, testHTMLWithMetadata()), nil } return mockResponse(http.StatusNotFound, nil, "not found"), nil }), } url := "https://chiaki.site/?/tools/watch_order/id/442" result, err := FetchWatchOrder(context.Background(), client, url) if err != nil { t.Fatalf("expected no error, got %v", err) } if result.ID != 442 { t.Fatalf("expected root id 442, got %d", result.ID) } if len(result.WatchOrder) != 1 { t.Fatalf("expected 1 watch_order entry, got %d", len(result.WatchOrder)) } entry := result.WatchOrder[0] if entry.ID != 442 { t.Fatalf("expected entry id 442, got %d", entry.ID) } if entry.Type != "Movie" { t.Fatalf("expected type Movie, got %q", entry.Type) } if entry.Title != "Naruto Movie 1" { t.Fatalf("expected title Naruto Movie 1, got %q", entry.Title) } if entry.TitleAlt != "Naruto the Movie 1" { t.Fatalf("expected title_alt Naruto the Movie 1, got %q", entry.TitleAlt) } } func TestFetchWatchOrder_NoRowsReturnsEmpty(t *testing.T) { client := &http.Client{ Timeout: time.Second, Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { if request.URL.RawQuery == "/tools/watch_order/id/1535" { return mockResponse(http.StatusOK, map[string]string{"Content-Type": "text/html; charset=utf-8"}, testHTMLEmptyRows()), nil } return mockResponse(http.StatusNotFound, nil, "not found"), nil }), } url := "https://chiaki.site/?/tools/watch_order/id/1535" result, err := FetchWatchOrder(context.Background(), client, url) if err != nil { t.Fatalf("expected no error, got %v", err) } if result.ID != 1535 { t.Fatalf("expected root id 1535, got %d", result.ID) } if len(result.WatchOrder) != 0 { t.Fatalf("expected no entries, got %d", len(result.WatchOrder)) } } func TestFetchWatchOrder_MissingMarkupFallsBackToProxy(t *testing.T) { proxyPayload := `Title: Jujutsu Kaisen / Watch Order URL Source: https://chiaki.site/?/tools/watch_order/id/40748 Markdown Content: Jujutsu Kaisen Oct 3, 2020 – Mar 27, 2021 | TV | 24ep × 23min. | ★8.51 | [](https://myanimelist.net/anime/40748) Jujutsu Kaisen 0 Movie Jujutsu Kaisen 0 Dec 24, 2021 | Movie | 1ep × 1hr. 44min. | ★8.36 | [](https://myanimelist.net/anime/48561) ` testClient := &http.Client{ Timeout: time.Second, Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { switch request.URL.Host { case "chiaki.site": return mockResponse(http.StatusForbidden, map[string]string{"Content-Type": "text/html; charset=utf-8"}, "blocked"), nil case "r.jina.ai": // Proxy response is plain text/markdown. return mockResponse(http.StatusOK, map[string]string{"Content-Type": "text/plain; charset=utf-8"}, proxyPayload), nil default: return mockResponse(http.StatusNotFound, nil, "not found"), nil } }), } result, err := FetchWatchOrder(context.Background(), testClient, "https://chiaki.site/?/tools/watch_order/id/40748") if err != nil { t.Fatalf("expected no error, got %v", err) } if len(result.WatchOrder) != 2 { t.Fatalf("expected 2 proxy entries, got %d", len(result.WatchOrder)) } if result.WatchOrder[0].ID != 40748 || result.WatchOrder[0].Type != "TV" { t.Fatalf("unexpected first entry: %+v", result.WatchOrder[0]) } if result.WatchOrder[1].ID != 48561 || result.WatchOrder[1].Type != "Movie" { t.Fatalf("unexpected second entry: %+v", result.WatchOrder[1]) } } func TestFetchWatchOrder_HTTPStatusErrorIncludesContext(t *testing.T) { client := &http.Client{ Timeout: time.Second, Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) { return mockResponse(http.StatusForbidden, map[string]string{ "Server": "cloudflare", "CF-Ray": "abc123", "Content-Type": "text/html; charset=utf-8", }, "access denied"), nil }), } url := "https://chiaki.site/?/tools/watch_order/id/1" _, err := fetchDocument(context.Background(), client, url) if err == nil { t.Fatalf("expected error, got nil") } var statusError *HTTPStatusError if !errors.As(err, &statusError) { t.Fatalf("expected HTTPStatusError, got %T", err) } if statusError.StatusCode != http.StatusForbidden { t.Fatalf("expected 403, got %d", statusError.StatusCode) } if statusError.CFRay != "abc123" { t.Fatalf("expected cf-ray abc123, got %q", statusError.CFRay) } if !strings.Contains(statusError.BodyPreview, "access denied") { t.Fatalf("expected body preview to include access denied, got %q", statusError.BodyPreview) } } type roundTripFunc func(*http.Request) (*http.Response, error) func (f roundTripFunc) RoundTrip(request *http.Request) (*http.Response, error) { return f(request) }