update

internal/proxy/handler_hook_test.go

@@ -0,0 +1,80 @@
+package proxy
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/gofiber/fiber/v3"
+	"github.com/valyala/fasthttp"
+
+	"github.com/any-hub/any-hub/internal/cache"
+	"github.com/any-hub/any-hub/internal/config"
+	"github.com/any-hub/any-hub/internal/proxy/hooks"
+	"github.com/any-hub/any-hub/internal/server"
+)
+
+func TestResolveUpstreamPrefersHook(t *testing.T) {
+	app := fiber.New()
+	defer app.Shutdown()
+
+	ctx := app.AcquireCtx(new(fasthttp.RequestCtx))
+	defer app.ReleaseCtx(ctx)
+	ctx.Request().SetRequestURI("/original/path?from=req")
+
+	base, _ := url.Parse("https://up.example")
+	route := &server.HubRoute{
+		Config: config.HubConfig{
+			Name: "demo",
+			Type: "custom",
+		},
+		UpstreamURL: base,
+	}
+	hook := &hookState{
+		ctx: &hooks.RequestContext{},
+		def: hooks.Hooks{
+			NormalizePath: func(_ *hooks.RequestContext, clean string, rawQuery []byte) (string, []byte) {
+				return clean, rawQuery
+			},
+			ResolveUpstream: func(_ *hooks.RequestContext, upstream string, clean string, rawQuery []byte) string {
+				return upstream + "/hooked"
+			},
+		},
+		hasHooks: true,
+		clean:    "/ignored",
+		rawQuery: []byte("ignored=1"),
+	}
+
+	target := resolveUpstreamURL(route, base, ctx, hook)
+	if target.String() != "https://up.example/hooked" {
+		t.Fatalf("expected hook override, got %s", target.String())
+	}
+}
+
+func TestCachePolicyHookOverrides(t *testing.T) {
+	route := &server.HubRoute{
+		Config: config.HubConfig{
+			Name: "demo",
+			Type: "npm",
+		},
+	}
+	locator := cacheLocatorForTest("demo", "/a.tgz")
+	hook := hooks.Hooks{
+		CachePolicy: func(_ *hooks.RequestContext, _ string, current hooks.CachePolicy) hooks.CachePolicy {
+			current.AllowCache = false
+			current.RequireRevalidate = false
+			return current
+		},
+	}
+	ctx := &hooks.RequestContext{Method: fiber.MethodGet}
+	policy := determineCachePolicyWithHook(route, locator, fiber.MethodGet, hook, true, ctx)
+	if policy.allowCache {
+		t.Fatalf("expected hook to disable cache")
+	}
+	if policy.requireRevalidate {
+		t.Fatalf("expected hook to disable revalidate")
+	}
+}
+
+func cacheLocatorForTest(hub, path string) cache.Locator {
+	return cache.Locator{HubName: hub, Path: path}
+}