128 lines
3.6 KiB
Go
128 lines
3.6 KiB
Go
package torrentclaw
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestCollections(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/api/v1/collections" {
|
|
t.Errorf("path = %q, want /api/v1/collections", r.URL.Path)
|
|
}
|
|
if got := r.URL.Query().Get("limit"); got != "12" {
|
|
t.Errorf("limit = %q, want 12", got)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(CollectionListResponse{
|
|
Items: []CollectionListItem{
|
|
{ID: 1, TMDbID: 10, Name: "Star Wars", MovieCount: 9, TotalSeeders: 5000, PartCount: 9},
|
|
},
|
|
Total: 50,
|
|
Page: 1,
|
|
PageSize: 12,
|
|
})
|
|
}))
|
|
defer srv.Close()
|
|
|
|
c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
|
|
resp, err := c.Collections(context.Background(), CollectionListParams{Limit: 12})
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if resp.Total != 50 {
|
|
t.Errorf("Total = %d, want 50", resp.Total)
|
|
}
|
|
if len(resp.Items) != 1 {
|
|
t.Fatalf("len(Items) = %d, want 1", len(resp.Items))
|
|
}
|
|
if resp.Items[0].Name != "Star Wars" {
|
|
t.Errorf("Name = %q, want Star Wars", resp.Items[0].Name)
|
|
}
|
|
if resp.Items[0].MovieCount != 9 {
|
|
t.Errorf("MovieCount = %d, want 9", resp.Items[0].MovieCount)
|
|
}
|
|
}
|
|
|
|
func TestCollectionByID(t *testing.T) {
|
|
overview := "The epic saga"
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/api/v1/collections/10" {
|
|
t.Errorf("path = %q, want /api/v1/collections/10", r.URL.Path)
|
|
}
|
|
if got := r.URL.Query().Get("locale"); got != "es" {
|
|
t.Errorf("locale = %q, want es", got)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(CollectionDetail{
|
|
ID: 1,
|
|
TMDbID: 10,
|
|
Name: "Star Wars",
|
|
MovieCount: 9,
|
|
TotalSeeders: 5000,
|
|
PartCount: 9,
|
|
Overview: &overview,
|
|
Movies: []PopularItem{
|
|
{ID: 100, Title: "A New Hope", ContentType: "movie", MaxSeeders: 800},
|
|
},
|
|
})
|
|
}))
|
|
defer srv.Close()
|
|
|
|
c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
|
|
resp, err := c.CollectionByID(context.Background(), 10, "es")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if resp.Name != "Star Wars" {
|
|
t.Errorf("Name = %q, want Star Wars", resp.Name)
|
|
}
|
|
if resp.Overview == nil || *resp.Overview != "The epic saga" {
|
|
t.Errorf("Overview = %v", resp.Overview)
|
|
}
|
|
if len(resp.Movies) != 1 {
|
|
t.Fatalf("len(Movies) = %d, want 1", len(resp.Movies))
|
|
}
|
|
if resp.Movies[0].Title != "A New Hope" {
|
|
t.Errorf("Title = %q, want A New Hope", resp.Movies[0].Title)
|
|
}
|
|
}
|
|
|
|
func TestCollectionByID_NotFound(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte(`{"error":"Collection not found"}`))
|
|
}))
|
|
defer srv.Close()
|
|
|
|
c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
|
|
_, err := c.CollectionByID(context.Background(), 999, "")
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
apiErr, ok := err.(*APIError)
|
|
if !ok {
|
|
t.Fatalf("expected *APIError, got %T", err)
|
|
}
|
|
if !apiErr.IsNotFound() {
|
|
t.Errorf("expected 404, got %d", apiErr.StatusCode)
|
|
}
|
|
}
|
|
|
|
func TestCollections_ServerError(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer srv.Close()
|
|
|
|
c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
|
|
_, err := c.Collections(context.Background(), CollectionListParams{})
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
}
|