kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-04-27 19:15:33 +08:00
Code Issues Packages 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
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
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()