aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2011-03-13 16:50:11 +1300
committerAldo Cortesi <aldo@nullcube.com>2011-03-13 16:50:11 +1300
commit324d7c3955bef29745183949baed48ee286c126d (patch)
tree1b5d72fd1a1f3e8befbe00e17bc602e69117ba71 /test
parent49c1532af76594338ffbd12e4daaed9501471f6f (diff)
downloadmitmproxy-324d7c3955bef29745183949baed48ee286c126d.tar.gz
mitmproxy-324d7c3955bef29745183949baed48ee286c126d.tar.bz2
mitmproxy-324d7c3955bef29745183949baed48ee286c126d.zip
Add client plaback to mitmproxy.
Diffstat (limited to 'test')
-rw-r--r--test/test_flow.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/test/test_flow.py b/test/test_flow.py
index 947d0d59..91e3f5f4 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -168,18 +168,20 @@ class uFlow(libpry.AutoTree):
assert f == f2
def test_kill(self):
+ s = flow.State()
+ fm = flow.FlowMaster(None, s)
f = tutils.tflow()
f.request = tutils.treq()
f.intercept()
assert not f.request.acked
- f.kill()
+ f.kill(fm)
assert f.request.acked
f.intercept()
f.response = tutils.tresp()
f.request = f.response.request
f.request.ack()
assert not f.response.acked
- f.kill()
+ f.kill(fm)
assert f.response.acked
def test_accept_intercept(self):