Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

100% test coverage for flow.py

Browse Source
This commit is contained in:
Aldo Cortesi 2012-02-10 15:55:58 +13:00
parent 1ad7e91527
commit da1ccfddeb
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/scripts/reqerr.py Normal file
View File

@ -0,0 +1,2 @@
def request(ctx, r):
raise ValueError

14
test/test_flow.py
View File

@ -453,9 +453,19 @@ class uFlowMaster(libpry.AutoTree):
s = flow.State()
fm = flow.FlowMaster(None, s)
assert not fm.load_script("scripts/a.py")
assert not fm.load_script("scripts/a.py")
assert not fm.load_script(None)
assert fm.load_script("nonexistent")
assert "ValueError" in fm.load_script("scripts/starterr.py")
def test_script_reqerr(self):
s = flow.State()
fm = flow.FlowMaster(None, s)
assert not fm.load_script("scripts/reqerr.py")
req = tutils.treq()
fm.handle_clientconnect(req.client_conn)
f = fm.handle_request(req)
def test_script(self):
s = flow.State()
fm = flow.FlowMaster(None, s)
@ -494,11 +504,15 @@ class uFlowMaster(libpry.AutoTree):
assert not fm.handle_response(rx)
dc = flow.ClientDisconnect(req.client_conn)
req.client_conn.requestcount = 1
fm.handle_clientdisconnect(dc)
err = flow.Error(f.request, "msg")
fm.handle_error(err)
fm.load_script("scripts/a.py")
fm.shutdown()
def test_client_playback(self):
s = flow.State()