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

pkg/testhelpers/errorx.go

@@ -0,0 +1,90 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package testhelpers
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/sirupsen/logrus"
+	"github.com/stretchr/testify/require"
+
+	"github.com/ory/x/httpx"
+	"github.com/ory/x/logrusx"
+
+	"github.com/ory/herodot"
+
+	"github.com/ory/kratos/driver/config"
+	"github.com/ory/kratos/identity"
+	"github.com/ory/kratos/selfservice/errorx"
+	"github.com/ory/kratos/session"
+	"github.com/ory/kratos/x"
+)
+
+func NewErrorTestServer(t *testing.T, reg interface {
+	errorx.PersistenceProvider
+	config.Provider
+},
+) *httptest.Server {
+	logger := logrusx.New("", "", logrusx.ForceLevel(logrus.TraceLevel))
+	writer := herodot.NewJSONWriter(logger)
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		e, err := reg.SelfServiceErrorPersister().ReadErrorContainer(r.Context(), x.ParseUUID(r.URL.Query().Get("id")))
+		require.NoError(t, err)
+		t.Logf("landed on error UI with: %s", e.Errors)
+		writer.Write(w, r, e.Errors)
+	}))
+	t.Cleanup(ts.Close)
+	reg.Config().MustSet(t.Context(), config.ViperKeySelfServiceErrorUI, ts.URL)
+	return ts
+}
+
+func NewRedirTS(t *testing.T, body string, conf *config.Config) *httptest.Server {
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if len(body) == 0 {
+			w.WriteHeader(http.StatusNoContent)
+			return
+		}
+		_, _ = w.Write([]byte(body))
+	}))
+	t.Cleanup(ts.Close)
+	conf.MustSet(context.Background(), config.ViperKeySelfServiceBrowserDefaultReturnTo, ts.URL)
+	return ts
+}
+
+func NewRedirSessionEchoTS(t *testing.T, reg interface {
+	httpx.WriterProvider
+	session.ManagementProvider
+	config.Provider
+},
+) *httptest.Server {
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// verify that the client has a session, and echo it back
+		sess, err := reg.SessionManager().FetchFromRequest(r.Context(), r, session.ExpandEverything, identity.ExpandEverything)
+		require.NoError(t, err, "Headers: %+v", r.Header)
+		reg.Writer().Write(w, r, sess)
+	}))
+	t.Cleanup(ts.Close)
+	reg.Config().MustSet(context.Background(), config.ViperKeySelfServiceBrowserDefaultReturnTo, ts.URL+"/return-ts")
+	return ts
+}
+
+func NewRedirNoSessionTS(t *testing.T, reg interface {
+	httpx.WriterProvider
+	session.ManagementProvider
+	config.Provider
+},
+) *httptest.Server {
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// verify that the client DOES NOT have a session
+		_, err := reg.SessionManager().FetchFromRequest(r.Context(), r, session.ExpandEverything, identity.ExpandEverything)
+		require.Error(t, err, "Headers: %+v", r.Header)
+		reg.Writer().Write(w, r, nil)
+	}))
+	t.Cleanup(ts.Close)
+	reg.Config().MustSet(context.Background(), config.ViperKeySelfServiceBrowserDefaultReturnTo, ts.URL+"/return-ts")
+	return ts
+}