diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2013-03-14 09:19:43 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2013-03-14 09:19:43 +1300 |
commit | cfb5ba89ce96594b2f8d51f27c9b2ee41ecf18e5 (patch) | |
tree | 7bdea72173b7f49e8409f6c5fb0c42e0fe3329ec /test/test_flow.py | |
parent | c34d1e3de6483b1ab8a1a3add53fb84364c68249 (diff) | |
download | mitmproxy-cfb5ba89ce96594b2f8d51f27c9b2ee41ecf18e5.tar.gz mitmproxy-cfb5ba89ce96594b2f8d51f27c9b2ee41ecf18e5.tar.bz2 mitmproxy-cfb5ba89ce96594b2f8d51f27c9b2ee41ecf18e5.zip |
Introduce a filtered flow writer, and use it in dump.py
Fixes #104
Diffstat (limited to 'test/test_flow.py')
-rw-r--r-- | test/test_flow.py | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/test/test_flow.py b/test/test_flow.py index 6aa898ad..c1ae1a9f 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -497,6 +497,23 @@ class TestSerialize: fm = flow.FlowMaster(None, s) fm.load_flows(r) assert len(s._flow_list) == 6 + + def test_filter(self): + sio = StringIO() + fl = filt.parse("~c 200") + w = flow.FilteredFlowWriter(sio, fl) + + f = tutils.tflow_full() + f.response.code = 200 + w.add(f) + + f = tutils.tflow_full() + f.response.code = 201 + w.add(f) + + sio.seek(0) + r = flow.FlowReader(sio) + assert len(list(r.stream())) def test_error(self): @@ -723,7 +740,7 @@ class TestFlowMaster: fm = flow.FlowMaster(None, s) tf = tutils.tflow_full() - fm.start_stream(file(p, "ab")) + fm.start_stream(file(p, "ab"), None) fm.handle_request(tf.request) fm.handle_response(tf.response) fm.stop_stream() @@ -731,7 +748,7 @@ class TestFlowMaster: assert r()[0].response tf = tutils.tflow_full() - fm.start_stream(file(p, "ab")) + fm.start_stream(file(p, "ab"), None) fm.handle_request(tf.request) fm.shutdown() |