Only test if pymongo is installed

redref · Nov 25, 2012 · 3c9c7ab · 3c9c7ab
1 parent 830e877
commit 3c9c7ab
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/collectors/mongodb/mongodb.py b/src/collectors/mongodb/mongodb.py
@@ -17,6 +17,7 @@
     pymongo  # workaround for pyflakes issue #13
 except ImportError:
     pymongo = None
+
 try:
     from pymongo import ReadPreference
     ReadPreference  # workaround for pyflakes issue #13

diff --git a/src/collectors/mongodb/test/testmongodb.py b/src/collectors/mongodb/test/testmongodb.py
@@ -14,6 +14,26 @@
 ################################################################################
 
 
+def run_only(func, predicate):
+    if predicate():
+        return func
+    else:
+        def f(arg):
+            pass
+        return f
+
+
+def run_only_if_pymongo_is_available(func):
+    try:
+        import pymongo
+        pymongo  # workaround for pyflakes issue #13
+    except ImportError:
+        pymongo = None
+    pred = lambda: pymongo is not None
+    return run_only(func, pred)
+
+
+@run_only_if_pymongo_is_available
 class TestMongoDBCollector(CollectorTestCase):
     def setUp(self):
         config = get_collector_config('MongoDBCollector', {