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
85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
// 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)
|
|
}
|