def test_get_secret_value_not_found(self, mock_client): """ Test that if a non-existent secret returns None """ mock_client.get_secret.side_effect = ResourceNotFoundError backend = AzureKeyVaultBackend() assert (backend._get_secret(path_prefix=backend.connections_prefix, secret_id='test_non_existent') is None)
def test_get_secret_value(self, mock_client): """ Test that get_secret returns the secret value """ mock_client.get_secret.return_value = mock.Mock(value='super-secret') backend = AzureKeyVaultBackend() secret_val = backend._get_secret('af-secrets', 'test_mysql_password') mock_client.get_secret.assert_called_with(name='af-secrets-test-mysql-password') self.assertEqual(secret_val, 'super-secret')