@@ -21,16 +21,17 @@ def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mo
2121 """Test get_azure_credential in dev environment."""
2222 # Arrange
2323 os .environ .update (mock_env_vars )
24- mock_azure_cli_credential = MagicMock ()
25- mock_azure_cli_credential .return_value = mock_azure_cli_credential
24+ mock_credential_instance = MagicMock ()
25+ mock_azure_cli_credential .return_value = mock_credential_instance
2626
2727 # Act
2828 credential = azure_credential_utils .get_azure_credential ()
2929
3030 # Assert
3131 mock_azure_cli_credential .assert_called_once ()
3232 mock_managed_identity_credential .assert_not_called ()
33- assert credential == mock_azure_cli_credential
33+ assert credential == mock_credential_instance
34+
3435
3536 @patch .dict (os .environ , {}, clear = True )
3637 @patch ("helpers.azure_credential_utils.AzureCliCredential" )
@@ -59,16 +60,16 @@ async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identit
5960 """Test get_azure_credential_async in dev environment."""
6061 # Arrange
6162 os .environ .update (mock_env_vars )
62- mock_aio_azure_cli_credential = MagicMock ()
63- mock_aio_azure_cli_credential .return_value = mock_aio_azure_cli_credential
63+ mock_credential_instance = MagicMock ()
64+ mock_aio_azure_cli_credential .return_value = mock_credential_instance
6465
6566 # Act
6667 credential = await azure_credential_utils .get_azure_credential_async ()
6768
6869 # Assert
6970 mock_aio_azure_cli_credential .assert_called_once ()
7071 mock_aio_managed_identity_credential .assert_not_called ()
71- assert credential == mock_aio_azure_cli_credential
72+ assert credential == mock_credential_instance
7273
7374 @pytest .mark .asyncio
7475 @patch .dict (os .environ , {}, clear = True )
0 commit comments