99 lines
3.5 KiB
Go
99 lines
3.5 KiB
Go
package composer
|
|
|
|
import (
|
|
"encoding/json"
|
|
"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 TestResolveMirrorDistUpstream(t *testing.T) {
|
|
resetComposerDistRegistry()
|
|
composerDists.remember("cache.example", "vendor/pkg", "abc123", "zip", "https://github.com/org/repo.zip")
|
|
ctx := &hooks.RequestContext{Domain: "cache.example"}
|
|
url := resolveDistUpstream(ctx, "", "/dists/vendor/pkg/abc123.zip", nil)
|
|
if url != "https://github.com/org/repo.zip" {
|
|
t.Fatalf("unexpected upstream %s", url)
|
|
}
|
|
}
|
|
|
|
func TestRewriteResponseUpdatesURLs(t *testing.T) {
|
|
resetComposerDistRegistry()
|
|
ctx := &hooks.RequestContext{Domain: "cache.example"}
|
|
body := []byte(`{"packages":{"a/b":{"1.0.0":{"dist":{"url":"https://api.github.com/repos/org/repo/zipball/ref","reference":"abc123","type":"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")
|
|
}
|
|
var payload map[string]any
|
|
if err := json.Unmarshal(rewritten, &payload); err != nil {
|
|
t.Fatalf("unmarshal: %v", err)
|
|
}
|
|
pkgs := payload["packages"].(map[string]any)
|
|
versions := pkgs["a/b"].(map[string]any)
|
|
version := versions["1.0.0"].(map[string]any)
|
|
dist := version["dist"].(map[string]any)
|
|
distURL := dist["url"].(string)
|
|
expected := "https://cache.example/dist/https/api.github.com/repos/org/repo/zipball/ref"
|
|
if distURL != expected {
|
|
t.Fatalf("expected dist url %s, got %s", expected, distURL)
|
|
}
|
|
}
|
|
|
|
func TestRewritePackagesRoot(t *testing.T) {
|
|
resetComposerDistRegistry()
|
|
ctx := &hooks.RequestContext{Domain: "cache.example"}
|
|
body := []byte(`{"metadata-url":"https://repo.packagist.org/p2/%package%.json","providers-url":"/p/%package%$%hash%.json"}`)
|
|
_, headers, rewritten, err := rewriteResponse(ctx, 200, map[string]string{}, body, "/packages.json")
|
|
if err != nil {
|
|
t.Fatalf("rewrite failed: %v", err)
|
|
}
|
|
if headers["Content-Type"] != "application/json" {
|
|
t.Fatalf("expected json content type")
|
|
}
|
|
var payload map[string]any
|
|
if err := json.Unmarshal(rewritten, &payload); err != nil {
|
|
t.Fatalf("unmarshal: %v", err)
|
|
}
|
|
if payload["metadata-url"] != "https://cache.example/p2/%package%.json" {
|
|
t.Fatalf("metadata URL not rewritten: %v", payload["metadata-url"])
|
|
}
|
|
if payload["providers-url"] != "https://cache.example/p/%package%$%hash%.json" {
|
|
t.Fatalf("providers URL not rewritten: %v", payload["providers-url"])
|
|
}
|
|
mirrors, _ := payload["mirrors"].([]any)
|
|
if len(mirrors) == 0 {
|
|
t.Fatalf("mirrors missing")
|
|
}
|
|
entry, _ := mirrors[0].(map[string]any)
|
|
if entry["dist-url"] != "https://cache.example/dists/%package%/%reference%.%type%" {
|
|
t.Fatalf("unexpected mirror dist-url: %v", entry["dist-url"])
|
|
}
|
|
if pref, _ := entry["preferred"].(bool); !pref {
|
|
t.Fatalf("mirror preferred flag missing")
|
|
}
|
|
}
|