feat: bootstrap reusable GoOmni framework

router_test.go

@@ -0,0 +1,69 @@
+package omni
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"git.yuntuo.hk/k2safe/go-omni/httpx"
+)
+
+func TestNewWithRouterKeepsHandlersTransportNeutral(t *testing.T) {
+	router := &testRouter{}
+	app := NewWithRouter(router)
+	app.GET("/health", func(ctx *Context) error {
+		if ctx.RequestID() == "" {
+			t.Fatal("expected request ID from core middleware")
+		}
+		return ctx.OK(map[string]string{"status": "ok"})
+	})
+
+	recorder := httptest.NewRecorder()
+	request := httptest.NewRequest(http.MethodGet, "/health", nil)
+	app.ServeHTTP(recorder, request)
+	if recorder.Code != http.StatusOK {
+		t.Fatalf("expected status 200, got %d", recorder.Code)
+	}
+}
+
+type testRouter struct {
+	handlers map[string]http.Handler
+}
+
+func (r *testRouter) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
+	if handler := r.handlers[request.Method+" "+request.URL.Path]; handler != nil {
+		handler.ServeHTTP(writer, request)
+		return
+	}
+	http.NotFound(writer, request)
+}
+
+func (r *testRouter) Handle(method, path string, handler http.Handler) {
+	if r.handlers == nil {
+		r.handlers = make(map[string]http.Handler)
+	}
+	r.handlers[method+" "+path] = handler
+}
+
+func (r *testRouter) Any(path string, handler http.Handler) { r.Handle(http.MethodGet, path, handler) }
+
+func (r *testRouter) Group(path string) httpx.RouterGroup {
+	return &testRouterGroup{router: r, prefix: path}
+}
+
+type testRouterGroup struct {
+	router *testRouter
+	prefix string
+}
+
+func (g *testRouterGroup) Handle(method, path string, handler http.Handler) {
+	g.router.Handle(method, g.prefix+path, handler)
+}
+
+func (g *testRouterGroup) Any(path string, handler http.Handler) {
+	g.Handle(http.MethodGet, path, handler)
+}
+
+func (g *testRouterGroup) Group(path string) httpx.RouterGroup {
+	return &testRouterGroup{router: g.router, prefix: g.prefix + path}
+}