Skip to content

Commit 0797935

Browse files
fix APP_ENV environment variable in testcase
1 parent 1f184b4 commit 0797935

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/tests/api/helpers/test_azure_credential_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
@pytest.fixture
1111
def mock_env_vars():
1212
return {
13-
"app_env": "dev"
13+
"APP_ENV": "dev"
1414
}
1515

1616
class TestAzureCredentialUtils:
@@ -38,7 +38,7 @@ def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mo
3838
def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_env_vars):
3939
"""Test get_azure_credential in non-dev environment."""
4040
# Arrange
41-
mock_env_vars["app_env"] = "prod"
41+
mock_env_vars["APP_ENV"] = "prod"
4242
os.environ.update(mock_env_vars)
4343
mock_managed_credential = MagicMock()
4444
mock_managed_identity_credential.return_value = mock_managed_credential
@@ -77,7 +77,7 @@ async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identit
7777
async def test_get_azure_credential_async_non_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_env_vars):
7878
"""Test get_azure_credential_async in non-dev environment."""
7979
# Arrange
80-
mock_env_vars["app_env"] = "prod"
80+
mock_env_vars["APP_ENV"] = "prod"
8181
os.environ.update(mock_env_vars)
8282
mock_aio_managed_credential = MagicMock()
8383
mock_aio_managed_identity_credential.return_value = mock_aio_managed_credential

0 commit comments

Comments
 (0)