[soc2010/test-refactor] Added skipIfDBEngine() decorator to django.test

PaulMcMillan · PaulMcMillan · commit e0bd235ffa78 · 2010-07-02T19:06:53.000Z
git-svn-id: http://code.djangoproject.com/svn/django/branches/soc2010/test-refactor@13416 bcc190cf-cafb-0310-a4f2-bffc1f526a37
diff --git a/django/test/__init__.py b/django/test/__init__.py
@@ -4,3 +4,4 @@
 
 from django.test.client import Client
 from django.test.testcases import TestCase, TransactionTestCase
+from django.test.utils import skipIfDBEngine
diff --git a/django/test/utils.py b/django/test/utils.py
@@ -2,9 +2,11 @@
 from django.conf import settings
 from django.core import mail
 from django.core.mail.backends import locmem
+from django.db import DEFAULT_DB_ALIAS
 from django.test import signals
 from django.template import Template
 from django.utils.translation import deactivate
+from django.utils.unittest import skipIf
 
 class ContextList(list):
     """A wrapper that provides direct key access to context items contained
@@ -77,3 +79,14 @@ def get_runner(settings):
     test_module = __import__(test_module_name, {}, {}, test_path[-1])
     test_runner = getattr(test_module, test_path[-1])
     return test_runner
+
+def skipIfDBEngine(engine, reason=None):
+    """
+    Decorator to skip tests on a given database engine.
+
+    Note that you can pass a single engine or an iterable here
+    """
+    if not reason:
+        reason = "not supported on this database"
+    return skipIf(settings.DATABASES[DEFAULT_DB_ALIAS]['ENGINE'] in engine,
+                  reason)

Original file line number	Diff line number	Diff line change
`@@ -4,3 +4,4 @@`
`4`	`4`
`5`	`5`	`from django.test.client import Client`
`6`	`6`	`from django.test.testcases import TestCase, TransactionTestCase`
	`7`	`+from django.test.utils import skipIfDBEngine`