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

fix(pytest): fix message filters in SessionDebugWrapper

Browse Source
This commit is contained in:
M1nd3r 2025-04-15 13:40:54 +02:00
parent 108e80a90b
commit 6675114b99
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1024,10 +1024,15 @@ class SessionDebugWrapper(Session):
return self.client
def _write(self, msg: t.Any) -> None:
if isinstance(self.client, TrezorClientDebugLink):
msg = self.client._filter_message(msg)
self._session._write(msg)
def _read(self) -> t.Any:
return self._session._read()
msg = self._session._read()
if isinstance(self.client, TrezorClientDebugLink):
msg = self.client._filter_message(msg)
return msg
def resume(self) -> None:
self._session.resume()
@ -1066,6 +1071,10 @@ class TrezorClientDebugLink(TrezorClient):
protocol: ProtocolV1Channel | ProtocolV2Channel
actual_responses: list[protobuf.MessageType] | None = None
filters: t.Dict[
t.Type[protobuf.MessageType],
t.Callable[[protobuf.MessageType], protobuf.MessageType] | None,
] = {}
def __init__(
self,