diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2015-05-30 12:03:28 +1200 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2015-05-30 12:03:28 +1200 |
| commit | a05a70d8168a07c92b2a3ecbbb1958d85532efe3 (patch) | |
| tree | bc2cc0b89d05e8213280a460cbf25978583c8337 /test/test_console_contentview.py | |
| parent | 1a106f4080d89a236a193f2cb007a00aa195a1ad (diff) | |
| download | mitmproxy-a05a70d8168a07c92b2a3ecbbb1958d85532efe3.tar.gz mitmproxy-a05a70d8168a07c92b2a3ecbbb1958d85532efe3.tar.bz2 mitmproxy-a05a70d8168a07c92b2a3ecbbb1958d85532efe3.zip | |
Add coding style check, reformat.
Diffstat (limited to 'test/test_console_contentview.py')
| -rw-r--r-- | test/test_console_contentview.py | 163 |
1 files changed, 81 insertions, 82 deletions
diff --git a/test/test_console_contentview.py b/test/test_console_contentview.py index 83dbbb8f..a296376e 100644 --- a/test/test_console_contentview.py +++ b/test/test_console_contentview.py @@ -31,40 +31,39 @@ class TestContentView: def test_view_auto(self): v = cv.ViewAuto() f = v( - odict.ODictCaseless(), - "foo", - 1000 - ) + odict.ODictCaseless(), + "foo", + 1000 + ) assert f[0] == "Raw" f = v( - odict.ODictCaseless( - [["content-type", "text/html"]], - ), - "<html></html>", - 1000 - ) + odict.ODictCaseless( + [["content-type", "text/html"]], + ), + "<html></html>", + 1000 + ) assert f[0] == "HTML" f = v( - odict.ODictCaseless( - [["content-type", "text/flibble"]], - ), - "foo", - 1000 - ) + odict.ODictCaseless( + [["content-type", "text/flibble"]], + ), + "foo", + 1000 + ) assert f[0] == "Raw" f = v( - odict.ODictCaseless( - [["content-type", "text/flibble"]], - ), - "<xml></xml>", - 1000 - ) + odict.ODictCaseless( + [["content-type", "text/flibble"]], + ), + "<xml></xml>", + 1000 + ) assert f[0].startswith("XML") - def test_view_urlencoded(self): d = utils.urlencode([("one", "two"), ("three", "four")]) v = cv.ViewURLEncoded() @@ -91,7 +90,7 @@ class TestContentView: v = cv.ViewJSON() assert v([], "{}", 1000) assert not v([], "{", 1000) - assert v([], "[" + ",".join(["0"]*cv.VIEW_CUTOFF) + "]", 1000) + assert v([], "[" + ",".join(["0"] * cv.VIEW_CUTOFF) + "]", 1000) assert v([], "[1, 2, 3, 4, 5]", 5) def test_view_xml(self): @@ -145,18 +144,18 @@ class TestContentView: def test_view_image(self): v = cv.ViewImage() p = tutils.test_data.path("data/image.png") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) p = tutils.test_data.path("data/image.gif") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) p = tutils.test_data.path("data/image-err1.jpg") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) p = tutils.test_data.path("data/image.ico") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) - assert not v([], "flibble", sys.maxint) + assert not v([], "flibble", sys.maxsize) def test_view_multipart(self): view = cv.ViewMultipart() @@ -187,71 +186,70 @@ Larry def test_get_content_view(self): r = cv.get_content_view( - cv.get("Raw"), - [["content-type", "application/json"]], - "[1, 2, 3]", - 1000, - lambda x, l: None, - False - ) + cv.get("Raw"), + [["content-type", "application/json"]], + "[1, 2, 3]", + 1000, + lambda x, l: None, + False + ) assert "Raw" in r[0] r = cv.get_content_view( - cv.get("Auto"), - [["content-type", "application/json"]], - "[1, 2, 3]", - 1000, - lambda x, l: None, - False - ) + cv.get("Auto"), + [["content-type", "application/json"]], + "[1, 2, 3]", + 1000, + lambda x, l: None, + False + ) assert r[0] == "JSON" r = cv.get_content_view( - cv.get("Auto"), - [["content-type", "application/json"]], - "[1, 2", - 1000, - lambda x, l: None, - False - ) + cv.get("Auto"), + [["content-type", "application/json"]], + "[1, 2", + 1000, + lambda x, l: None, + False + ) assert "Raw" in r[0] r = cv.get_content_view( - cv.get("AMF"), - [], - "[1, 2", - 1000, - lambda x, l: None, - False - ) + cv.get("AMF"), + [], + "[1, 2", + 1000, + lambda x, l: None, + False + ) assert "Raw" in r[0] - r = cv.get_content_view( - cv.get("Auto"), - [ - ["content-type", "application/json"], - ["content-encoding", "gzip"] - ], - encoding.encode('gzip', "[1, 2, 3]"), - 1000, - lambda x, l: None, - False - ) + cv.get("Auto"), + [ + ["content-type", "application/json"], + ["content-encoding", "gzip"] + ], + encoding.encode('gzip', "[1, 2, 3]"), + 1000, + lambda x, l: None, + False + ) assert "decoded gzip" in r[0] assert "JSON" in r[0] r = cv.get_content_view( - cv.get("XML"), - [ - ["content-type", "application/json"], - ["content-encoding", "gzip"] - ], - encoding.encode('gzip', "[1, 2, 3]"), - 1000, - lambda x, l: None, - False - ) + cv.get("XML"), + [ + ["content-type", "application/json"], + ["content-encoding", "gzip"] + ], + encoding.encode('gzip', "[1, 2, 3]"), + 1000, + lambda x, l: None, + False + ) assert "decoded gzip" in r[0] assert "Raw" in r[0] @@ -261,24 +259,25 @@ if pyamf: v = cv.ViewAMF() p = tutils.test_data.path("data/amf01") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) p = tutils.test_data.path("data/amf02") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) def test_view_amf_response(): v = cv.ViewAMF() p = tutils.test_data.path("data/amf03") - assert v([], file(p,"rb").read(), sys.maxint) + assert v([], file(p, "rb").read(), sys.maxsize) if cv.ViewProtobuf.is_available(): def test_view_protobuf_request(): v = cv.ViewProtobuf() p = tutils.test_data.path("data/protobuf01") - content_type, output = v([], file(p,"rb").read(), sys.maxint) + content_type, output = v([], file(p, "rb").read(), sys.maxsize) assert content_type == "Protobuf" assert output[0].text == '1: "3bbc333c-e61c-433b-819a-0b9a8cc103b8"' + def test_get_by_shortcut(): assert cv.get_by_shortcut("h") |