Converted model_inheritance_select_related tests from doctests to unittests. We have always been at war with doctests.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@14181 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Alex Gaynor 2010-10-12 02:09:13 +00:00
parent 977d588dc2
commit ceef628c19
2 changed files with 29 additions and 18 deletions

View File

@ -27,21 +27,3 @@ class Person(models.Model):
def __unicode__(self):
return self.name
__test__ = {'API_TESTS':"""
Regression test for #7246
>>> r1 = Restaurant.objects.create(name="Nobu", serves_sushi=True, serves_steak=False)
>>> r2 = Restaurant.objects.create(name="Craft", serves_sushi=False, serves_steak=True)
>>> p1 = Person.objects.create(name="John", favorite_restaurant=r1)
>>> p2 = Person.objects.create(name="Jane", favorite_restaurant=r2)
>>> Person.objects.order_by('name').select_related()
[<Person: Jane>, <Person: John>]
>>> jane = Person.objects.order_by('name').select_related('favorite_restaurant')[0]
>>> jane.favorite_restaurant.name
u'Craft'
"""}

View File

@ -0,0 +1,29 @@
from operator import attrgetter
from django.test import TestCase
from models import Restaurant, Person
class ModelInheritanceSelectRelatedTests(TestCase):
def test_inherited_select_related(self):
# Regression test for #7246
r1 = Restaurant.objects.create(
name="Nobu", serves_sushi=True, serves_steak=False
)
r2 = Restaurant.objects.create(
name="Craft", serves_sushi=False, serves_steak=True
)
p1 = Person.objects.create(name="John", favorite_restaurant=r1)
p2 = Person.objects.create(name="Jane", favorite_restaurant=r2)
self.assertQuerysetEqual(
Person.objects.order_by("name").select_related(), [
"Jane",
"John",
],
attrgetter("name")
)
jane = Person.objects.order_by("name").select_related("favorite_restaurant")[0]
self.assertEqual(jane.favorite_restaurant.name, "Craft")