diff --git a/sdk/tables/azure-data-tables/tests/_shared/testcase.py b/sdk/tables/azure-data-tables/tests/_shared/testcase.py index 8c855e220e7..5dd2263bc14 100644 --- a/sdk/tables/azure-data-tables/tests/_shared/testcase.py +++ b/sdk/tables/azure-data-tables/tests/_shared/testcase.py @@ -26,7 +26,7 @@ import re import logging from devtools_testutils import ( - AzureMgmtTestCase, + AzureTestCase, AzureMgmtPreparer, ResourceGroupPreparer, StorageAccountPreparer, @@ -140,7 +140,7 @@ def create_resource(self, name, **kwargs): } -class TableTestCase(AzureMgmtTestCase): +class TableTestCase(AzureTestCase): def __init__(self, *args, **kwargs): super(TableTestCase, self).__init__(*args, **kwargs) @@ -341,7 +341,7 @@ def __exit__(self, exc_type, exc_val, exc_tb): @pytest.fixture(scope="session") def storage_account(): - test_case = AzureMgmtTestCase("__init__") + test_case = AzureTestCase("__init__") rg_preparer = ResourceGroupPreparer(random_name_enabled=True, name_prefix='pystorage') storage_preparer = StorageAccountPreparer(random_name_enabled=True, name_prefix='pyacrstorage') cosmos_preparer = CosmosAccountPreparer(random_name_enabled=True, name_prefix='pycosmos')