test: extract method to generate a valid authorization code grant

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