diff options
Diffstat (limited to 'pkg')
| -rw-r--r-- | pkg/test/oidc_server.go | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/pkg/test/oidc_server.go b/pkg/test/oidc_server.go index 5e007d6..152d685 100644 --- a/pkg/test/oidc_server.go +++ b/pkg/test/oidc_server.go @@ -49,16 +49,20 @@ func NewOIDCServer(t *testing.T) *TestServer { } } -func (srv *TestServer) CreateTokenFor(user mockoidc.User) *oauth2.Token { +func (srv *TestServer) CreateAuthorizationCodeFor(user mockoidc.User) string { code := strconv.FormatInt(time.Now().Unix(), 10) srv.QueueUser(user) srv.QueueCode(code) - url := srv.AuthCodeURL("state") - response, err := http.Get(url) + response, err := http.Get(srv.AuthCodeURL("state")) require.NoError(srv, err) srv.Logf("%v\n", response) + return code +} + +func (srv *TestServer) CreateTokenFor(user mockoidc.User) *oauth2.Token { + code := srv.CreateAuthorizationCodeFor(user) token, err := srv.Exchange(srv.Context(), code) require.NoError(srv, err) return token |