From 08000c940ca2c6bcb2bf593961f6b88ee8426533 Mon Sep 17 00:00:00 2001 From: Shadab Zafar Date: Fri, 22 Jul 2016 11:45:14 +0530 Subject: [PATCH] Test marked property while serialization roundtrip --- test/mitmproxy/test_flow.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/mitmproxy/test_flow.py b/test/mitmproxy/test_flow.py index 36b212a78..74992130e 100644 --- a/test/mitmproxy/test_flow.py +++ b/test/mitmproxy/test_flow.py @@ -615,6 +615,7 @@ class TestSerialize: def test_roundtrip(self): sio = io.BytesIO() f = tutils.tflow() + f.marked = True f.request.content = bytes(bytearray(range(256))) w = flow.FlowWriter(sio) w.add(f) @@ -627,6 +628,7 @@ class TestSerialize: f2 = l[0] assert f2.get_state() == f.get_state() assert f2.request == f.request + assert f2.marked def test_load_flows(self): r = self._treader()