[Tables] Fix mypy errors (Azure#39170)

Author: Yalin Li

sdk/tables/azure-data-tables/azure/data/tables/_authentication.py

@@ -209,9 +209,15 @@ def on_challenge(self, request: PipelineRequest, response: PipelineResponse) ->
             return False
 
         if self._discover_tenant:
-            self.authorize_request(request, scope, tenant_id=challenge.tenant_id)
+            if isinstance(scope, str):
+                self.authorize_request(request, scope, tenant_id=challenge.tenant_id)
+            else:
+                self.authorize_request(request, *scope, tenant_id=challenge.tenant_id)
         else:
-            self.authorize_request(request, scope)
+            if isinstance(scope, str):
+                self.authorize_request(request, scope)
+            else:
+                self.authorize_request(request, *scope)
         return True
 
 

sdk/tables/azure-data-tables/azure/data/tables/aio/_authentication_async.py

@@ -65,9 +65,15 @@ async def on_challenge(self, request: PipelineRequest, response: PipelineRespons
             return False
 
         if self._discover_tenant:
-            await self.authorize_request(request, scope, tenant_id=challenge.tenant_id)
+            if isinstance(scope, str):
+                await self.authorize_request(request, scope, tenant_id=challenge.tenant_id)
+            else:
+                await self.authorize_request(request, *scope, tenant_id=challenge.tenant_id)
         else:
-            await self.authorize_request(request, scope)
+            if isinstance(scope, str):
+                await self.authorize_request(request, scope)
+            else:
+                await self.authorize_request(request, *scope)
         return True
 
 

sdk/tables/azure-data-tables/tests/test_challenge_auth.py

@@ -249,7 +249,7 @@ def get_token(*scopes, **kwargs):
             raise ValueError("unexpected token request")
 
         credential = Mock(spec_set=["get_token"], get_token=Mock(wraps=get_token))
-        policy = BearerTokenChallengePolicy(credential, "scope", discover_scopes=False)
+        policy = BearerTokenChallengePolicy(credential, ["scope1", "scope2"], discover_scopes=False)
         pipeline = Pipeline(policies=[policy], transport=Mock(send=send))
         pipeline.run(http_request("GET", "https://localhost"))
 
@@ -300,7 +300,9 @@ def get_token(*scopes, **kwargs):
             raise ValueError("unexpected token request")
 
         credential = Mock(spec_set=["get_token"], get_token=Mock(wraps=get_token))
-        policy = BearerTokenChallengePolicy(credential, "scope", discover_tenant=False, discover_scopes=False)
+        policy = BearerTokenChallengePolicy(
+            credential, ["scope1", "scope2"], discover_tenant=False, discover_scopes=False
+        )
         pipeline = Pipeline(policies=[policy], transport=Mock(send=send))
         pipeline.run(http_request("GET", "https://localhost"))
 

sdk/tables/azure-data-tables/tests/test_challenge_auth_async.py

@@ -250,7 +250,7 @@ async def get_token(*scopes, **kwargs):
             raise ValueError("unexpected token request")
 
         credential = Mock(spec_set=["get_token"], get_token=Mock(wraps=get_token))
-        policy = AsyncBearerTokenChallengePolicy(credential, "scope", discover_scopes=False)
+        policy = AsyncBearerTokenChallengePolicy(credential, ["scope1", "scope2"], discover_scopes=False)
         pipeline = AsyncPipeline(policies=[policy], transport=Mock(send=send))
         await pipeline.run(http_request("GET", "https://localhost"))
 
@@ -301,7 +301,9 @@ async def get_token(*scopes, **kwargs):
             raise ValueError("unexpected token request")
 
         credential = Mock(spec_set=["get_token"], get_token=Mock(wraps=get_token))
-        policy = AsyncBearerTokenChallengePolicy(credential, "scope", discover_tenant=False, discover_scopes=False)
+        policy = AsyncBearerTokenChallengePolicy(
+            credential, ["scope1", "scope2"], discover_tenant=False, discover_scopes=False
+        )
         pipeline = AsyncPipeline(policies=[policy], transport=Mock(send=send))
         await pipeline.run(http_request("GET", "https://localhost"))