init

internal/proxy/docker_path_test.go

@@ -0,0 +1,68 @@
+package proxy
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/any-hub/any-hub/internal/config"
+	"github.com/any-hub/any-hub/internal/server"
+)
+
+func TestApplyDockerHubNamespaceFallback(t *testing.T) {
+	route := dockerHubRoute(t, "https://registry-1.docker.io")
+
+	path, changed := applyDockerHubNamespaceFallback(route, "/v2/nginx/manifests/latest")
+	if !changed {
+		t.Fatalf("expected fallback to apply")
+	}
+	if path != "/v2/library/nginx/manifests/latest" {
+		t.Fatalf("unexpected normalized path: %s", path)
+	}
+
+	path, changed = applyDockerHubNamespaceFallback(route, "/v2/library/nginx/manifests/latest")
+	if changed {
+		t.Fatalf("expected no changes for already-namespaced repo")
+	}
+
+	path, changed = applyDockerHubNamespaceFallback(route, "/v2/rogee/nginx/manifests/latest")
+	if changed {
+		t.Fatalf("expected no changes for custom namespace")
+	}
+
+	path, changed = applyDockerHubNamespaceFallback(route, "/v2/_catalog")
+	if changed {
+		t.Fatalf("expected no changes for _catalog endpoint")
+	}
+
+	otherRoute := dockerHubRoute(t, "https://registry.example.com")
+	path, changed = applyDockerHubNamespaceFallback(otherRoute, "/v2/nginx/manifests/latest")
+	if changed || path != "/v2/nginx/manifests/latest" {
+		t.Fatalf("expected no changes for non-docker-hub upstream")
+	}
+}
+
+func TestSplitDockerRepoPath(t *testing.T) {
+	repo, rest, ok := splitDockerRepoPath("/v2/library/nginx/manifests/latest")
+	if !ok || repo != "library/nginx" || rest != "/manifests/latest" {
+		t.Fatalf("unexpected split result repo=%s rest=%s ok=%v", repo, rest, ok)
+	}
+
+	if _, _, ok := splitDockerRepoPath("/v2/_catalog"); ok {
+		t.Fatalf("expected catalog path to be ignored")
+	}
+}
+
+func dockerHubRoute(t *testing.T, upstream string) *server.HubRoute {
+	t.Helper()
+	parsed, err := url.Parse(upstream)
+	if err != nil {
+		t.Fatalf("invalid upstream: %v", err)
+	}
+	return &server.HubRoute{
+		Config: config.HubConfig{
+			Name: "docker",
+			Type: "docker",
+		},
+		UpstreamURL: parsed,
+	}
+}