29 lines
782 B
Python
29 lines
782 B
Python
|
from __future__ import with_statement
|
||
|
|
|
||
|
|
from django.test import TestCase
|
||
|
|
|
||
|
|
from models import Person
|
||
|
|
|
||
|
|
|
||
|
|
class AssertNumQueriesTests(TestCase):
|
||
|
|
def test_simple(self):
|
||
|
|
with self.assertNumQueries(0):
|
||
|
|
pass
|
||
|
|
|
||
|
|
with self.assertNumQueries(1):
|
||
|
|
Person.objects.count()
|
||
|
|
|
||
|
|
with self.assertNumQueries(2):
|
||
|
|
Person.objects.count()
|
||
|
|
Person.objects.count()
|
||
|
|
|
||
|
|
def test_failure(self):
|
||
|
|
with self.assertRaises(AssertionError) as exc_info:
|
||
|
|
with self.assertNumQueries(2):
|
||
|
|
Person.objects.count()
|
||
|
self.assertIn("1 queries executed, 2 expected", str(exc_info.exception))
|
||
|
|
|
||
|
|
with self.assertRaises(TypeError):
|
||
|
|
with self.assertNumQueries(4000):
|
||
|
|
raise TypeError
|