fix tests

nderkach · Nov 15, 2014 · 667fe0c · 667fe0c
1 parent a699bdc
commit 667fe0c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -84,6 +84,7 @@
         'examples': [
             "pytz",
             "harparser",
+            "beautifulsoup4"
         ]
     }
 )
diff --git a/test/test_examples.py b/test/test_examples.py
@@ -1,17 +1,17 @@
 import glob
-import mock
 from libmproxy import utils, script
 from libmproxy.proxy import config
 import tservers
 
-@mock.patch.dict("sys.modules", {"bs4": mock.Mock()})
 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: