package watchorder import ( "context" "errors" "net/http" "net/http/httptest" "strings" "testing" "time" ) func testServer(body string) *httptest.Server { handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "text/html; charset=utf-8") _, _ = w.Write([]byte(body)) }) return httptest.NewServer(handler) } func testHTMLWithMetadata() string { return `
Naruto Movie 1 Naruto the Movie 1
` } func testHTMLEmptyRows() string { return `
` } func testHTMLWithoutWatchOrderTable() string { return `

challenge page

` } func TestFetchWatchOrder_OutputShape(t *testing.T) { server := testServer(testHTMLWithMetadata()) defer server.Close() url := server.URL + "/?/tools/watch_order/id/442" result, err := FetchWatchOrder(context.Background(), &http.Client{Timeout: time.Second}, 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) { server := testServer(testHTMLEmptyRows()) defer server.Close() url := server.URL + "/?/tools/watch_order/id/1535" result, err := FetchWatchOrder(context.Background(), &http.Client{Timeout: time.Second}, 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_MissingMarkupReturnsError(t *testing.T) { server := testServer(testHTMLWithoutWatchOrderTable()) defer server.Close() url := server.URL + "/?/tools/watch_order/id/1535" _, err := FetchWatchOrder(context.Background(), &http.Client{Timeout: time.Second}, url) if !errors.Is(err, ErrWatchOrderMarkupNotFound) { t.Fatalf("expected ErrWatchOrderMarkupNotFound, got %v", err) } } func TestFetchWatchOrder_HTTPStatusErrorIncludesContext(t *testing.T) { server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "cloudflare") w.Header().Set("CF-Ray", "abc123") w.Header().Set("Content-Type", "text/html; charset=utf-8") w.WriteHeader(http.StatusForbidden) _, _ = w.Write([]byte("access denied")) })) defer server.Close() url := server.URL + "/?/tools/watch_order/id/1" _, err := FetchWatchOrder(context.Background(), &http.Client{Timeout: time.Second}, 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) } }