diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_flow.py | 28 |
1 files changed, 27 insertions, 1 deletions
diff --git a/test/test_flow.py b/test/test_flow.py index 170b16bb..99df9ed0 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -1,4 +1,4 @@ -import Queue, time +import Queue, time, os.path from cStringIO import StringIO import email.utils from libmproxy import filt, flow, controller, utils, tnetstring @@ -689,6 +689,32 @@ class TestFlowMaster: fm.handle_request(f.request) assert f.request.headers["authorization"] == ["foo"] + def test_stream(self): + with tutils.tmpdir() as tdir: + p = os.path.join(tdir, "foo") + def r(): + r = flow.FlowReader(open(p)) + return list(r.stream()) + + s = flow.State() + fm = flow.FlowMaster(None, s) + tf = tutils.tflow_full() + + fm.start_stream(file(p, "ab")) + fm.handle_request(tf.request) + fm.handle_response(tf.response) + fm.stop_stream() + + assert r()[0].response + + tf = tutils.tflow_full() + fm.start_stream(file(p, "ab")) + fm.handle_request(tf.request) + fm.shutdown() + + assert not r()[1].response + + class TestRequest: def test_simple(self): h = flow.ODictCaseless() |