diff options
author | Aldo Cortesi <aldo@corte.si> | 2016-09-10 12:39:21 +1200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-09-10 12:39:21 +1200 |
commit | 387e91f9a6deaad72231a91a0735638f52f06fca (patch) | |
tree | 81d69f6f427f2921244cc06cee89797f1ff13f1a /test | |
parent | 01cdaf4133753297b2554819266752e3d501caff (diff) | |
parent | 9b814ca067a81eed3c46bdffb7d3131b6ec26708 (diff) | |
download | mitmproxy-387e91f9a6deaad72231a91a0735638f52f06fca.tar.gz mitmproxy-387e91f9a6deaad72231a91a0735638f52f06fca.tar.bz2 mitmproxy-387e91f9a6deaad72231a91a0735638f52f06fca.zip |
Merge pull request #1548 from cortesi/opts
Make various replay-related options clearer
Diffstat (limited to 'test')
-rw-r--r-- | test/mitmproxy/builtins/test_serverplayback.py | 26 | ||||
-rw-r--r-- | test/mitmproxy/console/test_master.py | 2 | ||||
-rw-r--r-- | test/mitmproxy/test_dump.py | 8 |
3 files changed, 18 insertions, 18 deletions
diff --git a/test/mitmproxy/builtins/test_serverplayback.py b/test/mitmproxy/builtins/test_serverplayback.py index 72070c7a..4db509da 100644 --- a/test/mitmproxy/builtins/test_serverplayback.py +++ b/test/mitmproxy/builtins/test_serverplayback.py @@ -22,7 +22,7 @@ class TestServerPlayback: def test_ignore_host(self): sp = serverplayback.ServerPlayback() - sp.configure(options.Options(replay_ignore_host=True), []) + sp.configure(options.Options(server_replay_ignore_host=True), []) r = tutils.tflow(resp=True) r2 = tutils.tflow(resp=True) @@ -35,7 +35,7 @@ class TestServerPlayback: def test_ignore_content(self): s = serverplayback.ServerPlayback() - s.configure(options.Options(replay_ignore_content=False), []) + s.configure(options.Options(server_replay_ignore_content=False), []) r = tutils.tflow(resp=True) r2 = tutils.tflow(resp=True) @@ -46,7 +46,7 @@ class TestServerPlayback: r2.request.content = b"bar" assert not s._hash(r) == s._hash(r2) - s.configure(options.Options(replay_ignore_content=True), []) + s.configure(options.Options(server_replay_ignore_content=True), []) r = tutils.tflow(resp=True) r2 = tutils.tflow(resp=True) r.request.content = b"foo" @@ -63,8 +63,8 @@ class TestServerPlayback: s = serverplayback.ServerPlayback() s.configure( options.Options( - replay_ignore_content=True, - replay_ignore_payload_params=[ + server_replay_ignore_content=True, + server_replay_ignore_payload_params=[ "param1", "param2" ] ), @@ -87,8 +87,8 @@ class TestServerPlayback: s = serverplayback.ServerPlayback() s.configure( options.Options( - replay_ignore_content=False, - replay_ignore_payload_params=[ + server_replay_ignore_content=False, + server_replay_ignore_payload_params=[ "param1", "param2" ] ), @@ -127,7 +127,7 @@ class TestServerPlayback: def test_headers(self): s = serverplayback.ServerPlayback() - s.configure(options.Options(rheaders=["foo"]), []) + s.configure(options.Options(server_replay_use_headers=["foo"]), []) r = tutils.tflow(resp=True) r.request.headers["foo"] = "bar" @@ -167,9 +167,9 @@ class TestServerPlayback: assert not s.next_flow(r) - def test_load_with_nopop(self): + def test_load_with_server_replay_nopop(self): s = serverplayback.ServerPlayback() - s.configure(options.Options(nopop=True), []) + s.configure(options.Options(server_replay_nopop=True), []) r = tutils.tflow(resp=True) r.request.headers["key"] = "one" @@ -187,7 +187,7 @@ class TestServerPlayback: s = serverplayback.ServerPlayback() s.configure( options.Options( - replay_ignore_params=["param1", "param2"] + server_replay_ignore_params=["param1", "param2"] ), [] ) @@ -208,7 +208,7 @@ class TestServerPlayback: s = serverplayback.ServerPlayback() s.configure( options.Options( - replay_ignore_payload_params=["param1", "param2"] + server_replay_ignore_payload_params=["param1", "param2"] ), [] ) @@ -270,7 +270,7 @@ class TestServerPlayback: def test_server_playback_kill(self): state = flow.State() s = serverplayback.ServerPlayback() - o = options.Options(refresh_server_playback = True, kill=True) + o = options.Options(refresh_server_playback = True, replay_kill_extra=True) m = mastertest.RecordingMaster(o, None, state) m.addons.add(o, s) diff --git a/test/mitmproxy/console/test_master.py b/test/mitmproxy/console/test_master.py index fcb87e1b..8388a6bd 100644 --- a/test/mitmproxy/console/test_master.py +++ b/test/mitmproxy/console/test_master.py @@ -107,7 +107,7 @@ def test_format_keyvals(): def test_options(): - assert console.master.Options(kill=True) + assert console.master.Options(replay_kill_extra=True) class TestMaster(mastertest.MasterTest): diff --git a/test/mitmproxy/test_dump.py b/test/mitmproxy/test_dump.py index 40beeb0d..9fd5ff57 100644 --- a/test/mitmproxy/test_dump.py +++ b/test/mitmproxy/test_dump.py @@ -49,14 +49,14 @@ class TestDumpMaster(mastertest.MasterTest): assert "error" in o.tfile.getvalue() def test_replay(self): - o = dump.Options(server_replay=["nonexistent"], kill=True) + o = dump.Options(server_replay=["nonexistent"], replay_kill_extra=True) tutils.raises(exceptions.OptionsError, dump.DumpMaster, None, o) with tutils.tmpdir() as t: p = os.path.join(t, "rep") self.flowfile(p) - o = dump.Options(server_replay=[p], kill=True) + o = dump.Options(server_replay=[p], replay_kill_extra=True) o.verbosity = 0 o.flow_detail = 0 m = dump.DumpMaster(None, o) @@ -64,13 +64,13 @@ class TestDumpMaster(mastertest.MasterTest): self.cycle(m, b"content") self.cycle(m, b"content") - o = dump.Options(server_replay=[p], kill=False) + o = dump.Options(server_replay=[p], replay_kill_extra=False) o.verbosity = 0 o.flow_detail = 0 m = dump.DumpMaster(None, o) self.cycle(m, b"nonexistent") - o = dump.Options(client_replay=[p], kill=False) + o = dump.Options(client_replay=[p], replay_kill_extra=False) o.verbosity = 0 o.flow_detail = 0 m = dump.DumpMaster(None, o) |