aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/test_flow.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/test/test_flow.py b/test/test_flow.py
index b71ce6af..4dfa059e 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -1,3 +1,4 @@
+from cStringIO import StringIO
from libmproxy import console, proxy, filt, flow
import utils
import libpry
@@ -249,8 +250,24 @@ class uState(libpry.AutoTree):
c.accept_all()
+class uSerialize(libpry.AutoTree):
+ def test_roundtrip(self):
+ sio = StringIO()
+ f = utils.tflow()
+ w = flow.FlowWriter(sio)
+ w.add(f)
+
+ sio.seek(0)
+ r = flow.FlowReader(sio)
+ l = list(r.stream())
+ assert len(l) == 1
+ assert l[0] == f
+
+
+
tests = [
uFlow(),
uState(),
+ uSerialize()
]