mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
fix(coderd/externalauth): preserve scopes on entra v1 token refresh (#24851)
Without this, Entra silently narrows scopes to the default set.
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"net/url"
|
||||
@@ -850,6 +851,149 @@ func TestRefreshToken(t *testing.T) {
|
||||
})
|
||||
}
|
||||
|
||||
// TestRefreshTokenWithScopes verifies the refresh path echoes Config.Scopes on
|
||||
// the token-endpoint request and preserves the prior refresh_token when the
|
||||
// authorization server omits a new one (RFC 6749 §6).
|
||||
func TestRefreshTokenWithScopes(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
// fakeAS returns an http.Client + a pointer the test can read after
|
||||
// RefreshToken returns. The roundTripper captures the form body of every
|
||||
// outbound request and replies with tokenJSON to refresh requests.
|
||||
fakeAS := func(t *testing.T, tokenJSON []byte) (*http.Client, *url.Values) {
|
||||
t.Helper()
|
||||
captured := &url.Values{}
|
||||
client := &http.Client{Transport: roundTripper(func(req *http.Request) (*http.Response, error) {
|
||||
body, err := io.ReadAll(req.Body)
|
||||
require.NoError(t, err)
|
||||
values, err := url.ParseQuery(string(body))
|
||||
require.NoError(t, err)
|
||||
if values.Get("grant_type") == "refresh_token" {
|
||||
*captured = values
|
||||
}
|
||||
rec := httptest.NewRecorder()
|
||||
rec.Header().Set("Content-Type", "application/json")
|
||||
rec.WriteHeader(http.StatusOK)
|
||||
_, err = rec.Write(tokenJSON)
|
||||
return rec.Result(), err
|
||||
})}
|
||||
return client, captured
|
||||
}
|
||||
|
||||
newConfig := func(t *testing.T, scopes []string) *externalauth.Config {
|
||||
t.Helper()
|
||||
instrument := promoauth.NewFactory(prometheus.NewRegistry())
|
||||
configs, err := externalauth.ConvertConfig(instrument, []codersdk.ExternalAuthConfig{{
|
||||
ID: "test",
|
||||
Type: codersdk.EnhancedExternalAuthProviderAzureDevopsEntra.String(),
|
||||
ClientID: "id",
|
||||
ClientSecret: "secret",
|
||||
AuthURL: "https://login.microsoftonline.com/tenant/oauth2/authorize",
|
||||
TokenURL: "https://login.microsoftonline.com/tenant/oauth2/token",
|
||||
Scopes: scopes,
|
||||
}}, &url.URL{Scheme: "https", Host: "coder.example.com"})
|
||||
require.NoError(t, err)
|
||||
return configs[0]
|
||||
}
|
||||
|
||||
expired := dbtime.Now().Add(-time.Hour)
|
||||
|
||||
// mockDBPassthrough returns a mock store that echoes the
|
||||
// UpdateExternalAuthLink params back as a populated ExternalAuthLink,
|
||||
// letting the test read what RefreshToken decided to persist.
|
||||
mockDBPassthrough := func(t *testing.T) database.Store {
|
||||
t.Helper()
|
||||
ctrl := gomock.NewController(t)
|
||||
mDB := dbmock.NewMockStore(ctrl)
|
||||
mDB.EXPECT().UpdateExternalAuthLink(gomock.Any(), gomock.Any()).
|
||||
DoAndReturn(func(_ context.Context, p database.UpdateExternalAuthLinkParams) (database.ExternalAuthLink, error) {
|
||||
return database.ExternalAuthLink{
|
||||
ProviderID: p.ProviderID,
|
||||
UserID: p.UserID,
|
||||
OAuthAccessToken: p.OAuthAccessToken,
|
||||
OAuthRefreshToken: p.OAuthRefreshToken,
|
||||
OAuthExpiry: p.OAuthExpiry,
|
||||
}, nil
|
||||
}).AnyTimes()
|
||||
return mDB
|
||||
}
|
||||
|
||||
t.Run("EchoesConfiguredScopesOnRefresh", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
client, captured := fakeAS(t,
|
||||
[]byte(`{"access_token":"new","refresh_token":"new-r","token_type":"bearer","expires_in":3600}`))
|
||||
cfg := newConfig(t, []string{"openid", "offline_access", "api://app/session:role-any"})
|
||||
|
||||
ctx := context.WithValue(context.Background(), oauth2.HTTPClient, client)
|
||||
_, err := cfg.RefreshToken(ctx, mockDBPassthrough(t), database.ExternalAuthLink{
|
||||
OAuthAccessToken: "old",
|
||||
OAuthRefreshToken: "old-r",
|
||||
OAuthExpiry: expired,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Equal(t, "refresh_token", captured.Get("grant_type"))
|
||||
require.Equal(t, "old-r", captured.Get("refresh_token"))
|
||||
require.Equal(t, "openid offline_access api://app/session:role-any", captured.Get("scope"),
|
||||
"refresh request must echo configured scopes joined by space")
|
||||
})
|
||||
|
||||
t.Run("OmitsScopeParamWhenScopesEmpty", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
client, captured := fakeAS(t,
|
||||
[]byte(`{"access_token":"new","refresh_token":"new-r","token_type":"bearer","expires_in":3600}`))
|
||||
cfg := newConfig(t, nil)
|
||||
|
||||
ctx := context.WithValue(context.Background(), oauth2.HTTPClient, client)
|
||||
_, err := cfg.RefreshToken(ctx, mockDBPassthrough(t), database.ExternalAuthLink{
|
||||
OAuthAccessToken: "old",
|
||||
OAuthRefreshToken: "old-r",
|
||||
OAuthExpiry: expired,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Equal(t, "refresh_token", captured.Get("grant_type"))
|
||||
require.Equal(t, "old-r", captured.Get("refresh_token"))
|
||||
require.Empty(t, captured.Get("scope"),
|
||||
"refresh request must not send a scope param when Config.Scopes is empty")
|
||||
})
|
||||
|
||||
t.Run("PreservesPriorRefreshTokenWhenASOmitsNewOne", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
// Token response intentionally omits refresh_token.
|
||||
client, _ := fakeAS(t,
|
||||
[]byte(`{"access_token":"new","token_type":"bearer","expires_in":3600}`))
|
||||
cfg := newConfig(t, nil)
|
||||
|
||||
ctx := context.WithValue(context.Background(), oauth2.HTTPClient, client)
|
||||
link, err := cfg.RefreshToken(ctx, mockDBPassthrough(t), database.ExternalAuthLink{
|
||||
OAuthAccessToken: "old",
|
||||
OAuthRefreshToken: "prior-r",
|
||||
OAuthExpiry: expired,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, "prior-r", link.OAuthRefreshToken,
|
||||
"prior refresh_token must be preserved when AS omits a new one (RFC 6749 §6)")
|
||||
})
|
||||
|
||||
t.Run("AcceptsRotatedRefreshTokenWhenASReturnsOne", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
client, _ := fakeAS(t,
|
||||
[]byte(`{"access_token":"new","refresh_token":"rotated-r","token_type":"bearer","expires_in":3600}`))
|
||||
cfg := newConfig(t, nil)
|
||||
|
||||
ctx := context.WithValue(context.Background(), oauth2.HTTPClient, client)
|
||||
link, err := cfg.RefreshToken(ctx, mockDBPassthrough(t), database.ExternalAuthLink{
|
||||
OAuthAccessToken: "old",
|
||||
OAuthRefreshToken: "prior-r",
|
||||
OAuthExpiry: expired,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, "rotated-r", link.OAuthRefreshToken,
|
||||
"rotated refresh_token from AS must be persisted")
|
||||
})
|
||||
}
|
||||
|
||||
func TestValidateToken(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user