Update tests (Azure#28627)

Author: Yalin Li

sdk/tables/azure-data-tables/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/tables/azure-data-tables",
-  "Tag": "python/tables/azure-data-tables_9a03fa14fd"
+  "Tag": "python/tables/azure-data-tables_15f681b7d2"
 }

sdk/tables/azure-data-tables/tests/_shared/asynctestcase.py

@@ -42,9 +42,9 @@ class AsyncTableTestCase(TableTestCase):
     def get_token_credential(self):
         if is_live():
             return DefaultAzureCredential()
-        return self.generate_fake_token()
+        return self.generate_fake_token_credential()
 
-    def generate_fake_token(self):
+    def generate_fake_token_credential(self):
         return AsyncFakeTokenCredential()
 
     def _get_table_reference(self, prefix=TEST_TABLE_PREFIX):

sdk/tables/azure-data-tables/tests/_shared/testcase.py

@@ -109,9 +109,9 @@ def generate_sas_token(self):
     def get_token_credential(self):
         if is_live():
             return DefaultAzureCredential()
-        return self.generate_fake_token()
+        return self.generate_fake_token_credential()
 
-    def generate_fake_token(self):
+    def generate_fake_token_credential(self):
         return FakeTokenCredential()
 
     def _get_table_reference(self, prefix=TEST_TABLE_PREFIX):

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

@@ -26,7 +26,6 @@
     UpdateMode
 )
 from azure.data.tables._common_conversion import TZ_UTC
-from azure.identity import DefaultAzureCredential
 
 from azure.core import MatchConditions
 from azure.core.credentials import AzureSasCredential
@@ -2254,17 +2253,10 @@ def test_upsert_entity_with_invalid_key_type(self, tables_storage_account_name,
             self._tear_down()
 
     @tables_decorator
+    @recorded_by_proxy
     def test_list_tables_with_invalid_credential(self, tables_storage_account_name, tables_primary_storage_account_key):
         account_url = self.account_url(tables_storage_account_name, "table")
-        credential = DefaultAzureCredential(
-            exclude_environment_credential=True,
-            exclude_managed_identity_credential=False,
-            exclude_shared_token_cache_credential=True,
-            exclude_visual_studio_code_credential=True,
-            exclude_cli_credential=True,
-            exclude_interactive_browser_credential=True,
-            exclude_powershell_credential=True,
-        )
+        credential = self.generate_fake_token_credential()
         client = TableServiceClient(credential=credential, endpoint=account_url, api_version="2020-12-06")
         with pytest.raises(ClientAuthenticationError):
             for _ in client.list_tables():

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

@@ -5,12 +5,12 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from uuid import UUID
 import pytest
 
 from datetime import datetime, timedelta
 from dateutil.tz import tzutc, tzoffset
 from math import isnan
+from uuid import UUID
 
 from devtools_testutils import AzureRecordedTestCase
 from devtools_testutils.aio import recorded_by_proxy_async
@@ -36,7 +36,6 @@
 )
 from azure.data.tables.aio import TableServiceClient
 from azure.data.tables._common_conversion import TZ_UTC
-from azure.identity.aio import DefaultAzureCredential
 
 from _shared.asynctestcase import AsyncTableTestCase
 from async_preparers import tables_decorator_async
@@ -2250,19 +2249,11 @@ async def test_upsert_entity_with_invalid_key_type(self, tables_storage_account_
             await self._tear_down()
 
     @tables_decorator_async
+    @recorded_by_proxy_async
     async def test_list_tables_with_invalid_credential(self, tables_storage_account_name, tables_primary_storage_account_key):
         account_url = self.account_url(tables_storage_account_name, "table")
-        credential = DefaultAzureCredential(
-            exclude_environment_credential=True,
-            exclude_managed_identity_credential=False,
-            exclude_shared_token_cache_credential=True,
-            exclude_visual_studio_code_credential=True,
-            exclude_cli_credential=True,
-            exclude_interactive_browser_credential=True,
-            exclude_powershell_credential=True,
-        )
-        async with credential:
-            client = TableServiceClient(credential=credential, endpoint=account_url, api_version="2020-12-06")
-            with pytest.raises(ClientAuthenticationError):
-                async for _ in client.list_tables():
-                    pass
+        credential = self.generate_fake_token_credential()
+        client = TableServiceClient(credential=credential, endpoint=account_url, api_version="2020-12-06")
+        with pytest.raises(ClientAuthenticationError):
+            async for _ in client.list_tables():
+                pass