From da1ccfddeb38c534e8d511c45980783af76716c2 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Fri, 10 Feb 2012 15:55:58 +1300
Subject: [PATCH] 100% test coverage for flow.py

---
 test/scripts/reqerr.py |  2 ++
 test/test_flow.py      | 14 ++++++++++++++
 2 files changed, 16 insertions(+)
 create mode 100644 test/scripts/reqerr.py

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()