diff --git a/tests/modeltests/model_inheritance/tests.py b/tests/modeltests/model_inheritance/tests.py index b94ce26700..80dd0ded5c 100644 --- a/tests/modeltests/model_inheritance/tests.py +++ b/tests/modeltests/model_inheritance/tests.py @@ -266,12 +266,16 @@ class ModelInheritanceTests(TestCase): # select_related works with fields from the parent object as if they # were a normal part of the model. old_DEBUG = settings.DEBUG - starting_queries = len(connection.queries) - settings.DEBUG = True + try: + settings.DEBUG = True + starting_queries = len(connection.queries) + ItalianRestaurant.objects.all()[0].chef + self.assertEqual(len(connection.queries) - starting_queries, 2) + + starting_queries = len(connection.queries) + ItalianRestaurant.objects.select_related("chef")[0].chef + self.assertEqual(len(connection.queries) - starting_queries, 1) + finally: + settings.DEBUG = old_DEBUG - ItalianRestaurant.objects.all()[0].chef - self.assertEqual(len(connection.queries) - starting_queries, 2) - starting_queries = len(connection.queries) - ItalianRestaurant.objects.select_related("chef")[0].chef - self.assertEqual(len(connection.queries) - starting_queries, 1) diff --git a/tests/regressiontests/views/tests/defaults.py b/tests/regressiontests/views/tests/defaults.py index 5783a824cd..ffc347106a 100644 --- a/tests/regressiontests/views/tests/defaults.py +++ b/tests/regressiontests/views/tests/defaults.py @@ -60,11 +60,16 @@ class DefaultsTests(TestCase): The 404 page should have the csrf_token available in the context """ # See ticket #14565 - for url in self.non_existing_urls: - response = self.client.get(url) - csrf_token = response.context['csrf_token'] - self.assertNotEqual(str(csrf_token), 'NOTPROVIDED') - self.assertNotEqual(str(csrf_token), '') + old_DEBUG = settings.DEBUG + try: + settings.DEBUG = False # so we get real 404, not technical + for url in self.non_existing_urls: + response = self.client.get(url) + csrf_token = response.context['csrf_token'] + self.assertNotEqual(str(csrf_token), 'NOTPROVIDED') + self.assertNotEqual(str(csrf_token), '') + finally: + settings.DEBUG = old_DEBUG def test_server_error(self): "The server_error view raises a 500 status"