Rooba
/
mitmproxy
mirror of https://github.com/mitmproxy/mitmproxy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

100% test coverage for flow.py

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() s = flow.State()
fm = flow.FlowMaster(None, s) 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("scripts/a.py")
assert not fm.load_script(None)
assert fm.load_script("nonexistent") assert fm.load_script("nonexistent")
assert "ValueError" in fm.load_script("scripts/starterr.py") 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): def test_script(self):
s = flow.State() s = flow.State()
fm = flow.FlowMaster(None, s) fm = flow.FlowMaster(None, s)
@ -494,11 +504,15 @@ class uFlowMaster(libpry.AutoTree):
assert not fm.handle_response(rx) assert not fm.handle_response(rx)
dc = flow.ClientDisconnect(req.client_conn) dc = flow.ClientDisconnect(req.client_conn)
req.client_conn.requestcount = 1
fm.handle_clientdisconnect(dc) fm.handle_clientdisconnect(dc)
err = flow.Error(f.request, "msg") err = flow.Error(f.request, "msg")
fm.handle_error(err) fm.handle_error(err)
fm.load_script("scripts/a.py")
fm.shutdown()
def test_client_playback(self): def test_client_playback(self):
s = flow.State() s = flow.State()