22import os
33from unittest .mock import patch , MagicMock
44import pytest
5- import helpers .azure_credential_utils as azure_credential_utils
5+ import backend . batch . utilities . helpers .azure_credential_utils as azure_credential_utils
66
7- # Ensure src/backend is on the Python path for imports
87sys .path .insert (0 , os .path .abspath (os .path .join (os .path .dirname (__file__ ), ".." , ".." )))
98
109# Synchronous tests
1110
1211
13- @patch ("helpers.azure_credential_utils.os.getenv" )
14- @patch ("helpers.azure_credential_utils.DefaultAzureCredential" )
15- @patch ("helpers.azure_credential_utils.ManagedIdentityCredential" )
12+ @patch ("backend.batch.utilities.helpers.azure_credential_utils.os.getenv" )
13+ @patch ("backend.batch.utilities.helpers.azure_credential_utils.DefaultAzureCredential" )
14+ @patch (
15+ "backend.batch.utilities.helpers.azure_credential_utils.ManagedIdentityCredential"
16+ )
1617def test_get_azure_credential_dev_env (
1718 mock_managed_identity_credential , mock_default_azure_credential , mock_getenv
1819):
@@ -29,9 +30,11 @@ def test_get_azure_credential_dev_env(
2930 assert credential == mock_default_credential
3031
3132
32- @patch ("helpers.azure_credential_utils.os.getenv" )
33- @patch ("helpers.azure_credential_utils.DefaultAzureCredential" )
34- @patch ("helpers.azure_credential_utils.ManagedIdentityCredential" )
33+ @patch ("backend.batch.utilities.helpers.azure_credential_utils.os.getenv" )
34+ @patch ("backend.batch.utilities.helpers.azure_credential_utils.DefaultAzureCredential" )
35+ @patch (
36+ "backend.batch.utilities.helpers.azure_credential_utils.ManagedIdentityCredential"
37+ )
3538def test_get_azure_credential_non_dev_env (
3639 mock_managed_identity_credential , mock_default_azure_credential , mock_getenv
3740):
@@ -51,9 +54,13 @@ def test_get_azure_credential_non_dev_env(
5154
5255
5356@pytest .mark .asyncio
54- @patch ("helpers.azure_credential_utils.os.getenv" )
55- @patch ("helpers.azure_credential_utils.AioDefaultAzureCredential" )
56- @patch ("helpers.azure_credential_utils.AioManagedIdentityCredential" )
57+ @patch ("backend.batch.utilities.helpers.azure_credential_utils.os.getenv" )
58+ @patch (
59+ "backend.batch.utilities.helpers.azure_credential_utils.AioDefaultAzureCredential"
60+ )
61+ @patch (
62+ "backend.batch.utilities.helpers.azure_credential_utils.AioManagedIdentityCredential"
63+ )
5764async def test_get_azure_credential_async_dev_env (
5865 mock_aio_managed_identity_credential , mock_aio_default_azure_credential , mock_getenv
5966):
@@ -71,9 +78,13 @@ async def test_get_azure_credential_async_dev_env(
7178
7279
7380@pytest .mark .asyncio
74- @patch ("helpers.azure_credential_utils.os.getenv" )
75- @patch ("helpers.azure_credential_utils.AioDefaultAzureCredential" )
76- @patch ("helpers.azure_credential_utils.AioManagedIdentityCredential" )
81+ @patch ("backend.batch.utilities.helpers.azure_credential_utils.os.getenv" )
82+ @patch (
83+ "backend.batch.utilities.helpers.azure_credential_utils.AioDefaultAzureCredential"
84+ )
85+ @patch (
86+ "backend.batch.utilities.helpers.azure_credential_utils.AioManagedIdentityCredential"
87+ )
7788async def test_get_azure_credential_async_non_dev_env (
7889 mock_aio_managed_identity_credential , mock_aio_default_azure_credential , mock_getenv
7990):
0 commit comments