aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_flow.py
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2013-12-16 22:10:06 +0100
committerMaximilian Hils <git@maximilianhils.com>2013-12-16 22:10:06 +0100
commite643759ef653908dadffd82a832eb67f5223cd71 (patch)
tree54812017b647bc64e844546bbde12a5b04ff1298 /test/test_flow.py
parent6ec2e6f24f2e64854e80d81c04ae44c094486821 (diff)
parentafff1f17340b2a62de168ec42655ed7e5779c3c4 (diff)
downloadmitmproxy-e643759ef653908dadffd82a832eb67f5223cd71.tar.gz
mitmproxy-e643759ef653908dadffd82a832eb67f5223cd71.tar.bz2
mitmproxy-e643759ef653908dadffd82a832eb67f5223cd71.zip
merge smurfix/fix2, add serverconnect hook docs, adjust tests
Diffstat (limited to 'test/test_flow.py')
-rw-r--r--test/test_flow.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/test/test_flow.py b/test/test_flow.py
index 9844e0fd..c614960b 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -1,7 +1,7 @@
import Queue, time, os.path
from cStringIO import StringIO
import email.utils
-from libmproxy import filt, flow, controller, utils, tnetstring
+from libmproxy import filt, flow, controller, utils, tnetstring, proxy
import tutils
@@ -575,6 +575,10 @@ class TestFlowMaster:
req = tutils.treq()
fm.handle_clientconnect(req.client_conn)
assert fm.scripts[0].ns["log"][-1] == "clientconnect"
+ sc = proxy.ServerConnection(None, req.scheme, req.host, req.port, None)
+ sc.reply = controller.DummyReply()
+ fm.handle_serverconnection(sc)
+ assert fm.scripts[0].ns["log"][-1] == "serverconnect"
f = fm.handle_request(req)
assert fm.scripts[0].ns["log"][-1] == "request"
resp = tutils.tresp(req)