diff options
Diffstat (limited to 'pkg/web/oidc_server.go')
| -rw-r--r-- | pkg/web/oidc_server.go | 23 |
1 files changed, 19 insertions, 4 deletions
diff --git a/pkg/web/oidc_server.go b/pkg/web/oidc_server.go index 31ef572..86f4e7e 100644 --- a/pkg/web/oidc_server.go +++ b/pkg/web/oidc_server.go @@ -1,8 +1,10 @@ package web import ( + "net" "net/http" "strconv" + "strings" "testing" "time" @@ -20,15 +22,28 @@ type OIDCServer struct { } func NewOIDCServer(t *testing.T) *OIDCServer { - srv, err := mockoidc.Run() + srv, err := mockoidc.NewServer(nil) require.NoError(t, err) - srv.AddMiddleware(func(next http.Handler) http.Handler { + require.NoError(t, srv.AddMiddleware(func(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - t.Logf("%v %v %v\n", r.Method, r.URL.Path, r.URL.Query()) + t.Logf("mockoidc: %v %v %v\n", r.Method, r.URL.Path, r.URL.Query()) next.ServeHTTP(w, r) }) - }) + })) + + ln, err := net.Listen("tcp", "127.0.0.1:0") + require.NoError(t, err) + require.NoError(t, srv.Start(ln, nil)) + if srv.Server != nil { + mux := srv.Server.Handler.(*http.ServeMux) + mux.Handle(strings.Replace(mockoidc.AuthorizationEndpoint, "/oidc", "/oidc/oauth", 1), http.HandlerFunc(srv.Authorize)) + mux.Handle(strings.Replace(mockoidc.TokenEndpoint, "/oidc", "/oidc/oauth", 1), http.HandlerFunc(srv.Token)) + mux.Handle(strings.Replace(mockoidc.UserinfoEndpoint, "/oidc", "/oidc/oauth", 1), http.HandlerFunc(srv.Userinfo)) + mux.Handle(strings.Replace(mockoidc.JWKSEndpoint, "/oidc", "/oidc/oauth", 1), http.HandlerFunc(srv.JWKS)) + mux.Handle(strings.Replace(mockoidc.DiscoveryEndpoint, "/oidc", "/oidc/oauth", 1), http.HandlerFunc(srv.Discovery)) + } + provider, err := oidc.NewProvider(t.Context(), srv.Issuer()) require.NoError(t, err) |