Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
Maximilian Hils 2016-12-10 12:06:33 +01:00
parent 123ef043dc
commit 3e37cbd061
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/simple/modify_body_inject_iframe.py
View File

@ -11,7 +11,7 @@ class Injector:
def response(self, flow):
if flow.request.host in self.iframe_url:
return
html = BeautifulSoup(flow.response.content)
html = BeautifulSoup(flow.response.content, "html.parser")
if html.body:
iframe = html.new_tag(
"iframe",

2
mitmproxy/contentviews/auto.py
View File

@ -17,7 +17,7 @@ class ViewAuto(base.View):
if ct in contentviews.content_types_map:
return contentviews.content_types_map[ct][0](data, **metadata)
elif strutils.is_xml(data):
return contentviews.get("XML")(data, **metadata)
return contentviews.get("XML/HTML")(data, **metadata)
if metadata.get("query"):
return contentviews.get("Query")(data, **metadata)
if data and strutils.is_mostly_bin(data):

2
test/mitmproxy/contentviews/test_api.py
View File

@ -82,4 +82,4 @@ def test_get_message_content_view():
def test_get_by_shortcut():
assert contentviews.get_by_shortcut("h")
assert contentviews.get_by_shortcut("s")

2
test/mitmproxy/test_examples.py
View File

@ -56,7 +56,7 @@ class TestScripts(mastertest.MasterTest):
tscript("simple/modify_body_inject_iframe.py")
m, sc = tscript("simple/modify_body_inject_iframe.py", "http://example.org/evil_iframe")
f = tflow.tflow(resp=tutils.tresp(content=b"<html>mitmproxy</html>"))
f = tflow.tflow(resp=tutils.tresp(content=b"<html><body>mitmproxy</body></html>"))
m.response(f)
content = f.response.content
assert b'iframe' in content and b'evil_iframe' in content