updates

Author: xiangyan99

sdk/core/azure-core/azure/core/pipeline/policies/_authentication.py

@@ -4,6 +4,7 @@
 # license information.
 # -------------------------------------------------------------------------
 import time
+from urllib.parse import urlparse
 
 from . import HTTPPolicy, SansIOHTTPPolicy
 from ...exceptions import ServiceRequestError
@@ -35,7 +36,7 @@ def __init__(self, credential, *scopes, **kwargs):  # pylint:disable=unused-argu
         self._scopes = scopes
         self._credential = credential
         self._token = None  # type: Optional[AccessToken]
-        self._original_url = None
+        self._original_domain = None
         self._always_adding_header = kwargs.pop('always_adding_header', False)
 
     @staticmethod
@@ -75,10 +76,11 @@ def _need_new_token(self):
     def _need_adding_header(self, url):
         if self._always_adding_header:
             return True
-        if not self._original_url:
-            self._original_url = url
+        domain = urlparse(url).netloc
+        if not self._original_domain:
+            self._original_domain = domain
             return True
-        if self._original_url == url:
+        if self._original_domain == domain:
             return True
         return False
 

sdk/core/azure-core/azure/core/pipeline/policies/_authentication_async.py

@@ -5,6 +5,7 @@
 # -------------------------------------------------------------------------
 import asyncio
 import time
+from urllib.parse import urlparse
 from typing import TYPE_CHECKING
 
 from azure.core.pipeline.policies import AsyncHTTPPolicy
@@ -34,7 +35,7 @@ def __init__(self, credential: "AsyncTokenCredential", *scopes: str, **kwargs: "
         self._lock = asyncio.Lock()
         self._scopes = scopes
         self._token = None  # type: Optional[AccessToken]
-        self._original_url = None
+        self._original_domain = None
         self._always_adding_header = kwargs.pop('always_adding_header', False)
 
     async def on_request(self, request: "PipelineRequest") -> None:  # pylint:disable=invalid-overridden-method
@@ -136,9 +137,10 @@ def _need_new_token(self) -> bool:
     def _need_adding_header(self, url):
         if self._always_adding_header:
             return True
-        if not self._original_url:
-            self._original_url = url
+        domain = urlparse(url).netloc
+        if not self._original_domain:
+            self._original_domain = domain
             return True
-        if self._original_url == url:
+        if self._original_domain == domain:
             return True
         return False