diff options
author | Shadab Zafar <dufferzafar0@gmail.com> | 2016-03-04 03:13:40 +0530 |
---|---|---|
committer | Shadab Zafar <dufferzafar0@gmail.com> | 2016-03-04 03:13:40 +0530 |
commit | d17cd5c6eca72fdddafb73028687c8b0152a5fe6 (patch) | |
tree | 5ac67b42bf4c718783d3ac6779e056a8b1232678 /test | |
parent | 6a4f1ae7c7ea4b5036937312b837184162edaed3 (diff) | |
download | mitmproxy-d17cd5c6eca72fdddafb73028687c8b0152a5fe6.tar.gz mitmproxy-d17cd5c6eca72fdddafb73028687c8b0152a5fe6.tar.bz2 mitmproxy-d17cd5c6eca72fdddafb73028687c8b0152a5fe6.zip |
Use custom Context object
Diffstat (limited to 'test')
-rw-r--r-- | test/mitmproxy/test_har_extractor.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/test/mitmproxy/test_har_extractor.py b/test/mitmproxy/test_har_extractor.py index 26a092e6..7838f713 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() |