Skip to content

Commit dd1ad66

Browse files
fix testcase failure issue
1 parent bb500de commit dd1ad66

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

src/tests/api/helpers/test_azure_credential_utils.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)