Add batch tests (Azure#31788)

Author: Yalin Li

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

@@ -14,7 +14,7 @@
 from azure.core import MatchConditions
 from azure.core.pipeline.policies import HTTPPolicy
 from azure.core.credentials import AzureNamedKeyCredential, AzureSasCredential
-from azure.core.exceptions import ResourceNotFoundError, ClientAuthenticationError
+from azure.core.exceptions import ResourceNotFoundError, ClientAuthenticationError, HttpResponseError
 from azure.data.tables import (
     EdmType,
     EntityProperty,
@@ -965,6 +965,30 @@ def test_batch_with_specialchar_partitionkey(self, tables_storage_account_name,
         finally:
             self._tear_down()
 
+    # Playback doesn't work as test proxy issue: https://github.com/Azure/azure-sdk-tools/issues/2900
+    @pytest.mark.live_test_only
+    @tables_decorator
+    @recorded_by_proxy
+    def test_client_with_url_ends_with_table_name(
+        self, tables_storage_account_name, tables_primary_storage_account_key
+    ):
+        url = self.account_url(tables_storage_account_name, "table")
+        table_name = self.get_resource_name("mytable")
+        invalid_url = url + "/" + table_name
+        entity = {"PartitionKey": "test-partition", "RowKey": "test-key", "name": "test-name"}
+
+        valid_tc = TableClient(url, table_name, credential=tables_primary_storage_account_key)
+        valid_tc.create_table()
+
+        tc = TableClient(invalid_url, table_name, credential=tables_primary_storage_account_key)
+        with pytest.raises(HttpResponseError) as ex:
+            tc.submit_transaction([("upsert", entity)])
+        assert "None of the provided media types are supported" in str(ex.value)
+        assert ex.value.error_code == "MediaTypeNotSupported"
+        assert ex.value.status_code == 415
+
+        valid_tc.delete_table()
+
 
 class RequestCorrect(Exception):
     pass

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

@@ -17,7 +17,7 @@
 from azure.core import MatchConditions
 from azure.core.pipeline.policies import AsyncHTTPPolicy
 from azure.core.credentials import AzureSasCredential, AzureNamedKeyCredential
-from azure.core.exceptions import ResourceNotFoundError, ClientAuthenticationError
+from azure.core.exceptions import ResourceNotFoundError, ClientAuthenticationError, HttpResponseError
 from azure.data.tables.aio import TableServiceClient, TableClient
 from azure.data.tables import (
     TableEntity,
@@ -889,6 +889,32 @@ async def generate_entities(count):
         finally:
             await self._tear_down()
 
+    # Playback doesn't work as test proxy issue: https://github.com/Azure/azure-sdk-tools/issues/2900
+    @pytest.mark.live_test_only
+    @tables_decorator_async
+    @recorded_by_proxy_async
+    async def test_client_with_url_ends_with_table_name(
+        self, tables_storage_account_name, tables_primary_storage_account_key
+    ):
+        url = self.account_url(tables_storage_account_name, "table")
+        table_name = self.get_resource_name("mytable")
+        invalid_url = url + "/" + table_name
+        entity = {"PartitionKey": "test-partition", "RowKey": "test-key", "name": "test-name"}
+
+        valid_tc = TableClient(url, table_name, credential=tables_primary_storage_account_key)
+        await valid_tc.create_table()
+
+        tc = TableClient(invalid_url, table_name, credential=tables_primary_storage_account_key)
+        with pytest.raises(HttpResponseError) as ex:
+            await tc.submit_transaction([("upsert", entity)])
+        assert "None of the provided media types are supported" in str(ex.value)
+        assert ex.value.error_code == "MediaTypeNotSupported"
+        assert ex.value.status_code == 415
+
+        await valid_tc.delete_table()
+        await valid_tc.close()
+        await tc.close()
+
 
 class RequestCorrect(Exception):
     pass

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

@@ -13,7 +13,7 @@
 
 from azure.core import MatchConditions
 from azure.core.credentials import AzureSasCredential
-from azure.core.exceptions import ResourceNotFoundError
+from azure.core.exceptions import ResourceNotFoundError, HttpResponseError
 from azure.data.tables import (
     EdmType,
     EntityProperty,
@@ -690,6 +690,28 @@ def test_batch_with_specialchar_partitionkey(self, tables_cosmos_account_name, t
         finally:
             self._tear_down()
 
+    # Playback doesn't work as test proxy issue: https://github.com/Azure/azure-sdk-tools/issues/2900
+    @pytest.mark.live_test_only
+    @cosmos_decorator
+    @recorded_by_proxy
+    def test_client_with_url_ends_with_table_name(self, tables_cosmos_account_name, tables_primary_cosmos_account_key):
+        url = self.account_url(tables_cosmos_account_name, "cosmos")
+        table_name = self.get_resource_name("mytable")
+        invalid_url = url + "/" + table_name
+        entity = {"PartitionKey": "test-partition", "RowKey": "test-key", "name": "test-name"}
+
+        valid_tc = TableClient(url, table_name, credential=tables_primary_cosmos_account_key)
+        valid_tc.create_table()
+
+        tc = TableClient(invalid_url, table_name, credential=tables_primary_cosmos_account_key)
+        with pytest.raises(HttpResponseError) as ex:
+            tc.submit_transaction([("upsert", entity)])
+        assert "Request url is invalid" in str(ex.value)
+        assert ex.value.error_code == "InvalidInput"
+        assert ex.value.status_code == 400
+
+        valid_tc.delete_table()
+
 
 class TestBatchCosmosUnitTests(TableTestCase):
     tables_cosmos_account_name = "fake_cosmos_account"

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

@@ -17,6 +17,7 @@
 from azure.core.exceptions import (
     ResourceNotFoundError,
     ClientAuthenticationError,
+    HttpResponseError,
 )
 from azure.data.tables.aio import TableServiceClient, TableClient
 from azure.data.tables import (
@@ -796,6 +797,32 @@ async def generate_entities(count):
         finally:
             await self._tear_down()
 
+    # Playback doesn't work as test proxy issue: https://github.com/Azure/azure-sdk-tools/issues/2900
+    @pytest.mark.live_test_only
+    @cosmos_decorator_async
+    @recorded_by_proxy_async
+    async def test_client_with_url_ends_with_table_name(
+        self, tables_cosmos_account_name, tables_primary_cosmos_account_key
+    ):
+        url = self.account_url(tables_cosmos_account_name, "cosmos")
+        table_name = self.get_resource_name("mytable")
+        invalid_url = url + "/" + table_name
+        entity = {"PartitionKey": "test-partition", "RowKey": "test-key", "name": "test-name"}
+
+        valid_tc = TableClient(url, table_name, credential=tables_primary_cosmos_account_key)
+        await valid_tc.create_table()
+
+        tc = TableClient(invalid_url, table_name, credential=tables_primary_cosmos_account_key)
+        with pytest.raises(HttpResponseError) as ex:
+            await tc.submit_transaction([("upsert", entity)])
+        assert "Request url is invalid" in str(ex.value)
+        assert ex.value.error_code == "InvalidInput"
+        assert ex.value.status_code == 400
+
+        await valid_tc.delete_table()
+        await valid_tc.close()
+        await tc.close()
+
 
 class TestBatchCosmosAsyncUnitTests(AsyncTableTestCase):
     tables_cosmos_account_name = "fake_cosmos_account"

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

@@ -191,6 +191,9 @@ async def test_client_with_url_ends_with_table_name(
         assert ("URI does not match number of key properties for the resource") in str(exc.value)
         assert ("Please check your account URL.") in str(exc.value)
         await valid_tc.delete_table()
+        await valid_tc.close()
+        await tc.close()
+        await tc2.close()
 
     @tables_decorator_async
     @recorded_by_proxy_async

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

@@ -276,3 +276,6 @@ async def test_client_with_url_ends_with_table_name(
         assert ("Request url is invalid") in str(exc.value)
         assert ("Please check your account URL.") in str(exc.value)
         await valid_tc.delete_table()
+        await valid_tc.close()
+        await tc.close()
+        await tc2.close()