From 1c9aa0cf62ca01b82498723c02c33577a9f9421c Mon Sep 17 00:00:00 2001 From: Zunli Hu Date: Fri, 3 Jul 2020 14:54:42 +0800 Subject: [PATCH 1/2] allow large response --- .../storage/tests/latest/test_storage_account_scenarios.py | 1 + .../storage/tests/latest/test_storage_blob_scenarios.py | 2 ++ .../tests/latest/test_storage_container_immutability_policy.py | 2 ++ .../storage/tests/latest/test_storage_container_legal_hold.py | 2 ++ .../storage/tests/latest/test_storage_file_rm_scenarios.py | 2 ++ 5 files changed, 9 insertions(+) diff --git a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_account_scenarios.py b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_account_scenarios.py index 0c7db3da6e0..bd40b3af77c 100644 --- a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_account_scenarios.py +++ b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_account_scenarios.py @@ -11,6 +11,7 @@ from azure_devtools.scenario_tests import AllowLargeResponse +@AllowLargeResponse() @api_version_constraint(ResourceType.MGMT_STORAGE, min_api='2016-12-01') class StorageAccountTests(StorageScenarioMixin, ScenarioTest): @api_version_constraint(ResourceType.MGMT_STORAGE, min_api='2017-06-01') diff --git a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py index 1061f833348..b8191098d55 100644 --- a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py +++ b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py @@ -14,6 +14,7 @@ from azure.cli.command_modules.storage._client_factory import MISSING_CREDENTIALS_ERROR_MESSAGE from ..storage_test_util import StorageScenarioMixin +from azure_devtools.scenario_tests import AllowLargeResponse @api_version_constraint(ResourceType.MGMT_STORAGE, min_api='2016-12-01') @@ -486,6 +487,7 @@ def test_storage_blob_generate_sas_as_user(self, resource_group, storage_account self.assertIn('skv=', container_sas) self.assertIn('skv=', container_sas) + @AllowLargeResponse() @ResourceGroupPreparer() @StorageAccountPreparer() @api_version_constraint(resource_type=ResourceType.DATA_STORAGE_BLOB, min_api='2019-02-02') diff --git a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_immutability_policy.py b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_immutability_policy.py index 15c5dd10c15..d0ee9fc9256 100644 --- a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_immutability_policy.py +++ b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_immutability_policy.py @@ -5,8 +5,10 @@ from azure.cli.testsdk import (ScenarioTest, JMESPathCheck, ResourceGroupPreparer, StorageAccountPreparer) from msrestazure.azure_exceptions import CloudError +from azure_devtools.scenario_tests import AllowLargeResponse +@AllowLargeResponse() class StorageImmutabilityPolicy(ScenarioTest): @ResourceGroupPreparer() @StorageAccountPreparer(kind="StorageV2") diff --git a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_legal_hold.py b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_legal_hold.py index 538604c9922..40d00b8c422 100644 --- a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_legal_hold.py +++ b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_container_legal_hold.py @@ -4,8 +4,10 @@ # -------------------------------------------------------------------------------------------- from azure.cli.testsdk import (ScenarioTest, JMESPathCheck, ResourceGroupPreparer, StorageAccountPreparer) +from azure_devtools.scenario_tests import AllowLargeResponse +@AllowLargeResponse() class StorageLegalHold(ScenarioTest): @ResourceGroupPreparer() def test_legal_hold(self, resource_group): diff --git a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_file_rm_scenarios.py b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_file_rm_scenarios.py index 4d3c7ad5cd9..b68d3af8c02 100644 --- a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_file_rm_scenarios.py +++ b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_file_rm_scenarios.py @@ -8,8 +8,10 @@ JMESPathCheck) from azure.cli.core.profiles import ResourceType from ..storage_test_util import StorageScenarioMixin +from azure_devtools.scenario_tests import AllowLargeResponse +@AllowLargeResponse() class StorageFileShareRmScenarios(StorageScenarioMixin, ScenarioTest): @api_version_constraint(ResourceType.MGMT_STORAGE, min_api='2019-06-01') @ResourceGroupPreparer(name_prefix="cli", location="eastus") From 2b2563b4837861818a4e333e81753a8cfe08fb07 Mon Sep 17 00:00:00 2001 From: Zunli Hu Date: Fri, 3 Jul 2020 14:59:47 +0800 Subject: [PATCH 2/2] change allowLrageResponse --- .../storage/tests/latest/test_storage_blob_scenarios.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py index b8191098d55..13f800cd30f 100644 --- a/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py +++ b/src/azure-cli/azure/cli/command_modules/storage/tests/latest/test_storage_blob_scenarios.py @@ -17,6 +17,7 @@ from azure_devtools.scenario_tests import AllowLargeResponse +@AllowLargeResponse() @api_version_constraint(ResourceType.MGMT_STORAGE, min_api='2016-12-01') class StorageBlobUploadTests(StorageScenarioMixin, ScenarioTest): @ResourceGroupPreparer() @@ -487,7 +488,6 @@ def test_storage_blob_generate_sas_as_user(self, resource_group, storage_account self.assertIn('skv=', container_sas) self.assertIn('skv=', container_sas) - @AllowLargeResponse() @ResourceGroupPreparer() @StorageAccountPreparer() @api_version_constraint(resource_type=ResourceType.DATA_STORAGE_BLOB, min_api='2019-02-02')