Skip to content

Commit 1f184b4

Browse files
fix defaultcred unittestcase
1 parent a1dfb29 commit 1f184b4

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

src/tests/api/helpers/test_azure_credential_utils.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,12 @@ def mock_env_vars():
1414
}
1515

1616
class TestAzureCredentialUtils:
17-
@patch.dict(os.environ, clear=True)
17+
@patch.dict(os.environ, {}, clear=True)
1818
@patch("helpers.azure_credential_utils.DefaultAzureCredential")
1919
@patch("helpers.azure_credential_utils.ManagedIdentityCredential")
2020
def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_env_vars):
2121
"""Test get_azure_credential in dev environment."""
2222
# Arrange
23-
mock_env_vars["app_env"] = "dev"
2423
os.environ.update(mock_env_vars)
2524
mock_default_credential = MagicMock()
2625
mock_default_azure_credential.return_value = mock_default_credential
@@ -33,7 +32,7 @@ def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mo
3332
mock_managed_identity_credential.assert_not_called()
3433
assert credential == mock_default_credential
3534

36-
@patch.dict(os.environ, clear=True)
35+
@patch.dict(os.environ, {}, clear=True)
3736
@patch("helpers.azure_credential_utils.DefaultAzureCredential")
3837
@patch("helpers.azure_credential_utils.ManagedIdentityCredential")
3938
def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_env_vars):
@@ -53,13 +52,12 @@ def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential
5352
assert credential == mock_managed_credential
5453

5554
@pytest.mark.asyncio
56-
@patch.dict(os.environ, clear=True)
55+
@patch.dict(os.environ, {}, clear=True)
5756
@patch("helpers.azure_credential_utils.AioDefaultAzureCredential")
5857
@patch("helpers.azure_credential_utils.AioManagedIdentityCredential")
5958
async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_env_vars):
6059
"""Test get_azure_credential_async in dev environment."""
6160
# Arrange
62-
mock_env_vars["app_env"] = "dev"
6361
os.environ.update(mock_env_vars)
6462
mock_aio_default_credential = MagicMock()
6563
mock_aio_default_azure_credential.return_value = mock_aio_default_credential
@@ -73,7 +71,7 @@ async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identit
7371
assert credential == mock_aio_default_credential
7472

7573
@pytest.mark.asyncio
76-
@patch.dict(os.environ, clear=True)
74+
@patch.dict(os.environ, {}, clear=True)
7775
@patch("helpers.azure_credential_utils.AioDefaultAzureCredential")
7876
@patch("helpers.azure_credential_utils.AioManagedIdentityCredential")
7977
async def test_get_azure_credential_async_non_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_env_vars):

0 commit comments

Comments
 (0)