From 122235bf11d83aedb1e5a769f03d6a39477957d1 Mon Sep 17 00:00:00 2001 From: Stephen Rosen Date: Fri, 6 Dec 2024 17:06:51 -0600 Subject: [PATCH] Fix mocking of internal auth client --- tests/functional/test_login_command.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/functional/test_login_command.py b/tests/functional/test_login_command.py index 0f5888b6..886a32b2 100644 --- a/tests/functional/test_login_command.py +++ b/tests/functional/test_login_command.py @@ -16,7 +16,9 @@ def test_login_validates_token( # undo the validate_token disabling patch which is done for most tests disable_login_manager_validate_token.undo() - with mock.patch("globus_cli.login_manager.manager.internal_auth_client") as m: + with mock.patch( + "globus_cli.login_manager.tokenstore.CLITokenstorage.internal_auth_client" + ) as m: ac = mock.MagicMock(spec=globus_sdk.ConfidentialAppAuthClient) m.return_value = ac @@ -91,7 +93,8 @@ def test_login_gcs_different_identity( ) monkeypatch.setattr( - "globus_cli.commands.logout.internal_native_client", lambda: mock_auth_client + "globus_cli.login_manager.tokenstore.CLITokenstorage.internal_auth_client", + mock_auth_client, ) run_line("globus logout --yes") assert manager.token_storage.read_well_known_config("auth_user_data") is None