From 324d7c3955bef29745183949baed48ee286c126d Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Sun, 13 Mar 2011 16:50:11 +1300 Subject: Add client plaback to mitmproxy. --- test/test_flow.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'test') 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): -- cgit v1.2.3