aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_flow.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2012-03-16 11:24:18 +1300
committerAldo Cortesi <aldo@nullcube.com>2012-03-16 11:24:18 +1300
commitd138af72171d833659cfb53edc80eade121ca836 (patch)
treebbd8fbe35e5be084953df89dbf8ef8c4e167d044 /test/test_flow.py
parentd51b8cab0c0d1352865155865dfd258f66103ffe (diff)
downloadmitmproxy-d138af72171d833659cfb53edc80eade121ca836.tar.gz
mitmproxy-d138af72171d833659cfb53edc80eade121ca836.tar.bz2
mitmproxy-d138af72171d833659cfb53edc80eade121ca836.zip
replace() methods now decode and re-encode contents before substitution.
Diffstat (limited to 'test/test_flow.py')
-rw-r--r--test/test_flow.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/test/test_flow.py b/test/test_flow.py
index 56303881..ff35f899 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -278,6 +278,24 @@ class uFlow(libpry.AutoTree):
f.replace("error", "bar")
assert f.error.msg == "bar"
+ def test_replace_encoded(self):
+ f = tutils.tflow_full()
+ f.request.content = "afoob"
+ f.request.encode("gzip")
+ f.response.content = "afoob"
+ f.response.encode("gzip")
+
+ f.replace("foo", "bar")
+
+ assert f.request.content != "abarb"
+ f.request.decode()
+ assert f.request.content == "abarb"
+
+ assert f.response.content != "abarb"
+ f.response.decode()
+ assert f.response.content == "abarb"
+
+
class uState(libpry.AutoTree):
def test_backup(self):