Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Migrated select_related_regress doctests. Thanks to Stephan Jaekel.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13925 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit 30228b6459ce07bbaca554dfb4f2217da29e4667 1 parent 4747fe2
@freakboy3742 freakboy3742 authored
View
123 tests/regressiontests/select_related_regress/models.py
@@ -84,126 +84,3 @@ class Item(models.Model):
def __unicode__(self):
return self.name
-
-__test__ = {'API_TESTS': """
-Regression test for bug #7110. When using select_related(), we must query the
-Device and Building tables using two different aliases (each) in order to
-differentiate the start and end Connection fields. The net result is that both
-the "connections = ..." queries here should give the same results without
-pulling in more than the absolute minimum number of tables (history has
-shown that it's easy to make a mistake in the implementation and include some
-unnecessary bonus joins).
-
->>> b=Building.objects.create(name='101')
->>> dev1=Device.objects.create(name="router", building=b)
->>> dev2=Device.objects.create(name="switch", building=b)
->>> dev3=Device.objects.create(name="server", building=b)
->>> port1=Port.objects.create(port_number='4',device=dev1)
->>> port2=Port.objects.create(port_number='7',device=dev2)
->>> port3=Port.objects.create(port_number='1',device=dev3)
->>> c1=Connection.objects.create(start=port1, end=port2)
->>> c2=Connection.objects.create(start=port2, end=port3)
-
->>> connections=Connection.objects.filter(start__device__building=b, end__device__building=b).order_by('id')
->>> [(c.id, unicode(c.start), unicode(c.end)) for c in connections]
-[(1, u'router/4', u'switch/7'), (2, u'switch/7', u'server/1')]
-
->>> connections=Connection.objects.filter(start__device__building=b, end__device__building=b).select_related().order_by('id')
->>> [(c.id, unicode(c.start), unicode(c.end)) for c in connections]
-[(1, u'router/4', u'switch/7'), (2, u'switch/7', u'server/1')]
-
-# This final query should only join seven tables (port, device and building
-# twice each, plus connection once).
->>> connections.query.count_active_tables()
-7
-
-Regression test for bug #8106. Same sort of problem as the previous test, but
-this time there are more extra tables to pull in as part of the
-select_related() and some of them could potentially clash (so need to be kept
-separate).
-
->>> us = TUser.objects.create(name="std")
->>> usp = Person.objects.create(user=us)
->>> uo = TUser.objects.create(name="org")
->>> uop = Person.objects.create(user=uo)
->>> s = Student.objects.create(person = usp)
->>> o = Organizer.objects.create(person = uop)
->>> c = Class.objects.create(org=o)
->>> e = Enrollment.objects.create(std=s, cls=c)
-
->>> e_related = Enrollment.objects.all().select_related()[0]
->>> e_related.std.person.user.name
-u"std"
->>> e_related.cls.org.person.user.name
-u"org"
-
-Regression test for bug #8036: the first related model in the tests below
-("state") is empty and we try to select the more remotely related
-state__country. The regression here was not skipping the empty column results
-for country before getting status.
-
->>> australia = Country.objects.create(name='Australia')
->>> active = ClientStatus.objects.create(name='active')
->>> client = Client.objects.create(name='client', status=active)
-
->>> client.status
-<ClientStatus: ClientStatus object>
->>> Client.objects.select_related()[0].status
-<ClientStatus: ClientStatus object>
->>> Client.objects.select_related('state')[0].status
-<ClientStatus: ClientStatus object>
->>> Client.objects.select_related('state', 'status')[0].status
-<ClientStatus: ClientStatus object>
->>> Client.objects.select_related('state__country')[0].status
-<ClientStatus: ClientStatus object>
->>> Client.objects.select_related('state__country', 'status')[0].status
-<ClientStatus: ClientStatus object>
->>> Client.objects.select_related('status')[0].status
-<ClientStatus: ClientStatus object>
-
-Exercising select_related() with multi-table model inheritance.
->>> c1 = Child.objects.create(name="child1", value=42)
->>> _ = Item.objects.create(name="item1", child=c1)
->>> _ = Item.objects.create(name="item2")
->>> Item.objects.select_related("child").order_by("name")
-[<Item: item1>, <Item: item2>]
-
-# Regression for #12851 - Deferred fields are used correctly if you
-# select_related a subset of fields.
->>> wa = State.objects.create(name="Western Australia", country=australia)
->>> _ = Client.objects.create(name='Brian Burke', state=wa, status=active)
->>> burke = Client.objects.select_related('state').defer('state__name').get(name='Brian Burke')
->>> burke.name
-u'Brian Burke'
->>> burke.state.name
-u'Western Australia'
-
-# Still works if we're dealing with an inherited class
->>> _ = SpecialClient.objects.create(name='Troy Buswell', state=wa, status=active, value=42)
->>> troy = SpecialClient.objects.select_related('state').defer('state__name').get(name='Troy Buswell')
->>> troy.name
-u'Troy Buswell'
->>> troy.value
-42
->>> troy.state.name
-u'Western Australia'
-
-# Still works if we defer an attribute on the inherited class
->>> troy = SpecialClient.objects.select_related('state').defer('value', 'state__name').get(name='Troy Buswell')
->>> troy.name
-u'Troy Buswell'
->>> troy.value
-42
->>> troy.state.name
-u'Western Australia'
-
-# Also works if you use only, rather than defer
->>> troy = SpecialClient.objects.select_related('state').only('name').get(name='Troy Buswell')
->>> troy.name
-u'Troy Buswell'
->>> troy.value
-42
->>> troy.state.name
-u'Western Australia'
-
-"""}
View
134 tests/regressiontests/select_related_regress/tests.py
@@ -0,0 +1,134 @@
+from django.test import TestCase
+from regressiontests.select_related_regress.models import *
+
+class SelectRelatedRegressTests(TestCase):
+
+ def test_regression_7110(self):
+ """
+ Regression test for bug #7110.
+
+ When using select_related(), we must query the
+ Device and Building tables using two different aliases (each) in order to
+ differentiate the start and end Connection fields. The net result is that
+ both the "connections = ..." queries here should give the same results
+ without pulling in more than the absolute minimum number of tables
+ (history has shown that it's easy to make a mistake in the implementation
+ and include some unnecessary bonus joins).
+ """
+
+ b=Building.objects.create(name='101')
+ dev1=Device.objects.create(name="router", building=b)
+ dev2=Device.objects.create(name="switch", building=b)
+ dev3=Device.objects.create(name="server", building=b)
+ port1=Port.objects.create(port_number='4',device=dev1)
+ port2=Port.objects.create(port_number='7',device=dev2)
+ port3=Port.objects.create(port_number='1',device=dev3)
+ c1=Connection.objects.create(start=port1, end=port2)
+ c2=Connection.objects.create(start=port2, end=port3)
+
+ connections=Connection.objects.filter(start__device__building=b, end__device__building=b).order_by('id')
+ self.assertEquals([(c.id, unicode(c.start), unicode(c.end)) for c in connections],
+ [(1, u'router/4', u'switch/7'), (2, u'switch/7', u'server/1')])
+
+ connections=Connection.objects.filter(start__device__building=b, end__device__building=b).select_related().order_by('id')
+ self.assertEquals([(c.id, unicode(c.start), unicode(c.end)) for c in connections],
+ [(1, u'router/4', u'switch/7'), (2, u'switch/7', u'server/1')])
+
+ # This final query should only join seven tables (port, device and building
+ # twice each, plus connection once).
+ self.assertEquals(connections.query.count_active_tables(), 7)
+
+
+ def test_regression_8106(self):
+ """
+ Regression test for bug #8106.
+
+ Same sort of problem as the previous test, but this time there are
+ more extra tables to pull in as part of the select_related() and some
+ of them could potentially clash (so need to be kept separate).
+ """
+
+ us = TUser.objects.create(name="std")
+ usp = Person.objects.create(user=us)
+ uo = TUser.objects.create(name="org")
+ uop = Person.objects.create(user=uo)
+ s = Student.objects.create(person = usp)
+ o = Organizer.objects.create(person = uop)
+ c = Class.objects.create(org=o)
+ e = Enrollment.objects.create(std=s, cls=c)
+
+ e_related = Enrollment.objects.all().select_related()[0]
+ self.assertEquals(e_related.std.person.user.name, u"std")
+ self.assertEquals(e_related.cls.org.person.user.name, u"org")
+
+ def test_regression_8036(self):
+ """
+ Regression test for bug #8036
+
+ the first related model in the tests below
+ ("state") is empty and we try to select the more remotely related
+ state__country. The regression here was not skipping the empty column results
+ for country before getting status.
+ """
+
+ australia = Country.objects.create(name='Australia')
+ active = ClientStatus.objects.create(name='active')
+ client = Client.objects.create(name='client', status=active)
+
+ self.assertEquals(client.status, active)
+ self.assertEquals(Client.objects.select_related()[0].status, active)
+ self.assertEquals(Client.objects.select_related('state')[0].status, active)
+ self.assertEquals(Client.objects.select_related('state', 'status')[0].status, active)
+ self.assertEquals(Client.objects.select_related('state__country')[0].status, active)
+ self.assertEquals(Client.objects.select_related('state__country', 'status')[0].status, active)
+ self.assertEquals(Client.objects.select_related('status')[0].status, active)
+
+ def test_multi_table_inheritance(self):
+ """ Exercising select_related() with multi-table model inheritance. """
+ c1 = Child.objects.create(name="child1", value=42)
+ i1 = Item.objects.create(name="item1", child=c1)
+ i2 = Item.objects.create(name="item2")
+
+ self.assertQuerysetEqual(
+ Item.objects.select_related("child").order_by("name"),
+ ["<Item: item1>", "<Item: item2>"]
+ )
+
+ def test_regression_12851(self):
+ """
+ Regression for #12851
+
+ Deferred fields are used correctly if you select_related a subset
+ of fields.
+ """
+ australia = Country.objects.create(name='Australia')
+ active = ClientStatus.objects.create(name='active')
+
+ wa = State.objects.create(name="Western Australia", country=australia)
+ c1 = Client.objects.create(name='Brian Burke', state=wa, status=active)
+ burke = Client.objects.select_related('state').defer('state__name').get(name='Brian Burke')
+
+ self.assertEquals(burke.name, u'Brian Burke')
+ self.assertEquals(burke.state.name, u'Western Australia')
+
+ # Still works if we're dealing with an inherited class
+ sc1 = SpecialClient.objects.create(name='Troy Buswell', state=wa, status=active, value=42)
+ troy = SpecialClient.objects.select_related('state').defer('state__name').get(name='Troy Buswell')
+
+ self.assertEquals(troy.name, u'Troy Buswell')
+ self.assertEquals(troy.value, 42)
+ self.assertEquals(troy.state.name, u'Western Australia')
+
+ # Still works if we defer an attribute on the inherited class
+ troy = SpecialClient.objects.select_related('state').defer('value', 'state__name').get(name='Troy Buswell')
+
+ self.assertEquals(troy.name, u'Troy Buswell')
+ self.assertEquals(troy.value, 42)
+ self.assertEquals(troy.state.name, u'Western Australia')
+
+ # Also works if you use only, rather than defer
+ troy = SpecialClient.objects.select_related('state').only('name').get(name='Troy Buswell')
+
+ self.assertEquals(troy.name, u'Troy Buswell')
+ self.assertEquals(troy.value, 42)
+ self.assertEquals(troy.state.name, u'Western Australia')
Please sign in to comment.
Something went wrong with that request. Please try again.