diff options
Diffstat (limited to 'mitmproxy/test/test_examples.py')
-rw-r--r-- | mitmproxy/test/test_examples.py | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/mitmproxy/test/test_examples.py b/mitmproxy/test/test_examples.py new file mode 100644 index 00000000..bb7c596b --- /dev/null +++ b/mitmproxy/test/test_examples.py @@ -0,0 +1,30 @@ +import glob +from libmproxy import utils, script +from libmproxy.proxy import config +from . import tservers + + +def test_load_scripts(): + example_dir = utils.Data("libmproxy").path("../examples") + scripts = glob.glob("%s/*.py" % example_dir) + + tmaster = tservers.TestMaster(config.ProxyConfig()) + + for f in scripts: + if "har_extractor" in f: + continue + if "flowwriter" in f: + f += " -" + if "iframe_injector" in f: + f += " foo" # one argument required + if "filt" in f: + f += " ~a" + if "modify_response_body" in f: + f += " foo bar" # two arguments required + try: + s = script.Script(f, script.ScriptContext(tmaster)) # Loads the script file. + except Exception as v: + if "ImportError" not in str(v): + raise + else: + s.unload() |