@@ -14,13 +14,12 @@ def mock_env_vars():
1414 }
1515
1616class 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