Serialized data version check.

1 files changed, 12 insertions, 1 deletions

diff --git a/test/test_flow.py b/test/test_flow.py
index a7ab7cc2..74cf79f8 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -1,7 +1,7 @@
 import Queue, time
 from cStringIO import StringIO
 import email.utils
-from libmproxy import filt, flow, controller, utils
+from libmproxy import filt, flow, controller, utils, tnetstring
 import tutils
 import libpry
 
@@ -494,6 +494,17 @@ class uSerialize(libpry.AutoTree):
         f = flow.FlowReadError("foo")
         assert f.strerror == "foo"
 
+    def test_versioncheck(self):
+        f = tutils.tflow()
+        d = f._get_state()
+        d["version"] = (0, 0)
+        sio = StringIO()
+        tnetstring.dump(d, sio)
+        sio.seek(0)
+
+        r = flow.FlowReader(sio)
+        libpry.raises("version", list, r.stream())
+
 
 class uFlowMaster(libpry.AutoTree):
     def test_load_script(self):