1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
from unittest import mock
import pytest
from mitmproxy import contentviews
from mitmproxy.exceptions import ContentViewException
from mitmproxy.net.http import Headers
from mitmproxy.test import tutils
class TestContentView(contentviews.View):
name = "test"
prompt = ("test", "t")
content_types = ["test/123"]
def test_add_remove():
tcv = TestContentView()
contentviews.add(tcv)
assert tcv in contentviews.views
# repeated addition causes exception
with pytest.raises(ContentViewException, match="Duplicate view"):
contentviews.add(tcv)
tcv2 = TestContentView()
tcv2.name = "test2"
tcv2.prompt = ("test2", "t")
# Same shortcut doesn't work either.
with pytest.raises(ContentViewException, match="Duplicate view shortcut"):
contentviews.add(tcv2)
contentviews.remove(tcv)
assert tcv not in contentviews.views
def test_get_content_view():
desc, lines, err = contentviews.get_content_view(
contentviews.get("Raw"),
b"[1, 2, 3]",
)
assert "Raw" in desc
assert list(lines)
assert not err
desc, lines, err = contentviews.get_content_view(
contentviews.get("Auto"),
b"[1, 2, 3]",
headers=Headers(content_type="application/json")
)
assert desc == "JSON"
assert list(lines)
desc, lines, err = contentviews.get_content_view(
contentviews.get("JSON"),
b"[1, 2",
)
assert "Couldn't parse" in desc
with mock.patch("mitmproxy.contentviews.auto.ViewAuto.__call__") as view_auto:
view_auto.side_effect = ValueError
desc, lines, err = contentviews.get_content_view(
contentviews.get("Auto"),
b"[1, 2",
)
assert err
assert "Couldn't parse" in desc
def test_get_message_content_view():
r = tutils.treq()
desc, lines, err = contentviews.get_message_content_view("raw", r)
assert desc == "Raw"
desc, lines, err = contentviews.get_message_content_view("unknown", r)
assert desc == "Raw"
r.encode("gzip")
desc, lines, err = contentviews.get_message_content_view("raw", r)
assert desc == "[decoded gzip] Raw"
r.headers["content-encoding"] = "deflate"
desc, lines, err = contentviews.get_message_content_view("raw", r)
assert desc == "[cannot decode] Raw"
r.content = None
desc, lines, err = contentviews.get_message_content_view("raw", r)
assert list(lines) == [[("error", "content missing")]]
def test_get_by_shortcut():
assert contentviews.get_by_shortcut("s")
assert not contentviews.get_by_shortcut("b")
|
