Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 08:11:00 +00:00
Code Issues Projects Releases Wiki Activity
11c63dcb9f
mitmproxy/test/test_console_contentview.py

226 lines
6.2 KiB
Python
Raw Normal View History

Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
import sys
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
import libmproxy.console.contentview as cv
from libmproxy import utils, flow, encoding
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
import tutils
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
class TestContentView:
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
def test_trailer(self):
txt = []
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
cv.trailer(5, txt, 1000)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
assert not txt
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
cv.trailer(cv.VIEW_CUTOFF + 10, txt, cv.VIEW_CUTOFF)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
assert txt
Pretty view now indents Javascript. Thanks to the JSBeautifier project, which is now included in the contrib directory.
2012-03-24 21:56:45 +00:00
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
def test_get_view_func(self):
f = cv.get_view_func(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Hex"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
flow.ODictCaseless(),
"foo"
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert f.name == "Hex"
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
f = cv.get_view_func(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
flow.ODictCaseless(),
"foo"
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert f.name == "Raw"
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
f = cv.get_view_func(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
flow.ODictCaseless(
[["content-type", "text/html"]],
),
"foo"
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert f.name == "HTML"
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
f = cv.get_view_func(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
flow.ODictCaseless(
[["content-type", "text/flibble"]],
),
"foo"
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert f.name == "Raw"
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
f = cv.get_view_func(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
flow.ODictCaseless(
[["content-type", "text/flibble"]],
),
"<xml></xml>"
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert f.name == "XML"
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Optional AMF decoding support If PyAMF is installed, enable AMF decoding.
2012-08-17 06:27:47 +00:00
try:
import pyamf
f = cv.get_view_func(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Optional AMF decoding support If PyAMF is installed, enable AMF decoding.
2012-08-17 06:27:47 +00:00
flow.ODictCaseless(
[["content-type", "application/x-amf"]],
),
""
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert f.name == "AMF"
Optional AMF decoding support If PyAMF is installed, enable AMF decoding.
2012-08-17 06:27:47 +00:00
except ImportError:
pass
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
def test_view_urlencoded(self):
d = utils.urlencode([("one", "two"), ("three", "four")])
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewURLEncoded()
assert v([], d, 100)
assert not v([], "foo", 100)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
def test_view_html(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewHTML()
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
s = "<html><br><br></br><p>one</p></html>"
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], s, 1000)
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
s = "gobbledygook"
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert not v([], s, 1000)
def test_view_html_outline(self):
v = cv.ViewHTMLOutline()
s = "<html><br><br></br><p>one</p></html>"
assert v([], s, 1000)
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
def test_view_json(self):
cv.VIEW_CUTOFF = 100
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewJSON()
assert v([], "{}", 1000)
assert not v([], "{", 1000)
assert v([], "[" + ",".join(["0"]*cv.VIEW_CUTOFF) + "]", 1000)
assert v([], "[1, 2, 3, 4, 5]", 5)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
def test_view_xml(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewXML()
assert v([], "<foo></foo>", 1000)
assert not v([], "<foo>", 1000)
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
s = """<?xml version="1.0" encoding="UTF-8"?>
<?xml-stylesheet title="XSL_formatting"?>
XML/HTML pretty view tweaks.
2012-04-07 10:15:31 +00:00
<rss
Integrate lxml for pretty-printing HTML and XML. Tackling the pretty-printing performance problem head-on, at the cost of a major dependency.
2012-04-07 01:47:03 +00:00
xmlns:media="http://search.yahoo.com/mrss/"
xmlns:atom="http://www.w3.org/2005/Atom"
version="2.0">
</rss>
"""
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], s, 1000)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
def test_view_raw(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewRaw()
assert v([], "foo", 1000)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Pretty view now indents Javascript. Thanks to the JSBeautifier project, which is now included in the contrib directory.
2012-03-24 21:56:45 +00:00
def test_view_javascript(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewJavaScript()
assert v([], "[1, 2, 3]", 100)
assert v([], "[1, 2, 3", 100)
assert v([], "function(a){[1, 2, 3]}", 100)
Pretty view now indents Javascript. Thanks to the JSBeautifier project, which is now included in the contrib directory.
2012-03-24 21:56:45 +00:00
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
def test_view_hex(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewHex()
assert v([], "foo", 1000)
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Add a pretty-viewer for images. This shows basic image information like dimensions, plus extracted EXIF tags and other metadata.
2012-03-25 22:26:02 +00:00
def test_view_image(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewImage()
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
p = tutils.test_data.path("data/image.png")
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], file(p).read(), sys.maxint)
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
p = tutils.test_data.path("data/image.gif")
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], file(p).read(), sys.maxint)
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
p = tutils.test_data.path("data/image-err1.jpg")
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], file(p).read(), sys.maxint)
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
p = tutils.test_data.path("data/image.ico")
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], file(p).read(), sys.maxint)
Port mitmproxy test suite entirely to nose.
2012-06-09 01:42:43 +00:00
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert not v([], "flibble", sys.maxint)
Add a pretty-viewer for images. This shows basic image information like dimensions, plus extracted EXIF tags and other metadata.
2012-03-25 22:26:02 +00:00
Optional AMF decoding support If PyAMF is installed, enable AMF decoding.
2012-08-17 06:27:47 +00:00
def test_view_amf(self):
try:
import pyamf
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
v = cv.ViewAMF()
Optional AMF decoding support If PyAMF is installed, enable AMF decoding.
2012-08-17 06:27:47 +00:00
p = tutils.test_data.path("data/test.amf")
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert v([], file(p).read(), sys.maxint)
Optional AMF decoding support If PyAMF is installed, enable AMF decoding.
2012-08-17 06:27:47 +00:00
except ImportError:
pass
Re-enable simple multipart form parsing and preview.
2012-03-24 21:10:48 +00:00
def test_view_multipart(self):
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
view = cv.ViewMultipart()
Re-enable simple multipart form parsing and preview.
2012-03-24 21:10:48 +00:00
v = """
--AaB03x
Content-Disposition: form-data; name="submit-name"
Larry
--AaB03x
""".strip()
h = flow.ODictCaseless(
[("Content-Type", "multipart/form-data; boundary=AaB03x")]
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert view(h, v, 1000)
Re-enable simple multipart form parsing and preview.
2012-03-24 21:10:48 +00:00
h = flow.ODictCaseless()
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert not view(h, v, 1000)
Re-enable simple multipart form parsing and preview.
2012-03-24 21:10:48 +00:00
h = flow.ODictCaseless(
[("Content-Type", "multipart/form-data")]
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert not view(h, v, 1000)
Re-enable simple multipart form parsing and preview.
2012-03-24 21:10:48 +00:00
h = flow.ODictCaseless(
[("Content-Type", "unparseable")]
)
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
assert not view(h, v, 1000)
Re-enable simple multipart form parsing and preview.
2012-03-24 21:10:48 +00:00
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
def test_get_content_view(self):
r = cv.get_content_view(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Raw"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
[["content-type", "application/json"]],
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
"[1, 2, 3]",
1000
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
)
Start rationalizing content views. We now no longer have distinction between "pretty" view and hex/raw. Instead, we simply a default AUTO view with a global override (M) and a local override (m).
2012-04-01 22:30:35 +00:00
assert "Raw" in r[0]
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
r = cv.get_content_view(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
[["content-type", "application/json"]],
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
"[1, 2, 3]",
1000
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
)
assert r[0] == "JSON"
r = cv.get_content_view(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
[["content-type", "application/json"]],
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
"[1, 2",
1000
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
)
Refactor proxy.Server to fix a crash when replaying with -n
2012-04-02 01:24:51 +00:00
assert "Raw" in r[0]
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
r = cv.get_content_view(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("Auto"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
[
["content-type", "application/json"],
["content-encoding", "gzip"]
],
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
encoding.encode('gzip', "[1, 2, 3]"),
1000
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
)
assert "decoded gzip" in r[0]
assert "JSON" in r[0]
r = cv.get_content_view(
Huge cleanup of content viewers.
2012-08-18 05:08:17 +00:00
cv.get("XML"),
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
[
["content-type", "application/json"],
["content-encoding", "gzip"]
],
Add an "f" shortcut key to load full body contents.
2012-04-08 07:23:05 +00:00
encoding.encode('gzip', "[1, 2, 3]"),
1000
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
)
assert "decoded gzip" in r[0]
XML/HTML pretty view tweaks.
2012-04-07 10:15:31 +00:00
assert "Raw" in r[0]
Refactor pretty view mechanism. Also start adding unit tests for this subsystem.
2012-03-24 01:02:41 +00:00
Reference in New Issue Copy Permalink