diff --git a/test/scripts/reqerr.py b/test/scripts/reqerr.py new file mode 100644 index 000000000..e7c503a87 --- /dev/null +++ b/test/scripts/reqerr.py @@ -0,0 +1,2 @@ +def request(ctx, r): + raise ValueError diff --git a/test/test_flow.py b/test/test_flow.py index 5a8eb0f2e..45daf0c5c 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -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()