1 files changed, 11 insertions, 6 deletions
diff --git a/test/test_examples.py b/test/test_examples.py
index fa57b7eb..d18b5862 100644
--- a/test/test_examples.py
+++ b/test/test_examples.py
@@ -1,12 +1,17 @@
-from libmproxy import utils, script
 import glob
+from libmproxy import utils, script
 from libmproxy.proxy import config
 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():
+
+    example_dir = utils.Data("libmproxy").path("../examples")
+    scripts = glob.glob("%s/*.py" % example_dir)
+
+    tmaster = tservers.TestMaster(config.ProxyConfig())
 
-for f in scripts:
-    script.Script(f, tmaster)  # Loads the script file.
-\ No newline at end of file
+    for f in scripts:
+        if "modify_response_body" in f:
+            f += " foo bar"  # two arguments required
+        script.Script(f, tmaster)  # Loads the script file.
+\ No newline at end of file