How to Mock Azure Key Vault Secret Get in Python

import unittest
from unittest.mock import MagicMock, patch


def get_secret(key_vault_client, secret_name):
    """Retrieve a secret value from an Azure Key Vault client."""
    secret = key_vault_client.get_secret(secret_name)
    return secret.value


class TestKeyVaultSecretGet(unittest.TestCase):
    def test_get_mocked_secret(self):
        # Create a real mock of the Azure Key Vault client
        mock_client = MagicMock()

        # Configure the mock to return a specific secret value
        mock_secret = MagicMock()
        mock_secret.value = "mock-secret-value-123"
        mock_client.get_secret.return_value = mock_secret

        # Call the function under test
        result = get_secret(mock_client, "my-test-secret")

        # Assert the result
        self.assertEqual(result, "mock-secret-value-123")
        mock_client.get_secret.assert_called_once_with("my-test-secret")

    def test_get_secret_with_patch(self):
        # Alternative: patch the client before calling
        with patch("__main__.get_secret") as mock_get:
            mock_get.return_value = "patched-secret"
            result = get_secret(MagicMock(), "another-secret")
            self.assertEqual(result, "patched-secret")


if __name__ == "__main__":
    unittest.main()