From 9f7cefd5059e78d75f1e1a852b51a6286ecec728 Mon Sep 17 00:00:00 2001
From: Claude Paroz <claude@2xlibre.net>
Date: Sat, 24 Nov 2012 23:43:13 +0100
Subject: [PATCH] Fixed #18417 -- Raised exception when unittest.TestCase is
 decorated with override_settings

---
 django/test/utils.py                          |  6 ++++-
 tests/regressiontests/settings_tests/tests.py | 24 ++++++++++++-------
 2 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/django/test/utils.py b/django/test/utils.py
index f10d388227..99ef6b9fa2 100644
--- a/django/test/utils.py
+++ b/django/test/utils.py
@@ -188,7 +188,11 @@ class override_settings(object):
 
     def __call__(self, test_func):
         from django.test import TransactionTestCase
-        if isinstance(test_func, type) and issubclass(test_func, TransactionTestCase):
+        if isinstance(test_func, type):
+            if not issubclass(test_func, TransactionTestCase):
+                raise Exception(
+                    "Only subclasses of Django TransactionTestCase can be decorated "
+                    "with override_settings")
             original_pre_setup = test_func._pre_setup
             original_post_teardown = test_func._post_teardown
 
diff --git a/tests/regressiontests/settings_tests/tests.py b/tests/regressiontests/settings_tests/tests.py
index aaf8bcffcf..45190daa23 100644
--- a/tests/regressiontests/settings_tests/tests.py
+++ b/tests/regressiontests/settings_tests/tests.py
@@ -6,6 +6,7 @@ from django.core.exceptions import ImproperlyConfigured
 from django.http import HttpRequest
 from django.test import TransactionTestCase, TestCase, signals
 from django.test.utils import override_settings
+from django.utils import unittest, six
 
 
 @override_settings(TEST='override')
@@ -67,11 +68,6 @@ class ClassDecoratedTestCase(ClassDecoratedTestCaseSuper):
             self.fail()
 
 
-class SettingGetter(object):
-    def __init__(self):
-        self.test = getattr(settings, 'TEST', 'undefined')
-
-
 class SettingsTests(TestCase):
     def setUp(self):
         self.testvalue = None
@@ -122,10 +118,20 @@ class SettingsTests(TestCase):
         self.assertRaises(AttributeError, getattr, settings, 'TEST')
 
     def test_class_decorator(self):
-        self.assertEqual(SettingGetter().test, 'undefined')
-        DecoratedSettingGetter = override_settings(TEST='override')(SettingGetter)
-        self.assertEqual(DecoratedSettingGetter().test, 'override')
-        self.assertRaises(AttributeError, getattr, settings, 'TEST')
+        # TransactionTestCase can be decorated by override_settings, but not ut.TestCase
+        class TransactionTestCaseSubclass(TransactionTestCase):
+            pass
+
+        class UnittestTestCaseSubclass(unittest.TestCase):
+            pass
+
+        decorated = override_settings(TEST='override')(TransactionTestCaseSubclass)
+        self.assertIsInstance(decorated, type)
+        self.assertTrue(issubclass(decorated, TransactionTestCase))
+
+        with six.assertRaisesRegex(self, Exception,
+                "Only subclasses of Django TransactionTestCase*"):
+            decorated = override_settings(TEST='override')(UnittestTestCaseSubclass)
 
     def test_signal_callback_context_manager(self):
         self.assertRaises(AttributeError, getattr, settings, 'TEST')