From df6595fb471ec724a5cc2fbeee195ba74caf06e6 Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
Date: Wed, 27 Nov 2024 11:28:17 +0100
Subject: [PATCH] test: fix client.reset_debug_features for codec [no
 changelog]

---
 python/src/trezorlib/debuglink.py | 10 +++++-----
 tests/conftest.py                 |  2 +-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/python/src/trezorlib/debuglink.py b/python/src/trezorlib/debuglink.py
index dbf416f244..5145639b98 100644
--- a/python/src/trezorlib/debuglink.py
+++ b/python/src/trezorlib/debuglink.py
@@ -1301,7 +1301,7 @@ class TrezorClientDebugLink(TrezorClient):
         self.transport = transport
         self.ui: DebugUI = DebugUI(self.debug)
 
-        self.reset_debug_features()
+        self.reset_debug_features(new_management_session=True)
         self.sync_responses()
         # So that we can choose right screenshotting logic (T1 vs TT)
         # and know the supported debug capabilities
@@ -1324,7 +1324,7 @@ class TrezorClientDebugLink(TrezorClient):
     def get_new_client(self) -> TrezorClientDebugLink:
         return TrezorClientDebugLink(self.transport, self.debug.allow_interactions)
 
-    def reset_debug_features(self) -> None:
+    def reset_debug_features(self, new_management_session: bool = False) -> None:
         """
         Prepare the debugging client for a new testcase.
 
@@ -1339,8 +1339,8 @@ class TrezorClientDebugLink(TrezorClient):
             t.Type[protobuf.MessageType],
             t.Callable[[protobuf.MessageType], protobuf.MessageType] | None,
         ] = {}
-
-        self._management_session = self.get_management_session(new_session=True)
+        if new_management_session:
+            self._management_session = self.get_management_session(new_session=True)
 
     @property
     def button_callback(self):
@@ -1531,7 +1531,7 @@ class TrezorClientDebugLink(TrezorClient):
         else:
             input_flow = None
 
-        self.reset_debug_features()
+        self.reset_debug_features(new_management_session=False)
 
         if exc_type is None:
             # If no other exception was raised, evaluate missed responses
diff --git a/tests/conftest.py b/tests/conftest.py
index 41c6c4917e..9dca5c8b65 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -291,7 +291,7 @@ def client(
 
     test_ui = request.config.getoption("ui")
 
-    _raw_client.reset_debug_features()
+    _raw_client.reset_debug_features(new_management_session=True)
     _raw_client.open()
     if isinstance(_raw_client.protocol, ProtocolV1):
         try: