diff --git a/sdk/identity/azure-identity/tests/integration/test_azure_functions.py b/sdk/identity/azure-identity/tests/integration/test_azure_functions.py index 0b4576c03769..d98733845cf2 100644 --- a/sdk/identity/azure-identity/tests/integration/test_azure_functions.py +++ b/sdk/identity/azure-identity/tests/integration/test_azure_functions.py @@ -5,9 +5,10 @@ import os import pytest -from azure.core import PipelineClient from azure.core.rest import HttpRequest, HttpResponse +from utils import get_pipeline_client + @pytest.fixture(scope="module") def base_url(): @@ -24,7 +25,7 @@ class TestAzureFunctionsIntegration: ) def test_azure_functions_integration_sync(self, base_url): """Test the Azure Functions endpoint where the sync MI credential is used.""" - client = PipelineClient(base_url) + client = get_pipeline_client(base_url) request = HttpRequest("GET", f"{base_url}RunTest") response: HttpResponse = client.send_request(request) assert response.status_code == 200 @@ -35,7 +36,7 @@ def test_azure_functions_integration_sync(self, base_url): ) def test_azure_functions_integration_async(self, base_url): """Test the Azure Functions endpoint where the async MI credential is used.""" - client = PipelineClient(base_url) + client = get_pipeline_client(base_url) request = HttpRequest("GET", f"{base_url}RunAsyncTest") response: HttpResponse = client.send_request(request) assert response.status_code == 200 diff --git a/sdk/identity/azure-identity/tests/integration/test_azure_web_apps.py b/sdk/identity/azure-identity/tests/integration/test_azure_web_apps.py index 43e21e9623f5..dab5ee11702c 100644 --- a/sdk/identity/azure-identity/tests/integration/test_azure_web_apps.py +++ b/sdk/identity/azure-identity/tests/integration/test_azure_web_apps.py @@ -5,9 +5,10 @@ import os import pytest -from azure.core import PipelineClient from azure.core.rest import HttpRequest, HttpResponse +from utils import get_pipeline_client + @pytest.fixture(scope="module") def base_url(): @@ -24,7 +25,7 @@ class TestAzureWebAppsIntegration: ) def test_azure_web_app_integration_sync(self, base_url): """Test the Azure Web App endpoint where the sync MI credential is used.""" - client = PipelineClient(base_url) + client = get_pipeline_client(base_url) request = HttpRequest("GET", f"{base_url}sync") response: HttpResponse = client.send_request(request) assert response.status_code == 200 @@ -35,7 +36,7 @@ def test_azure_web_app_integration_sync(self, base_url): ) def test_azure_web_app_integration_async(self, base_url): """Test the Azure Web App endpoint where the async MI credential is used.""" - client = PipelineClient(base_url) + client = get_pipeline_client(base_url) request = HttpRequest("GET", f"{base_url}async") response: HttpResponse = client.send_request(request) assert response.status_code == 200 diff --git a/sdk/identity/azure-identity/tests/integration/utils.py b/sdk/identity/azure-identity/tests/integration/utils.py index 77349b4c6c99..cccfe2bc9a6f 100644 --- a/sdk/identity/azure-identity/tests/integration/utils.py +++ b/sdk/identity/azure-identity/tests/integration/utils.py @@ -5,6 +5,9 @@ import subprocess import sys +from azure.core import PipelineClient +from azure.core.pipeline import policies + def run_command(command, exit_on_failure=True) -> str: try: @@ -16,3 +19,11 @@ def run_command(command, exit_on_failure=True) -> str: print(result) sys.exit(1) return result + + +def get_pipeline_client(base_url: str) -> PipelineClient: + policy_list = [ + policies.RetryPolicy(), + policies.ContentDecodePolicy(), + ] + return PipelineClient(base_url, policies=policy_list) diff --git a/sdk/identity/test-resources-post.ps1 b/sdk/identity/test-resources-post.ps1 index d81c350e5b42..2344ff6fa68d 100644 --- a/sdk/identity/test-resources-post.ps1 +++ b/sdk/identity/test-resources-post.ps1 @@ -131,6 +131,9 @@ Write-Host "Deploying Azure Container Instance" az container create -g $($DeploymentOutputs['IDENTITY_RESOURCE_GROUP']) -n $($DeploymentOutputs['IDENTITY_CONTAINER_INSTANCE_NAME']) --image $image ` --acr-identity $($DeploymentOutputs['IDENTITY_USER_DEFINED_IDENTITY']) ` --assign-identity [system] $($DeploymentOutputs['IDENTITY_USER_DEFINED_IDENTITY']) ` + --cpu 1 ` + --memory 1.0 ` + --os-type Linux ` --role "Storage Blob Data Reader" ` --scope $($DeploymentOutputs['IDENTITY_STORAGE_ID_1']) ` -e IDENTITY_STORAGE_NAME=$($DeploymentOutputs['IDENTITY_STORAGE_NAME_1']) `