Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-01-30 14:58:38 +00:00
Code Issues Projects Releases Wiki Activity

Use custom Context object

Browse Source
This commit is contained in:
Shadab Zafar 2016-03-04 03:13:40 +05:30
parent 6a4f1ae7c7
commit d17cd5c6ec
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/mitmproxy/test_har_extractor.py
View File

@ -2,10 +2,13 @@ import json
import netlib.tutils import netlib.tutils
from . import tutils from . import tutils
from mitmproxy import script, flow
from examples import har_extractor from examples import har_extractor
class Context(object):
pass
trequest = netlib.tutils.treq( trequest = netlib.tutils.treq(
timestamp_start=746203272, timestamp_start=746203272,
timestamp_end=746203272, timestamp_end=746203272,
@ -18,14 +21,11 @@ tresponse = netlib.tutils.tresp(
def test_start(): def test_start():
fm = flow.FlowMaster(None, flow.State()) tutils.raises(ValueError, har_extractor.start, Context(), [])
ctx = script.ScriptContext(fm)
tutils.raises(ValueError, har_extractor.start, ctx, [])
def test_response(): def test_response():
fm = flow.FlowMaster(None, flow.State()) ctx = Context()
ctx = script.ScriptContext(fm)
ctx.HARLog = har_extractor._HARLog([]) ctx.HARLog = har_extractor._HARLog([])
ctx.seen_server = set() ctx.seen_server = set()