diff --git a/test/mitmproxy/test_har_extractor.py b/test/mitmproxy/test_har_extractor.py index 26a092e6c..7838f7133 100644 --- a/test/mitmproxy/test_har_extractor.py +++ b/test/mitmproxy/test_har_extractor.py @@ -2,10 +2,13 @@ import json import netlib.tutils from . import tutils -from mitmproxy import script, flow from examples import har_extractor +class Context(object): + pass + + trequest = netlib.tutils.treq( timestamp_start=746203272, timestamp_end=746203272, @@ -18,14 +21,11 @@ tresponse = netlib.tutils.tresp( def test_start(): - fm = flow.FlowMaster(None, flow.State()) - ctx = script.ScriptContext(fm) - tutils.raises(ValueError, har_extractor.start, ctx, []) + tutils.raises(ValueError, har_extractor.start, Context(), []) def test_response(): - fm = flow.FlowMaster(None, flow.State()) - ctx = script.ScriptContext(fm) + ctx = Context() ctx.HARLog = har_extractor._HARLog([]) ctx.seen_server = set()