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

fix #876

Browse Source
This commit is contained in:
Maximilian Hils 2016-02-08 01:06:18 +01:00
parent b0477fd8c9
commit 2654cd87af
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libmproxy/models/http.py
View File

@ -91,10 +91,12 @@ class MessageMixin(stateobject.StateObject):
Returns the number of replacements made.
"""
with decoded(self):
self.content, count = utils.safe_subn(
pattern, repl, self.content, *args, **kwargs
)
count = 0
if self.content:
with decoded(self):
self.content, count = utils.safe_subn(
pattern, repl, self.content, *args, **kwargs
)
fields = []
for name, value in self.headers.fields:
name, c = utils.safe_subn(pattern, repl, name, *args, **kwargs)

5
test/test_flow.py
View File

@ -463,6 +463,11 @@ class TestFlow(object):
f.response.content = "\xc2foo"
f.replace("foo", u"bar")
def test_replace_no_content(self):
f = tutils.tflow()
f.request.content = CONTENT_MISSING
assert f.replace("foo", "bar") == 0
def test_replace(self):
f = tutils.tflow(resp=True)
f.request.headers["foo"] = "foo"