From 39207ffdd280af854d521f810f6082d42943eefa Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Wed, 23 Feb 2011 10:54:51 +1300 Subject: Add a way for users to specify header significance in server replay. Also add the --rheader command-line option to mitmdump to let the user specify an arbitrary number of significant headers. The default is to treat no headers as significant. --- test/test_flow.py | 23 +++++++++++++++++++---- 1 file changed, 19 insertions(+), 4 deletions(-) (limited to 'test') diff --git a/test/test_flow.py b/test/test_flow.py index 9963ad18..e574f545 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -6,7 +6,7 @@ import libpry class uServerPlaybackState(libpry.AutoTree): def test_hash(self): - s = flow.ServerPlaybackState() + s = flow.ServerPlaybackState(None) r = utils.tflow() r2 = utils.tflow() @@ -17,8 +17,23 @@ class uServerPlaybackState(libpry.AutoTree): r.request.path = "voing" assert s._hash(r) != s._hash(r2) + def test_headers(self): + s = flow.ServerPlaybackState(["foo"]) + r = utils.tflow_full() + r.request.headers["foo"] = ["bar"] + r2 = utils.tflow_full() + assert not s._hash(r) == s._hash(r2) + r2.request.headers["foo"] = ["bar"] + assert s._hash(r) == s._hash(r2) + r2.request.headers["oink"] = ["bar"] + assert s._hash(r) == s._hash(r2) + + r = utils.tflow_full() + r2 = utils.tflow_full() + assert s._hash(r) == s._hash(r2) + def test_load(self): - s = flow.ServerPlaybackState() + s = flow.ServerPlaybackState(None) r = utils.tflow_full() r.request.headers["key"] = ["one"] @@ -319,10 +334,10 @@ class uFlowMaster(libpry.AutoTree): fm = flow.FlowMaster(None, s) assert not fm.do_playback(utils.tflow()) - fm.start_playback(pb, False) + fm.start_playback(pb, False, []) assert fm.do_playback(utils.tflow()) - fm.start_playback(pb, False) + fm.start_playback(pb, False, []) r = utils.tflow() r.request.content = "gibble" assert not fm.do_playback(r) -- cgit v1.2.3