diff options
Diffstat (limited to 'pkg')
| -rw-r--r-- | pkg/web/oidc_server.go (renamed from pkg/oidc/test_server.go) | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/pkg/oidc/test_server.go b/pkg/web/oidc_server.go index 80f2c9a..31ef572 100644 --- a/pkg/oidc/test_server.go +++ b/pkg/web/oidc_server.go @@ -1,4 +1,4 @@ -package oidc +package web import ( "net/http" @@ -12,14 +12,14 @@ import ( "golang.org/x/oauth2" ) -type TestServer struct { +type OIDCServer struct { *mockoidc.MockOIDC *oauth2.Config *oidc.Provider *testing.T } -func NewTestServer(t *testing.T) *TestServer { +func NewOIDCServer(t *testing.T) *OIDCServer { srv, err := mockoidc.Run() require.NoError(t, err) @@ -32,7 +32,7 @@ func NewTestServer(t *testing.T) *TestServer { provider, err := oidc.NewProvider(t.Context(), srv.Issuer()) require.NoError(t, err) - return &TestServer{ + return &OIDCServer{ srv, &oauth2.Config{ ClientID: srv.ClientID, @@ -46,7 +46,7 @@ func NewTestServer(t *testing.T) *TestServer { } } -func (srv *TestServer) CreateAuthorizationCodeFor(user mockoidc.User) string { +func (srv *OIDCServer) CreateAuthorizationCodeFor(user mockoidc.User) string { code := strconv.FormatInt(time.Now().Unix(), 10) srv.QueueUser(user) srv.QueueCode(code) @@ -56,21 +56,21 @@ func (srv *TestServer) CreateAuthorizationCodeFor(user mockoidc.User) string { return code } -func (srv *TestServer) CreateTokenFor(user mockoidc.User) *oauth2.Token { +func (srv *OIDCServer) CreateTokenFor(user mockoidc.User) *oauth2.Token { code := srv.CreateAuthorizationCodeFor(user) token, err := srv.Exchange(srv.Context(), code) require.NoError(srv, err) return token } -func (srv *TestServer) CreateTokensFor(user mockoidc.User) (*oauth2.Token, string) { +func (srv *OIDCServer) CreateTokensFor(user mockoidc.User) (*oauth2.Token, string) { token := srv.CreateTokenFor(user) rawIDToken, ok := token.Extra("id_token").(string) require.True(srv, ok) return token, rawIDToken } -func (srv *TestServer) Verify(rawIDToken string) *oidc.IDToken { +func (srv *OIDCServer) Verify(rawIDToken string) *oidc.IDToken { idToken, err := srv. Verifier(&oidc.Config{ClientID: srv.MockOIDC.Config().ClientID}). Verify(srv.Context(), rawIDToken) @@ -79,6 +79,6 @@ func (srv *TestServer) Verify(rawIDToken string) *oidc.IDToken { return idToken } -func (s *TestServer) Close() { +func (s *OIDCServer) Close() { s.Shutdown() } |
