From c4efa9d5bed8f37b3b6e4793cf5111dcc9e34b49 Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
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 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):