Switch several assertNumQueries to use the context manager, which is much more beautiful.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@16986 bcc190cf-cafb-0310-a4f2-bffc1f526a37

tests/regressiontests/comment_tests/tests/templatetag_tests.py

@@ -45,9 +45,8 @@ class CommentTemplateTagTests(CommentTestCase):
         self.testRenderCommentForm("{% render_comment_form for a %}")
 
     def testRenderCommentFormFromObjectWithQueryCount(self):
-        def test():
+        with self.assertNumQueries(1):
             self.testRenderCommentFormFromObject()
-        self.assertNumQueries(1, test)
 
     def verifyGetCommentCount(self, tag=None):
         t = "{% load comments %}" + (tag or "{% get_comment_count for comment_tests.article a.id as cc %}") + "{{ cc }}"

tests/regressiontests/defer_regress/tests.py

@@ -1,4 +1,4 @@
-from __future__ import absolute_import
+from __future__ import with_statement, absolute_import
 
 from operator import attrgetter
 
@@ -21,22 +21,18 @@ class DeferRegressionTest(TestCase):
         obj = Item.objects.only("name", "other_value").get(name="first")
         # Accessing "name" doesn't trigger a new database query. Accessing
         # "value" or "text" should.
-        def test():
+        with self.assertNumQueries(0):
             self.assertEqual(obj.name, "first")
             self.assertEqual(obj.other_value, 0)
-        self.assertNumQueries(0, test)
 
-        def test():
+        with self.assertNumQueries(1):
             self.assertEqual(obj.value, 42)
-        self.assertNumQueries(1, test)
 
-        def test():
+        with self.assertNumQueries(1):
             self.assertEqual(obj.text, "xyzzy")
-        self.assertNumQueries(1, test)
 
-        def test():
+        with self.assertNumQueries(0):
             self.assertEqual(obj.text, "xyzzy")
-        self.assertNumQueries(0, test)
 
         # Regression test for #10695. Make sure different instances don't
         # inadvertently share data in the deferred descriptor objects.

tests/regressiontests/forms/tests/models.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from __future__ import absolute_import
+from __future__ import with_statement, absolute_import
 
 import datetime
 
@@ -28,11 +28,10 @@ class TestTicket12510(TestCase):
         self.groups = [Group.objects.create(name=name) for name in 'abc']
 
     def test_choices_not_fetched_when_not_rendering(self):
-        def test():
+        # only one query is required to pull the model from DB
+        with self.assertNumQueries(1):
             field = ModelChoiceField(Group.objects.order_by('-name'))
             self.assertEqual('a', field.clean(self.groups[0].pk).name)
-        # only one query is required to pull the model from DB
-        self.assertNumQueries(1, test)
 
 class ModelFormCallableModelDefault(TestCase):
     def test_no_empty_option(self):

tests/regressiontests/select_related_onetoone/tests.py

@@ -1,10 +1,11 @@
-from __future__ import absolute_import
+from __future__ import with_statement, absolute_import
 
 from django.test import TestCase
 
 from .models import (User, UserProfile, UserStat, UserStatResult, StatDetails,
     AdvancedUserStat, Image, Product)
 
+
 class ReverseSelectRelatedTestCase(TestCase):
     def setUp(self):
         user = User.objects.create(username="test")
@@ -22,65 +23,56 @@ class ReverseSelectRelatedTestCase(TestCase):
         StatDetails.objects.create(base_stats=advstat, comments=250)
 
     def test_basic(self):
-        def test():
+        with self.assertNumQueries(1):
             u = User.objects.select_related("userprofile").get(username="test")
             self.assertEqual(u.userprofile.state, "KS")
-        self.assertNumQueries(1, test)
 
     def test_follow_next_level(self):
-        def test():
+        with self.assertNumQueries(1):
             u = User.objects.select_related("userstat__results").get(username="test")
             self.assertEqual(u.userstat.posts, 150)
             self.assertEqual(u.userstat.results.results, 'first results')
-        self.assertNumQueries(1, test)
 
     def test_follow_two(self):
-        def test():
+        with self.assertNumQueries(1):
             u = User.objects.select_related("userprofile", "userstat").get(username="test")
             self.assertEqual(u.userprofile.state, "KS")
             self.assertEqual(u.userstat.posts, 150)
-        self.assertNumQueries(1, test)
 
     def test_follow_two_next_level(self):
-        def test():
+        with self.assertNumQueries(1):
             u = User.objects.select_related("userstat__results", "userstat__statdetails").get(username="test")
             self.assertEqual(u.userstat.results.results, 'first results')
             self.assertEqual(u.userstat.statdetails.comments, 259)
-        self.assertNumQueries(1, test)
 
     def test_forward_and_back(self):
-        def test():
+        with self.assertNumQueries(1):
             stat = UserStat.objects.select_related("user__userprofile").get(user__username="test")
             self.assertEqual(stat.user.userprofile.state, 'KS')
             self.assertEqual(stat.user.userstat.posts, 150)
-        self.assertNumQueries(1, test)
 
     def test_back_and_forward(self):
-        def test():
+        with self.assertNumQueries(1):
             u = User.objects.select_related("userstat").get(username="test")
             self.assertEqual(u.userstat.user.username, 'test')
-        self.assertNumQueries(1, test)
 
     def test_not_followed_by_default(self):
-        def test():
+        with self.assertNumQueries(2):
             u = User.objects.select_related().get(username="test")
             self.assertEqual(u.userstat.posts, 150)
-        self.assertNumQueries(2, test)
 
     def test_follow_from_child_class(self):
-        def test():
+        with self.assertNumQueries(1):
             stat = AdvancedUserStat.objects.select_related('user', 'statdetails').get(posts=200)
             self.assertEqual(stat.statdetails.comments, 250)
             self.assertEqual(stat.user.username, 'bob')
-        self.assertNumQueries(1, test)
 
     def test_follow_inheritance(self):
-        def test():
+        with self.assertNumQueries(1):
             stat = UserStat.objects.select_related('user', 'advanceduserstat').get(posts=200)
             self.assertEqual(stat.advanceduserstat.posts, 200)
             self.assertEqual(stat.user.username, 'bob')
             self.assertEqual(stat.advanceduserstat.user.username, 'bob')
-        self.assertNumQueries(1, test)
 
     def test_nullable_relation(self):
         im = Image.objects.create(name="imag1")