diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/test_flow.py | 17 |
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() ] |