From c0017eaed90c352a3aabe07c12af7c66eaa4c303 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Thu, 5 Dec 2024 11:52:04 +0100 Subject: [PATCH] feat(python): add lock() to TrezorClientDebugLink [no changelog] --- python/src/trezorlib/debuglink.py | 4 ++++ tests/conftest.py | 1 + 2 files changed, 5 insertions(+) diff --git a/python/src/trezorlib/debuglink.py b/python/src/trezorlib/debuglink.py index 5b446c8130..c9054070db 100644 --- a/python/src/trezorlib/debuglink.py +++ b/python/src/trezorlib/debuglink.py @@ -1437,6 +1437,10 @@ class TrezorClientDebugLink(TrezorClient): # TODO is this needed? # self.debug.close() + def lock(self) -> None: + s = SessionDebugWrapper(self.get_management_session()) + s.lock() + def get_session( self, passphrase: str | object | None = "", diff --git a/tests/conftest.py b/tests/conftest.py index 9df5463d76..cad38a337f 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -401,6 +401,7 @@ def client( if use_passphrase and isinstance(setup_params["passphrase"], str): _raw_client.use_passphrase(setup_params["passphrase"]) + _raw_client.lock() # TODO _raw_client.clear_session() with ui_tests.screen_recording(_raw_client, request):