arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-01 10:20:59 +00:00
Code Issues Releases Wiki Activity

test: fix client.reset_debug_features for codec

Browse Source 
[no changelog]
This commit is contained in:
M1nd3r 2024-11-27 11:28:17 +01:00
parent c27e3d63b7
commit df6595fb47
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/src/trezorlib/debuglink.py
View File

@ -1301,7 +1301,7 @@ class TrezorClientDebugLink(TrezorClient):
self.transport = transport self.transport = transport
self.ui: DebugUI = DebugUI(self.debug) self.ui: DebugUI = DebugUI(self.debug)
self.reset_debug_features() self.reset_debug_features(new_management_session=True)
self.sync_responses() self.sync_responses()
# So that we can choose right screenshotting logic (T1 vs TT) # So that we can choose right screenshotting logic (T1 vs TT)
# and know the supported debug capabilities # and know the supported debug capabilities
@ -1324,7 +1324,7 @@ class TrezorClientDebugLink(TrezorClient):
def get_new_client(self) -> TrezorClientDebugLink: def get_new_client(self) -> TrezorClientDebugLink:
return TrezorClientDebugLink(self.transport, self.debug.allow_interactions) 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. Prepare the debugging client for a new testcase.
@ -1339,7 +1339,7 @@ class TrezorClientDebugLink(TrezorClient):
t.Type[protobuf.MessageType], t.Type[protobuf.MessageType],
t.Callable[[protobuf.MessageType], protobuf.MessageType] | None, t.Callable[[protobuf.MessageType], protobuf.MessageType] | None,
] = {} ] = {}
if new_management_session:
self._management_session = self.get_management_session(new_session=True) self._management_session = self.get_management_session(new_session=True)
@property @property
@ -1531,7 +1531,7 @@ class TrezorClientDebugLink(TrezorClient):
else: else:
input_flow = None input_flow = None
self.reset_debug_features() self.reset_debug_features(new_management_session=False)
if exc_type is None: if exc_type is None:
# If no other exception was raised, evaluate missed responses # If no other exception was raised, evaluate missed responses

2
tests/conftest.py
View File

@ -291,7 +291,7 @@ def client(
test_ui = request.config.getoption("ui") test_ui = request.config.getoption("ui")
_raw_client.reset_debug_features() _raw_client.reset_debug_features(new_management_session=True)
_raw_client.open() _raw_client.open()
if isinstance(_raw_client.protocol, ProtocolV1): if isinstance(_raw_client.protocol, ProtocolV1):
try: try: