diff options
author | Arushit Mudgal <rshtmudgal@gmail.com> | 2018-01-25 20:36:19 +0530 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2018-01-25 16:06:19 +0100 |
commit | f41d521ce51084f37c094a7a5368b77a7d0cd225 (patch) | |
tree | 9947db80c55bd891bffaceef670d37811780b411 | |
parent | 6dd336fcec6dc32c4986b6c20189bf1b2132153c (diff) | |
download | mitmproxy-f41d521ce51084f37c094a7a5368b77a7d0cd225.tar.gz mitmproxy-f41d521ce51084f37c094a7a5368b77a7d0cd225.tar.bz2 mitmproxy-f41d521ce51084f37c094a7a5368b77a7d0cd225.zip |
fix #2800
-rw-r--r-- | mitmproxy/contentviews/base.py | 5 | ||||
-rw-r--r-- | test/mitmproxy/contentviews/test_base.py | 18 |
2 files changed, 20 insertions, 3 deletions
diff --git a/mitmproxy/contentviews/base.py b/mitmproxy/contentviews/base.py index bdab1e99..dbaa6ccc 100644 --- a/mitmproxy/contentviews/base.py +++ b/mitmproxy/contentviews/base.py @@ -49,8 +49,9 @@ def format_dict( ] entries, where key is padded to a uniform width. """ - max_key_len = max(len(k) for k in d.keys()) - max_key_len = min(max_key_len, KEY_MAX) + + max_key_len = max((len(k) for k in d.keys()), default=0) + max_key_len = min((max_key_len, KEY_MAX), default=0) for key, value in d.items(): if isinstance(key, bytes): key += b":" diff --git a/test/mitmproxy/contentviews/test_base.py b/test/mitmproxy/contentviews/test_base.py index 777ab4dd..c94d8be2 100644 --- a/test/mitmproxy/contentviews/test_base.py +++ b/test/mitmproxy/contentviews/test_base.py @@ -1 +1,17 @@ -# TODO: write tests +import pytest +from mitmproxy.contentviews import base + + +def test_format_dict(): + d = {"one": "two", "three": "four"} + f_d = base.format_dict(d) + assert next(f_d) + + d = {"adsfa": ""} + f_d = base.format_dict(d) + assert next(f_d) + + d = {} + f_d = base.format_dict(d) + with pytest.raises(StopIteration): + next(f_d) |