From 9f16a84a9e771dbbc3314daafaf63449de73d3eb Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Fri, 11 Mar 2011 15:16:31 +1300
Subject: Make mitmdump handle invalid serialized data gracefully.

---
 test/test_flow.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'test')

diff --git a/test/test_flow.py b/test/test_flow.py
index 78892b5d..d2cb85dc 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -356,6 +356,16 @@ class uSerialize(libpry.AutoTree):
         assert len(l) == 1
         assert l[0] == f
 
+    def test_error(self):
+        sio = StringIO()
+        sio.write("bogus")
+        sio.seek(0)
+        r = flow.FlowReader(sio)
+        libpry.raises(flow.FlowReadError, list, r.stream())
+
+        f = flow.FlowReadError("foo")
+        assert f.strerror == "foo"
+
 
 class uFlowMaster(libpry.AutoTree):
     def test_all(self):
-- 
cgit v1.2.3