Use clear_default_identity correctly

bluesky · Oct 1, 2024 · c05daac · c05daac
1 parent c009522
commit c05daac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tiled/_tests/test_access_control.py b/tiled/_tests/test_access_control.py
@@ -160,7 +160,9 @@ def test_top_level_access_control(context, enter_username_password):
     with pytest.raises(KeyError):
         bob_client["b"]
     alice_client.logout()
-    bob_client.logout()
+
+    # Make sure clearing default identity works without raising an error.
+    bob_client.logout(clear_default=True)
 
 
 def test_access_control_with_api_key_auth(context, enter_username_password):

diff --git a/tiled/client/context.py b/tiled/client/context.py
@@ -727,7 +727,7 @@ def logout(self, clear_default=False):
 
         # If requested, automatically clear the default identity
         if clear_default:
-            self.clear_default_identity()
+            clear_default_identity(self.api_uri)
 
     def revoke_session(self, session_id):
         """