Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 18:18:25 +00:00
Code Issues Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Maximilian Hils 2014-07-26 11:51:27 +02:00
parent 41f49ff0d4
commit a73ad1d945
2 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/modify_response_body.py
View File

@ -7,8 +7,7 @@ from libmproxy.protocol.http import decoded
def start(ctx, argv): def start(ctx, argv):
if len(argv) != 3: if len(argv) != 3:
ctx.log('Usage: -s "modify-response-body.py old new"') sys.exit('Usage: -s "modify-response-body.py old new"')
sys.exit(1)
global old, new # In larger scripts, a centralized options object (as returned by argparse) is encouraged global old, new # In larger scripts, a centralized options object (as returned by argparse) is encouraged
old, new = argv[1:] old, new = argv[1:]

17
test/test_examples.py
View File

@ -1,12 +1,17 @@
from libmproxy import utils, script
import glob import glob
from libmproxy import utils, script
from libmproxy.proxy import config from libmproxy.proxy import config
import tservers import tservers
example_dir = utils.Data("libmproxy").path("../examples")
scripts = glob.glob("%s/*.py" % example_dir)
tmaster = tservers.TestMaster(config.ProxyConfig()) def test_load_scripts():
for f in scripts: example_dir = utils.Data("libmproxy").path("../examples")
script.Script(f, tmaster) # Loads the script file. scripts = glob.glob("%s/*.py" % example_dir)
tmaster = tservers.TestMaster(config.ProxyConfig())
for f in scripts:
if "modify_response_body" in f:
f += " foo bar" # two arguments required
script.Script(f, tmaster) # Loads the script file.