feat: implement TorrentClaw Go API client v0.1.0

torrent_test.go

@@ -0,0 +1,108 @@
+package torrentclaw
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+func TestTorrentDownloadURL_CustomBase(t *testing.T) {
+	c := NewClient(WithBaseURL("https://custom.example.com"))
+	got := c.TorrentDownloadURL("deadbeef")
+	want := "https://custom.example.com/api/v1/torrent/deadbeef"
+	if got != want {
+		t.Errorf("TorrentDownloadURL = %q, want %q", got, want)
+	}
+}
+
+func TestGetTorrentFile_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.GetTorrentFile(context.Background(), "abc123")
+	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 TestGetTorrentFile_RetryOn503(t *testing.T) {
+	attempts := 0
+	torrentData := []byte{0xDE, 0xAD, 0xBE, 0xEF}
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		attempts++
+		if attempts < 3 {
+			w.WriteHeader(http.StatusServiceUnavailable)
+			return
+		}
+		w.Header().Set("Content-Type", "application/x-bittorrent")
+		w.Write(torrentData)
+	}))
+	defer srv.Close()
+
+	c := NewClient(
+		WithBaseURL(srv.URL),
+		WithRetry(3, 1*time.Millisecond, 10*time.Millisecond),
+	)
+	data, err := c.GetTorrentFile(context.Background(), "abc123")
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if len(data) != len(torrentData) {
+		t.Errorf("len(data) = %d, want %d", len(data), len(torrentData))
+	}
+	if attempts != 3 {
+		t.Errorf("attempts = %d, want 3", attempts)
+	}
+}
+
+func TestGetTorrentFile_RetryExhausted(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusBadGateway)
+	}))
+	defer srv.Close()
+
+	c := NewClient(
+		WithBaseURL(srv.URL),
+		WithRetry(1, 1*time.Millisecond, 10*time.Millisecond),
+	)
+	_, err := c.GetTorrentFile(context.Background(), "abc123")
+	if err == nil {
+		t.Fatal("expected error after retries exhausted")
+	}
+	apiErr, ok := err.(*APIError)
+	if !ok {
+		t.Fatalf("expected *APIError, got %T", err)
+	}
+	if apiErr.StatusCode != 502 {
+		t.Errorf("StatusCode = %d, want 502", apiErr.StatusCode)
+	}
+}
+
+func TestGetTorrentFile_ContextCanceled(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		time.Sleep(1 * time.Second)
+		w.WriteHeader(http.StatusOK)
+	}))
+	defer srv.Close()
+
+	c := NewClient(WithBaseURL(srv.URL), WithRetry(0, 0, 0))
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	_, err := c.GetTorrentFile(ctx, "abc123")
+	if err == nil {
+		t.Fatal("expected error for canceled context")
+	}
+}