From c6a5a47f05d92e4ca1089b8d0fe35b4a550d605b Mon Sep 17 00:00:00 2001 From: mo khan Date: Thu, 17 Apr 2025 13:31:33 -0600 Subject: test: extract method to generate a valid authorization code grant --- pkg/test/oidc_server.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'pkg/test') 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 -- cgit v1.2.3