refactor: move initial context setting out of try block

bluesky · Nov 25, 2024 · 39f1d0d · 39f1d0d
1 parent 3011c9c
commit 39f1d0d
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tiled/_tests/utils.py b/tiled/_tests/utils.py
@@ -61,11 +61,12 @@ def enter_username_password(username, password):
     ...     # Run code that calls prompt_for_credentials and subsequently getpass.getpass().
     """
 
+    original_prompt = context.PROMPT_FOR_REAUTHENTICATION
+    original_credentials = context.prompt_for_credentials
+    context.PROMPT_FOR_REAUTHENTICATION = True
+    context.prompt_for_credentials = lambda u, p: (username, password)
     try:
-        original_prompt = context.PROMPT_FOR_REAUTHENTICATION
-        original_credentials = context.prompt_for_credentials
-        context.PROMPT_FOR_REAUTHENTICATION = True
-        context.prompt_for_credentials = lambda u, p: (username, password)
+        # Ensures that raise in calling routine does not prevent context from being exited.
         yield
     finally:
         context.PROMPT_FOR_REAUTHENTICATION = original_prompt