Commit 9b077ee4 authored by Alex Gaynor's avatar Alex Gaynor
Browse files

[1.2.X] Converted model_inheritance_select_related tests from doctests to...

[1.2.X] Converted model_inheritance_select_related tests from doctests to unittests.  We have always been at war with doctests.  Backport of [14181].

git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@14182 bcc190cf-cafb-0310-a4f2-bffc1f526a37
parent 9ab383b3
Loading
Loading
Loading
Loading
+0 −18
Original line number Diff line number Diff line
@@ -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'

"""}
+29 −0
Original line number Diff line number Diff line
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")