mitmproxy/test/test_examples.py

import glob
from libmproxy import utils, script
from libmproxy.proxy import config
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:
            f += " foo"
        if "iframe_injector" in f:
            f += " foo"  # one argument required
        if "modify_response_body" in f:
            f += " foo bar"  # two arguments required
        s = script.Script(f, tmaster)  # Loads the script file.
        s.unload()