Skip to content

Commit a5b9161

Browse files
committed
fix: rename internal refresh token attribute to _refresh_token_val and add _refresh_token wrapper
Rename the stored refresh token attribute from _refresh_token to _refresh_token_val to avoid a name collision with a newly added _refresh_token(request) wrapper that delegates to refresh(request). Update all references and tests accordingly to match changes in fix test_aws_based_external_account system test Fixes #1279
1 parent 0387bb9 commit a5b9161

File tree

2 files changed

+11
-8
lines changed

2 files changed

+11
-8
lines changed

google/oauth2/credentials.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def __init__(
139139
super(Credentials, self).__init__()
140140
self.token = token
141141
self.expiry = expiry
142-
self._refresh_token = refresh_token
142+
self._refresh_token_val = refresh_token
143143
self._id_token = id_token
144144
self._scopes = scopes
145145
self._default_scopes = default_scopes
@@ -178,7 +178,7 @@ def __setstate__(self, d):
178178
all the attributes."""
179179
self.token = d.get("token")
180180
self.expiry = d.get("expiry")
181-
self._refresh_token = d.get("_refresh_token")
181+
self._refresh_token_val = d.get("_refresh_token_val")
182182
self._id_token = d.get("_id_token")
183183
self._scopes = d.get("_scopes")
184184
self._default_scopes = d.get("_default_scopes")
@@ -203,7 +203,7 @@ def __setstate__(self, d):
203203
@property
204204
def refresh_token(self):
205205
"""Optional[str]: The OAuth 2.0 refresh token."""
206-
return self._refresh_token
206+
return self._refresh_token_val
207207

208208
@property
209209
def scopes(self):
@@ -350,6 +350,9 @@ def with_universe_domain(self, universe_domain):
350350
def _metric_header_for_usage(self):
351351
return metrics.CRED_TYPE_USER
352352

353+
def _refresh_token(self, request):
354+
return self.refresh(request)
355+
353356
@_helpers.copy_docstring(credentials.Credentials)
354357
def refresh(self, request):
355358
if self._universe_domain != credentials.DEFAULT_UNIVERSE_DOMAIN:
@@ -368,7 +371,7 @@ def refresh(self, request):
368371
# available. This is useful in general when tokens are obtained by calling
369372
# some external process on demand. It is particularly useful for retrieving
370373
# downscoped tokens from a token broker.
371-
if self._refresh_token is None and self.refresh_handler:
374+
if self._refresh_token_val is None and self.refresh_handler:
372375
token, expiry = self.refresh_handler(request, scopes=scopes)
373376
# Validate returned data.
374377
if not isinstance(token, str):
@@ -389,7 +392,7 @@ def refresh(self, request):
389392
return
390393

391394
if (
392-
self._refresh_token is None
395+
self._refresh_token_val is None
393396
or self._token_uri is None
394397
or self._client_id is None
395398
or self._client_secret is None
@@ -409,7 +412,7 @@ def refresh(self, request):
409412
) = reauth.refresh_grant(
410413
request,
411414
self._token_uri,
412-
self._refresh_token,
415+
self._refresh_token_val,
413416
self._client_id,
414417
self._client_secret,
415418
scopes=scopes,
@@ -419,7 +422,7 @@ def refresh(self, request):
419422

420423
self.token = access_token
421424
self.expiry = expiry
422-
self._refresh_token = refresh_token
425+
self._refresh_token_val = refresh_token
423426
self._id_token = grant_response.get("id_token")
424427
self._rapt_token = rapt_token
425428

tests/test__oauth2client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def test__convert_oauth2_credentials():
5252
new_credentials = _oauth2client._convert_oauth2_credentials(old_credentials)
5353

5454
assert new_credentials.token == old_credentials.access_token
55-
assert new_credentials._refresh_token == old_credentials.refresh_token
55+
assert new_credentials._refresh_token_val == old_credentials.refresh_token
5656
assert new_credentials._client_id == old_credentials.client_id
5757
assert new_credentials._client_secret == old_credentials.client_secret
5858
assert new_credentials._token_uri == old_credentials.token_uri

0 commit comments

Comments
 (0)