arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-23 06:48:16 +00:00
Code Issues Releases Wiki Activity

tests: limit number of reported expected responses

Browse Source 
so that our test logs are not spammed, esp. in that one test that has
20k messages
This commit is contained in:
matejcik 2018-12-13 13:35:35 +01:00
parent b0a227866a
commit bfb6a196c0
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
trezorlib/debuglink.py
View File

@ -22,6 +22,8 @@ from . import messages as proto, protobuf, tools
from .client import TrezorClient from .client import TrezorClient
from .tools import expect from .tools import expect
EXPECTED_RESPONSES_CONTEXT_LINES = 3
class DebugLink: class DebugLink:
def __init__(self, transport): def __init__(self, transport):
@ -289,9 +291,17 @@ class TrezorClientDebugLink(TrezorClient):
def _raise_unexpected_response(self, msg): def _raise_unexpected_response(self, msg):
__tracebackhide__ = True # for pytest # pylint: disable=W0612 __tracebackhide__ = True # for pytest # pylint: disable=W0612
start_at = max(self.current_response - EXPECTED_RESPONSES_CONTEXT_LINES, 0)
stop_at = min(
self.current_response + EXPECTED_RESPONSES_CONTEXT_LINES + 1,
len(self.expected_responses),
)
output = [] output = []
output.append("Expected responses:") output.append("Expected responses:")
for i, exp in enumerate(self.expected_responses): if start_at > 0:
output.append(" (...{} previous responses omitted)".format(start_at))
for i in range(start_at, stop_at):
exp = self.expected_responses[i]
prefix = " " if i != self.current_response else ">>> " prefix = " " if i != self.current_response else ">>> "
set_fields = { set_fields = {
key: value key: value
@ -304,10 +314,15 @@ class TrezorClientDebugLink(TrezorClient):
"{}{}({})".format(prefix, exp.__class__.__name__, oneline_str) "{}{}({})".format(prefix, exp.__class__.__name__, oneline_str)
) )
else: else:
output.append("{}{}(".format(prefix, exp.__class__.__name__)) item = []
item.append("{}{}(".format(prefix, exp.__class__.__name__))
for key, value in set_fields.items(): for key, value in set_fields.items():
output.append("{} {}={!r}".format(prefix, key, value)) item.append("{} {}={!r}".format(prefix, key, value))
output.append("{})".format(prefix)) item.append("{})".format(prefix))
output.append("\n".join(item))
if stop_at < len(self.expected_responses):
omitted = len(self.expected_responses) - stop_at
output.append(" (...{} following responses omitted)".format(omitted))
output.append("") output.append("")
if msg is not None: if msg is not None: