package watchorder import ( "context" "errors" "net/http" "net/http/httptest" "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) } }