arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-18 05:28:40 +00:00
Code Issues Releases Wiki Activity

feat(python/debuglink): make input flows part of the traceback

Browse Source
This commit is contained in:
matejcik 2023-11-15 16:10:40 +01:00 committed by matejcik
parent 8c9f5ebbca
commit 863435cc9d
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1114,6 +1114,13 @@ class TrezorClientDebugLink(TrezorClient):
# copy expected/actual responses before clearing them # copy expected/actual responses before clearing them
expected_responses = self.expected_responses expected_responses = self.expected_responses
actual_responses = self.actual_responses actual_responses = self.actual_responses
# grab a copy of the inputflow generator to raise an exception through it
if isinstance(self.ui, DebugUI):
input_flow = self.ui.input_flow
else:
input_flow = None
self.reset_debug_features() self.reset_debug_features()
if exc_type is None: if exc_type is None:
@ -1121,6 +1128,11 @@ class TrezorClientDebugLink(TrezorClient):
# (raises AssertionError on mismatch) # (raises AssertionError on mismatch)
self._verify_responses(expected_responses, actual_responses) self._verify_responses(expected_responses, actual_responses)
elif isinstance(input_flow, Generator):
# Propagate the exception through the input flow, so that we see in
# traceback where it is stuck.
input_flow.throw(exc_type, value, traceback)
def set_expected_responses( def set_expected_responses(
self, expected: List[Union["ExpectedMessage", Tuple[bool, "ExpectedMessage"]]] self, expected: List[Union["ExpectedMessage", Tuple[bool, "ExpectedMessage"]]]
) -> None: ) -> None: