diff --git a/python/src/trezorlib/debuglink.py b/python/src/trezorlib/debuglink.py index 707401cf1b..0f81bcfa6a 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 32d0545d1c..99ed7135e7 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):