diff options
Diffstat (limited to 'test/mitmproxy/test_flow_format_compat.py')
| -rw-r--r-- | test/mitmproxy/test_flow_format_compat.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/test/mitmproxy/test_flow_format_compat.py b/test/mitmproxy/test_flow_format_compat.py index cc80db81..4997aff9 100644 --- a/test/mitmproxy/test_flow_format_compat.py +++ b/test/mitmproxy/test_flow_format_compat.py @@ -1,11 +1,11 @@ -from mitmproxy.flow import FlowReader -from mitmproxy.exceptions import FlowReadException +from mitmproxy import io +from mitmproxy import exceptions from . import tutils def test_load(): with open(tutils.test_data.path("data/dumpfile-011"), "rb") as f: - flow_reader = FlowReader(f) + flow_reader = io.FlowReader(f) flows = list(flow_reader.stream()) assert len(flows) == 1 assert flows[0].request.url == "https://example.com/" @@ -13,6 +13,6 @@ def test_load(): def test_cannot_convert(): with open(tutils.test_data.path("data/dumpfile-010"), "rb") as f: - flow_reader = FlowReader(f) - with tutils.raises(FlowReadException): + flow_reader = io.FlowReader(f) + with tutils.raises(exceptions.FlowReadException): list(flow_reader.stream()) |