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

View File

@ -91,6 +91,8 @@ class MessageMixin(stateobject.StateObject):
Returns the number of replacements made.
"""
count = 0
if self.content:
with decoded(self):
self.content, count = utils.safe_subn(
pattern, repl, self.content, *args, **kwargs

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"