unittest setup: Automatically find modules to import for coverage.

kombu/tests/__init__.py

@@ -1,6 +1,8 @@
 from __future__ import absolute_import
 
 import anyjson
+import os
+import sys
 
 from ..exceptions import VersionMismatch
 
@@ -11,51 +13,55 @@ try:
 except ImportError:
     anyjson.force_implementation("simplejson")
 
-moduleindex = ("kombu.abstract",
-               "kombu.clocks",
-               "kombu.common",
-               "kombu.compat",
-               "kombu.compression",
-               "kombu.connection",
-               "kombu.entity",
-               "kombu.exceptions",
-               "kombu.log",
-               "kombu.messaging",
-               "kombu.mixins",
-               "kombu.pidbox",
-               "kombu.pools",
-               "kombu.serialization",
-               "kombu.simple",
-               "kombu.utils",
-               "kombu.utils.compat",
-               "kombu.utils.debug",
-               "kombu.utils.encoding",
-               "kombu.utils.functional",
-               "kombu.utils.limits",
-               "kombu.transport",
-               "kombu.transport.amqplib",
-               "kombu.transport.base",
-               "kombu.transport.beanstalk",
-               "kombu.transport.couchdb",
-               "kombu.transport.django",
-               "kombu.transport.django.managers",
-               "kombu.transport.django.models",
-               "kombu.transport.memory",
-               "kombu.transport.mongodb",
-               "kombu.transport.pika",
-               "kombu.transport.redis",
-               "kombu.transport.SQS",
-               "kombu.transport.sqlalchemy",
-               "kombu.transport.virtual",
-               "kombu.transport.virtual.exchange",
-               "kombu.transport.virtual.scheduling")
+
+def find_distribution_modules(name=__name__, file=__file__):
+    current_dist_depth = len(name.split(".")) - 1
+    current_dist = os.path.join(os.path.dirname(file),
+                                *([os.pardir] * current_dist_depth))
+    abs = os.path.abspath(current_dist)
+    dist_name = os.path.basename(abs)
+
+    for dirpath, dirnames, filenames in os.walk(abs):
+        package = (dist_name + dirpath[len(abs):]).replace("/", ".")
+        if "__init__.py" in filenames:
+            yield package
+            for filename in filenames:
+                if filename.endswith(".py") and filename != "__init__.py":
+                    yield ".".join([package, filename])[:-3]
+
+
+def import_all_modules(name=__name__, file=__file__, skip=[]):
+    for module in find_distribution_modules(name, file):
+        if module not in skip:
+            print("preimporting %r for coverage..." % (module, ))
+            try:
+                __import__(module)
+            except (ImportError, VersionMismatch, AttributeError):
+                pass
+
+
+def is_in_coverage():
+    return (os.environ.get("COVER_ALL_MODULES") or
+            "--with-coverage3" in sys.argv)
+
+
+def setup_django_env():
+    try:
+        from django.conf import settings
+    except ImportError:
+        return
+
+    if not settings.configured:
+        settings.configure(DATABASES={"default": {
+                                "ENGINE": "django.db.backends.sqlite3",
+                                "NAME": ":memory:"}},
+                           DATABASE_ENGINE="sqlite3",
+                           DATABASE_NAME=":memory:",
+                           INSTALLED_APPS=("kombu.transport.django", ))
 
 
 def setup():
     # so coverage sees all our modules.
-    for module in moduleindex:
-        print("preimporting %r for coverage..." % (module, ))
-        try:
-            __import__(module)
-        except (ImportError, VersionMismatch):
-            pass
+    setup_django_env()
+    if is_in_coverage():
+        import_all_modules()