Skip to content

Commit a1dfb29

Browse files
fix unittestcase
1 parent 41bf085 commit a1dfb29

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/tests/api/helpers/test_azure_credential_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ class TestAzureCredentialUtils:
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"
2324
os.environ.update(mock_env_vars)
2425
mock_default_credential = MagicMock()
2526
mock_default_azure_credential.return_value = mock_default_credential
@@ -58,6 +59,7 @@ def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential
5859
async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_env_vars):
5960
"""Test get_azure_credential_async in dev environment."""
6061
# Arrange
62+
mock_env_vars["app_env"] = "dev"
6163
os.environ.update(mock_env_vars)
6264
mock_aio_default_credential = MagicMock()
6365
mock_aio_default_azure_credential.return_value = mock_aio_default_credential

0 commit comments

Comments
 (0)