diff options
author | Shadab Zafar <dufferzafar0@gmail.com> | 2016-03-07 22:52:55 +0530 |
---|---|---|
committer | Shadab Zafar <dufferzafar0@gmail.com> | 2016-03-11 17:57:57 +0530 |
commit | af9442a9bae834c82d50d3a9f31796377c29a9ad (patch) | |
tree | 6d3bc558b30f71bd6afc0ddca9b541f5f4a6761e | |
parent | b95cc63b87ea87d9a6105ef72ad1d653c5c01b06 (diff) | |
download | mitmproxy-af9442a9bae834c82d50d3a9f31796377c29a9ad.tar.gz mitmproxy-af9442a9bae834c82d50d3a9f31796377c29a9ad.tar.bz2 mitmproxy-af9442a9bae834c82d50d3a9f31796377c29a9ad.zip |
Add contextmanager to load an example
-rw-r--r-- | test/mitmproxy/test_examples.py | 25 |
1 files changed, 15 insertions, 10 deletions
diff --git a/test/mitmproxy/test_examples.py b/test/mitmproxy/test_examples.py index a76d56a8..3244ebf2 100644 --- a/test/mitmproxy/test_examples.py +++ b/test/mitmproxy/test_examples.py @@ -1,4 +1,6 @@ import glob +import os +from contextlib import contextmanager from mitmproxy import utils, script from mitmproxy.proxy import config @@ -6,25 +8,28 @@ from netlib import tutils as netutils from netlib.http import Headers from . import tservers, tutils -from examples import ( - add_header, - custom_contentviews, - iframe_injector, - modify_form, - modify_querystring, - modify_response_body, - redirect_requests, -) +example_dir = utils.Data(__name__).path("../../examples") + + +@contextmanager +def example(command): + command = os.path.join(example_dir, command) + # tmaster = tservers.TestMaster(config.ProxyConfig()) + # ctx = script.ScriptContext(tmaster) + ctx = DummyContext() + s = script.Script(command, ctx) + yield s + s.unload() class DummyContext(object): + """Emulate script.ScriptContext() functionality.""" def log(self, *args, **kwargs): pass def test_load_scripts(): - example_dir = utils.Data(__name__).path("../../examples") scripts = glob.glob("%s/*.py" % example_dir) tmaster = tservers.TestMaster(config.ProxyConfig()) |