snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

pkg/testhelpers/server.go

@@ -0,0 +1,84 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package testhelpers
+
+import (
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/urfave/negroni"
+
+	"github.com/ory/kratos/driver"
+	"github.com/ory/kratos/driver/config"
+	"github.com/ory/kratos/x"
+	"github.com/ory/kratos/x/nosurfx"
+	"github.com/ory/x/httprouterx"
+)
+
+func NewKratosServer(t *testing.T, reg driver.Registry) (public, admin *httptest.Server) {
+	return NewKratosServerWithRouters(t, reg, httprouterx.NewRouterPublic(), httprouterx.NewRouterAdminWithPrefix())
+}
+
+func NewKratosServerWithCSRF(t *testing.T, reg driver.Registry) (public, admin *httptest.Server) {
+	public, admin, _, _ = NewKratosServerWithCSRFAndRouters(t, reg)
+	return
+}
+
+func NewKratosServerWithCSRFAndRouters(t *testing.T, reg driver.Registry) (public, admin *httptest.Server, rp *httprouterx.RouterPublic, ra *httprouterx.RouterAdmin) {
+	rp, ra = httprouterx.NewRouterPublic(), httprouterx.NewRouterAdminWithPrefix()
+	csrfHandler := nosurfx.NewTestCSRFHandler(rp, reg)
+	reg.WithCSRFHandler(csrfHandler)
+
+	ran := negroni.New()
+	ran.UseFunc(httprouterx.AddAdminPrefixIfNotPresentNegroni)
+	ran.UseHandler(ra)
+
+	rpn := negroni.New()
+	rpn.UseFunc(x.HTTPLoaderContextMiddleware(reg))
+	rpn.UseHandler(rp)
+
+	public = httptest.NewServer(nosurfx.NewTestCSRFHandler(rpn, reg))
+	admin = httptest.NewServer(ran)
+
+	public.URL = strings.ReplaceAll(public.URL, "127.0.0.1", "localhost")
+
+	initKratosServers(t, reg, public, admin, rp, ra)
+
+	t.Cleanup(public.Close)
+	t.Cleanup(admin.Close)
+	return
+}
+
+func NewKratosServerWithRouters(t *testing.T, reg driver.Registry, rp *httprouterx.RouterPublic, ra *httprouterx.RouterAdmin) (public, admin *httptest.Server) {
+	np := negroni.New()
+	np.UseFunc(httprouterx.TrimTrailingSlashNegroni)
+	np.UseHandler(rp)
+
+	na := negroni.New()
+	na.UseFunc(httprouterx.AddAdminPrefixIfNotPresentNegroni)
+	na.UseFunc(httprouterx.TrimTrailingSlashNegroni)
+	na.UseHandler(ra)
+
+	public = httptest.NewServer(np)
+	admin = httptest.NewServer(na)
+
+	initKratosServers(t, reg, public, admin, rp, ra)
+
+	t.Cleanup(public.Close)
+	t.Cleanup(admin.Close)
+	return
+}
+
+func initKratosServers(t *testing.T, reg driver.Registry, public, admin *httptest.Server, rp *httprouterx.RouterPublic, ra *httprouterx.RouterAdmin) {
+	ctx := t.Context()
+	if len(reg.Config().GetProvider(ctx).String(config.ViperKeySelfServiceLoginUI)) == 0 {
+		reg.Config().MustSet(ctx, config.ViperKeySelfServiceLoginUI, "http://NewKratosServerWithRouters/you-forgot-to-set-me/login")
+	}
+	reg.Config().MustSet(ctx, config.ViperKeyPublicBaseURL, public.URL)
+	reg.Config().MustSet(ctx, config.ViperKeyAdminBaseURL, admin.URL)
+
+	reg.RegisterPublicRoutes(ctx, rp)
+	reg.RegisterAdminRoutes(ctx, ra)
+}