From 748336d2e9895c5a00cbf34a30121f3b6b79be7a Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Tue, 15 Apr 2025 13:40:54 +0200 Subject: [PATCH] fix(pytest): fix message filters in SessionDebugWrapper --- python/src/trezorlib/debuglink.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/python/src/trezorlib/debuglink.py b/python/src/trezorlib/debuglink.py index 7a3f9be390..a4aaecfdeb 100644 --- a/python/src/trezorlib/debuglink.py +++ b/python/src/trezorlib/debuglink.py @@ -995,10 +995,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() @@ -1037,6 +1042,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,