From df8966591939e6aa4f747429b762ec33a6721543 Mon Sep 17 00:00:00 2001 From: Yujong Lee Date: Mon, 21 Sep 2026 20:37:57 +0000 Subject: [PATCH] test: narrow Azure parity exception assertion Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../secret_managers/test_secret_manager_handler.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/secret_managers/test_secret_manager_handler.py b/tests/test_litellm/secret_managers/test_secret_manager_handler.py index 0925198992b..b4838912d27 100644 --- a/tests/test_litellm/secret_managers/test_secret_manager_handler.py +++ b/tests/test_litellm/secret_managers/test_secret_manager_handler.py @@ -89,7 +89,9 @@ def test_azure_key_vault_matches_rust_parity_fixture() -> None: value=secret, ) if case.expected.missing or case.expected.error: - with pytest.raises(Exception): + with pytest.raises( + AzureResourceNotFoundError if case.expected.missing else AzureHttpResponseError + ): get_secret_from_manager( secret_name=case.secret_name, key_manager=KeyManagementSystem.AZURE_KEY_VAULT.value,