update

internal/hubmodule/template/module_test.go

@@ -0,0 +1,73 @@
+package template
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/any-hub/any-hub/internal/proxy/hooks"
+)
+
+// This test shows a full hook lifecycle that module authors can copy when creating a new hook.
+func TestTemplateHookFlow(t *testing.T) {
+	baseURL := "https://example.com"
+	ctx := &hooks.RequestContext{
+		HubName:   "demo",
+		ModuleKey: "template",
+	}
+
+	h := hooks.Hooks{
+		NormalizePath: func(_ *hooks.RequestContext, clean string, rawQuery []byte) (string, []byte) {
+			return "/normalized" + clean, rawQuery
+		},
+		ResolveUpstream: func(_ *hooks.RequestContext, upstream string, clean string, rawQuery []byte) string {
+			if len(rawQuery) > 0 {
+				return upstream + clean + "?" + string(rawQuery)
+			}
+			return upstream + clean
+		},
+		CachePolicy: func(_ *hooks.RequestContext, path string, current hooks.CachePolicy) hooks.CachePolicy {
+			current.AllowCache = path != ""
+			current.AllowStore = true
+			return current
+		},
+		ContentType: func(_ *hooks.RequestContext, path string) string {
+			if path == "/normalized/index.json" {
+				return "application/json"
+			}
+			return ""
+		},
+		RewriteResponse: func(_ *hooks.RequestContext, status int, headers map[string]string, body []byte, _ string) (int, map[string]string, []byte, error) {
+			if headers == nil {
+				headers = map[string]string{}
+			}
+			headers["X-Demo"] = "ok"
+			return status, headers, body, nil
+		},
+	}
+
+	normalized, _ := h.NormalizePath(ctx, "/index.json", nil)
+	if normalized != "/normalized/index.json" {
+		t.Fatalf("expected normalized path, got %s", normalized)
+	}
+	u := h.ResolveUpstream(ctx, baseURL, normalized, nil)
+	if u != baseURL+normalized {
+		t.Fatalf("expected upstream %s, got %s", baseURL+normalized, u)
+	}
+	policy := h.CachePolicy(ctx, normalized, hooks.CachePolicy{})
+	if !policy.AllowCache || !policy.AllowStore {
+		t.Fatalf("expected policy to allow cache/store, got %#v", policy)
+	}
+	status, headers, body, err := h.RewriteResponse(ctx, http.StatusOK, map[string]string{}, []byte("ok"), normalized)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if headers["X-Demo"] != "ok" {
+		t.Fatalf("expected rewrite to set header, got %s", headers["X-Demo"])
+	}
+	if status != http.StatusOK || string(body) != "ok" {
+		t.Fatalf("expected unchanged status/body, got %d/%s", status, string(body))
+	}
+	if ct := h.ContentType(ctx, normalized); ct != "application/json" {
+		t.Fatalf("expected content type application/json, got %s", ct)
+	}
+}