c60cc7ca75
Some checks failed
CI / Run tests and lints (push) Failing after 29s
CI / Build CLI docs (push) Has been skipped
CI / Run Playwright end-to-end tests (mysql) (push) Failing after 3s
CI / Run Playwright end-to-end tests (postgres) (push) Failing after 2s
CI / Run Playwright end-to-end tests (sqlite) (push) Failing after 3s
CodeQL / Analyze (go) (push) Failing after 4s
CodeQL / Analyze (javascript) (push) Failing after 3s
Docker Image Scanners / scanners (push) Failing after 4s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Run end-to-end tests (postgres) (push) Failing after 19s
CI / Run end-to-end tests (sqlite) (push) Failing after 2s
CI / Run Playwright end-to-end tests (cockroach) (push) Failing after 2s
CI / Draft newsletter (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
91 lines
2.8 KiB
Go
91 lines
2.8 KiB
Go
// 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
|
|
}
|