Skip to content

Commit 4826c98

Browse files
authored
[Core] aaz: Fix AAZBearerTokenCredentialPolicy to support MSIAuthenticationWrapper crendital (#23471)
1 parent 0625878 commit 4826c98

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/azure-cli-core/azure/cli/core/aaz/_http_policy.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,8 @@ def _need_new_token(self):
5858

5959
@property
6060
def _need_new_aux_tokens(self):
61+
if not hasattr(self._credential, 'get_auxiliary_tokens'):
62+
return False
6163
if not self._aux_tokens:
6264
return True
6365
for token in self._aux_tokens:
@@ -91,8 +93,11 @@ def authorize_request(self, request, *scopes, **kwargs):
9193
:param ~azure.core.pipeline.PipelineRequest request: the request
9294
:param str scopes: required scopes of authentication
9395
"""
94-
self._token = self._credential.get_token(*scopes, **kwargs)
95-
self._aux_tokens = self._credential.get_auxiliary_tokens(*self._scopes)
96+
if self._need_new_token:
97+
self._token = self._credential.get_token(*scopes, **kwargs)
98+
99+
if self._need_new_aux_tokens:
100+
self._aux_tokens = self._credential.get_auxiliary_tokens(*self._scopes)
96101

97102
self._update_headers(request.http_request.headers)
98103

0 commit comments

Comments
 (0)