feat: implement TorrentClaw Go API client v0.1.0

stats_test.go

@@ -0,0 +1,92 @@
+package torrentclaw
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestStats_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.Stats(context.Background())
+	if err == nil {
+		t.Fatal("expected error")
+	}
+	apiErr, ok := err.(*APIError)
+	if !ok {
+		t.Fatalf("expected *APIError, got %T", err)
+	}
+	if apiErr.StatusCode != 500 {
+		t.Errorf("StatusCode = %d, want 500", apiErr.StatusCode)
+	}
+}
+
+func TestStats_EmptyIngestions(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		json.NewEncoder(w).Encode(StatsResponse{
+			Content:          ContentStats{Movies: 100, Shows: 50, TMDbEnriched: 80},
+			Torrents:         TorrentStats{Total: 500, WithSeeders: 300, BySource: map[string]int{}},
+			RecentIngestions: []IngestionRecord{},
+		})
+	}))
+	defer srv.Close()
+
+	c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
+	resp, err := c.Stats(context.Background())
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp.Content.TMDbEnriched != 80 {
+		t.Errorf("TMDbEnriched = %d, want 80", resp.Content.TMDbEnriched)
+	}
+	if resp.Torrents.WithSeeders != 300 {
+		t.Errorf("WithSeeders = %d, want 300", resp.Torrents.WithSeeders)
+	}
+	if len(resp.RecentIngestions) != 0 {
+		t.Errorf("len(RecentIngestions) = %d, want 0", len(resp.RecentIngestions))
+	}
+}
+
+func TestStats_WithCompletedAt(t *testing.T) {
+	completedAt := "2025-01-15T10:05:00Z"
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		json.NewEncoder(w).Encode(StatsResponse{
+			Content:  ContentStats{},
+			Torrents: TorrentStats{BySource: map[string]int{}},
+			RecentIngestions: []IngestionRecord{
+				{
+					Source:      "yts",
+					Status:      "completed",
+					StartedAt:   "2025-01-15T10:00:00Z",
+					CompletedAt: &completedAt,
+					Fetched:     100,
+					New:         10,
+					Updated:     5,
+				},
+			},
+		})
+	}))
+	defer srv.Close()
+
+	c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
+	resp, err := c.Stats(context.Background())
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if len(resp.RecentIngestions) != 1 {
+		t.Fatalf("len(RecentIngestions) = %d, want 1", len(resp.RecentIngestions))
+	}
+	rec := resp.RecentIngestions[0]
+	if rec.CompletedAt == nil || *rec.CompletedAt != completedAt {
+		t.Errorf("CompletedAt = %v, want %q", rec.CompletedAt, completedAt)
+	}
+}