fix tests

2 files changed, 3 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 1d3c6205..79398a18 100644
--- a/setup.py
+++ b/setup.py
@@ -84,6 +84,7 @@ setup(
         'examples': [
             "pytz",
             "harparser",
+            "beautifulsoup4"
         ]
     }
 )
diff --git a/test/test_examples.py b/test/test_examples.py
index fd42e6f0..a5a212cd 100644
--- a/test/test_examples.py
+++ b/test/test_examples.py
@@ -1,10 +1,8 @@
 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)
@@ -12,6 +10,8 @@ def test_load_scripts():
     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: