mirror of
https://github.com/mitmproxy/mitmproxy.git
synced 2025-03-05 20:27:48 +00:00
Fixed old tests
This commit is contained in:
parent
93565392cd
commit
5b5b79f5c4
@ -87,7 +87,7 @@ class uClientPlaybackState(libpry.AutoTree):
|
||||
|
||||
class uServerPlaybackState(libpry.AutoTree):
|
||||
def test_hash(self):
|
||||
s = flow.ServerPlaybackState(None, [], False)
|
||||
s = flow.ServerPlaybackState(None, [], False, False)
|
||||
r = tutils.tflow()
|
||||
r2 = tutils.tflow()
|
||||
|
||||
@ -99,7 +99,7 @@ class uServerPlaybackState(libpry.AutoTree):
|
||||
assert s._hash(r) != s._hash(r2)
|
||||
|
||||
def test_headers(self):
|
||||
s = flow.ServerPlaybackState(["foo"], [], False)
|
||||
s = flow.ServerPlaybackState(["foo"], [], False, False)
|
||||
r = tutils.tflow_full()
|
||||
r.request.headers["foo"] = ["bar"]
|
||||
r2 = tutils.tflow_full()
|
||||
@ -120,7 +120,7 @@ class uServerPlaybackState(libpry.AutoTree):
|
||||
r2 = tutils.tflow_full()
|
||||
r2.request.headers["key"] = ["two"]
|
||||
|
||||
s = flow.ServerPlaybackState(None, [r, r2], False)
|
||||
s = flow.ServerPlaybackState(None, [r, r2], False, False)
|
||||
assert s.count() == 2
|
||||
assert len(s.fmap.keys()) == 1
|
||||
|
||||
@ -547,7 +547,7 @@ class uFlowMaster(libpry.AutoTree):
|
||||
f = tutils.tflow_full()
|
||||
pb = [tutils.tflow_full(), f]
|
||||
fm = flow.FlowMaster(None, s)
|
||||
assert not fm.start_server_playback(pb, False, [], False)
|
||||
assert not fm.start_server_playback(pb, False, [], False, False)
|
||||
assert not fm.start_client_playback(pb, False)
|
||||
|
||||
q = Queue.Queue()
|
||||
@ -568,10 +568,10 @@ class uFlowMaster(libpry.AutoTree):
|
||||
fm.refresh_server_playback = True
|
||||
assert not fm.do_server_playback(tutils.tflow())
|
||||
|
||||
fm.start_server_playback(pb, False, [], False)
|
||||
fm.start_server_playback(pb, False, [], False, False)
|
||||
assert fm.do_server_playback(tutils.tflow())
|
||||
|
||||
fm.start_server_playback(pb, False, [], True)
|
||||
fm.start_server_playback(pb, False, [], True, False)
|
||||
r = tutils.tflow()
|
||||
r.request.content = "gibble"
|
||||
assert not fm.do_server_playback(r)
|
||||
|
Loading…
x
Reference in New Issue
Block a user