Fix tests to reflect change in exception type

alan-turing-institute · Nov 15, 2024 · 7ae72ab · 7ae72ab
1 parent 041b5a4
commit 7ae72ab
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/external/api/test_credentials.py b/tests/external/api/test_credentials.py
@@ -4,8 +4,8 @@
     DeviceCodeCredential,
 )
 
+from click.exceptions import Exit
 from data_safe_haven.directories import config_dir
-from data_safe_haven.exceptions import DataSafeHavenAzureError
 from data_safe_haven.external.api.credentials import (
     AzureSdkCredential,
     DeferredCredential,
@@ -37,8 +37,7 @@ def test_confirm_credentials_interactive_fail(
         DeferredCredential.cache_ = set()
         credential = AzureSdkCredential(skip_confirmation=False)
         with pytest.raises(
-            DataSafeHavenAzureError,
-            match="Error getting account information from Azure CLI.",
+            Exit
         ):
             credential.get_credential()
 
@@ -62,8 +61,7 @@ def test_decode_token_error(
     ):
         credential = AzureSdkCredential(skip_confirmation=True)
         with pytest.raises(
-            DataSafeHavenAzureError,
-            match="Error getting account information from Azure CLI.",
+            Exit
         ):
             credential.decode_token(credential.token)