Adding decorator requires(ModuleName) for tests

2 files changed, 43 insertions, 0 deletions

diff --git a/tests/util/helper.py b/tests/util/helper/__init__.py
index a010934ae..a010934ae 100644
--- a/tests/util/helper.py
+++ b/tests/util/helper/__init__.py
diff --git a/tests/util/helper/decorators.py b/tests/util/helper/decorators.py
new file mode 100644
index 000000000..ca7f9cdaf
--- /dev/null
+++ b/tests/util/helper/decorators.py
@@ -0,0 +1,43 @@
+
+'''Decorators for skipping test methods and test cases'''
+
+import logging
+import unittest
+
+
+class requires(object):
+    '''Skip if the given module is not found
+    
+    Usage:
+
+    @requires('RequiredModule')
+    class MyTestCase(unittest.TestCase):
+        ...
+    '''
+
+    def __init__(self, *args):
+        '''Setup this decorator. Args should be a list of
+        module names'''
+        self.skip = False
+
+        for module in args:
+            try:
+                __import__(module)
+            except ImportError:
+                self.skip = True
+                self.skipped = module
+                break
+
+    def __call__(self, klass):
+        '''Replace a skipped class with a dummy testCase'''
+        if not self.skip:
+            return klass
+
+        logging.warning('Module %s not found. Skipping %s' % (self.skipped,
+                                                            klass.__name__))
+
+        class SkipWrapper(unittest.TestCase):
+            '''Dummy wrapper'''
+            pass
+
+        return SkipWrapper