22import sys
33import os
44from unittest .mock import patch , MagicMock
5+ import backend .helpers .azure_credential_utils as azure_credential_utils
56
67# Ensure src/backend is on the Python path for imports
78sys .path .insert (0 , os .path .abspath (os .path .join (os .path .dirname (__file__ ), '..' , '..' )))
89
9- import helpers .azure_credential_utils as azure_credential_utils
10-
1110# Synchronous tests
1211
12+
1313@patch ("helpers.azure_credential_utils.os.getenv" )
1414@patch ("helpers.azure_credential_utils.DefaultAzureCredential" )
1515@patch ("helpers.azure_credential_utils.ManagedIdentityCredential" )
@@ -26,6 +26,7 @@ def test_get_azure_credential_dev_env(mock_managed_identity_credential, mock_def
2626 mock_managed_identity_credential .assert_not_called ()
2727 assert credential == mock_default_credential
2828
29+
2930@patch ("helpers.azure_credential_utils.os.getenv" )
3031@patch ("helpers.azure_credential_utils.DefaultAzureCredential" )
3132@patch ("helpers.azure_credential_utils.ManagedIdentityCredential" )
@@ -43,6 +44,7 @@ def test_get_azure_credential_non_dev_env(mock_managed_identity_credential, mock
4344
4445# Asynchronous tests
4546
47+
4648@pytest .mark .asyncio
4749@patch ("helpers.azure_credential_utils.os.getenv" )
4850@patch ("helpers.azure_credential_utils.AioDefaultAzureCredential" )
@@ -60,6 +62,7 @@ async def test_get_azure_credential_async_dev_env(mock_aio_managed_identity_cred
6062 mock_aio_managed_identity_credential .assert_not_called ()
6163 assert credential == mock_aio_default_credential
6264
65+
6366@pytest .mark .asyncio
6467@patch ("helpers.azure_credential_utils.os.getenv" )
6568@patch ("helpers.azure_credential_utils.AioDefaultAzureCredential" )
@@ -75,4 +78,4 @@ async def test_get_azure_credential_async_non_dev_env(mock_aio_managed_identity_
7578 mock_getenv .assert_called_once_with ("APP_ENV" , "prod" )
7679 mock_aio_managed_identity_credential .assert_called_once_with (client_id = "test-client-id" )
7780 mock_aio_default_azure_credential .assert_not_called ()
78- assert credential == mock_aio_managed_credential
81+ assert credential == mock_aio_managed_credential
0 commit comments