update

internal/hubmodule/composer/hooks_test.go

@@ -0,0 +1,43 @@
+package composer
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/any-hub/any-hub/internal/proxy/hooks"
+)
+
+func TestNormalizePathDropsDistQuery(t *testing.T) {
+	path, raw := normalizePath(nil, "/dist/https/example.com/file.zip", []byte("token=1"))
+	if raw != nil {
+		t.Fatalf("expected query to be dropped")
+	}
+	if path != "/dist/https/example.com/file.zip" {
+		t.Fatalf("unexpected path %s", path)
+	}
+}
+
+func TestResolveDistUpstream(t *testing.T) {
+	url := resolveDistUpstream(nil, "", "/dist/https/example.com/file.zip", []byte("token=1"))
+	if url != "https://example.com/file.zip?token=1" {
+		t.Fatalf("unexpected upstream %s", url)
+	}
+}
+
+func TestRewriteResponseUpdatesURLs(t *testing.T) {
+	ctx := &hooks.RequestContext{Domain: "cache.example"}
+	body := []byte(`{"packages":{"a/b":{"1.0.0":{"dist":{"url":"https://pkg.example/dist.zip"}}}}}`)
+	_, headers, rewritten, err := rewriteResponse(ctx, 200, map[string]string{}, body, "/p2/a/b.json")
+	if err != nil {
+		t.Fatalf("rewrite failed: %v", err)
+	}
+	if string(rewritten) == string(body) {
+		t.Fatalf("expected rewrite to modify payload")
+	}
+	if headers["Content-Type"] != "application/json" {
+		t.Fatalf("expected json content type")
+	}
+	if !strings.Contains(string(rewritten), "https://cache.example/dist/https/pkg.example/dist.zip") {
+		t.Fatalf("expected rewritten URL, got %s", string(rewritten))
+	}
+}