mirror of
https://github.com/django/django.git
synced 2025-10-31 09:41:08 +00:00
Refs #33476 -- Reformatted code with Black.
This commit is contained in:
django-bot
committed by
Mariusz Felisiak
Mariusz Felisiak
parent
f68fa8b45d
commit
9c19aff7c7
@@ -6,12 +6,12 @@ class Building(models.Model):
|
||||
|
||||
|
||||
class Device(models.Model):
|
||||
building = models.ForeignKey('Building', models.CASCADE)
|
||||
building = models.ForeignKey("Building", models.CASCADE)
|
||||
name = models.CharField(max_length=10)
|
||||
|
||||
|
||||
class Port(models.Model):
|
||||
device = models.ForeignKey('Device', models.CASCADE)
|
||||
device = models.ForeignKey("Device", models.CASCADE)
|
||||
port_number = models.CharField(max_length=10)
|
||||
|
||||
def __str__(self):
|
||||
@@ -22,16 +22,17 @@ class Connection(models.Model):
|
||||
start = models.ForeignKey(
|
||||
Port,
|
||||
models.CASCADE,
|
||||
related_name='connection_start',
|
||||
related_name="connection_start",
|
||||
unique=True,
|
||||
)
|
||||
end = models.ForeignKey(
|
||||
Port,
|
||||
models.CASCADE,
|
||||
related_name='connection_end',
|
||||
related_name="connection_end",
|
||||
unique=True,
|
||||
)
|
||||
|
||||
|
||||
# Another non-tree hierarchy that exercises code paths similar to the above
|
||||
# example, but in a slightly different configuration.
|
||||
|
||||
@@ -60,6 +61,7 @@ class Enrollment(models.Model):
|
||||
std = models.ForeignKey(Student, models.CASCADE)
|
||||
cls = models.ForeignKey(Class, models.CASCADE)
|
||||
|
||||
|
||||
# Models for testing bug #8036.
|
||||
|
||||
|
||||
@@ -85,6 +87,7 @@ class Client(models.Model):
|
||||
class SpecialClient(Client):
|
||||
value = models.IntegerField()
|
||||
|
||||
|
||||
# Some model inheritance exercises
|
||||
|
||||
|
||||
@@ -103,6 +106,7 @@ class Item(models.Model):
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
|
||||
# Models for testing bug #19870.
|
||||
|
||||
|
||||
|
||||
@@ -1,14 +1,32 @@
|
||||
from django.test import TestCase
|
||||
|
||||
from .models import (
|
||||
A, B, Building, C, Chick, Child, Class, Client, ClientStatus, Connection,
|
||||
Country, Device, Enrollment, Hen, Item, Organizer, Person, Port,
|
||||
SpecialClient, State, Student, TUser,
|
||||
A,
|
||||
B,
|
||||
Building,
|
||||
C,
|
||||
Chick,
|
||||
Child,
|
||||
Class,
|
||||
Client,
|
||||
ClientStatus,
|
||||
Connection,
|
||||
Country,
|
||||
Device,
|
||||
Enrollment,
|
||||
Hen,
|
||||
Item,
|
||||
Organizer,
|
||||
Person,
|
||||
Port,
|
||||
SpecialClient,
|
||||
State,
|
||||
Student,
|
||||
TUser,
|
||||
)
|
||||
|
||||
|
||||
class SelectRelatedRegressTests(TestCase):
|
||||
|
||||
def test_regression_7110(self):
|
||||
"""
|
||||
Regression test for bug #7110.
|
||||
@@ -22,31 +40,34 @@ class SelectRelatedRegressTests(TestCase):
|
||||
and include some unnecessary bonus joins).
|
||||
"""
|
||||
|
||||
b = Building.objects.create(name='101')
|
||||
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)
|
||||
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')
|
||||
connections = Connection.objects.filter(
|
||||
start__device__building=b, end__device__building=b
|
||||
).order_by("id")
|
||||
self.assertEqual(
|
||||
[(c.id, str(c.start), str(c.end)) for c in connections],
|
||||
[(c1.id, 'router/4', 'switch/7'), (c2.id, 'switch/7', 'server/1')]
|
||||
[(c1.id, "router/4", "switch/7"), (c2.id, "switch/7", "server/1")],
|
||||
)
|
||||
|
||||
connections = (
|
||||
Connection.objects
|
||||
.filter(start__device__building=b, end__device__building=b)
|
||||
Connection.objects.filter(
|
||||
start__device__building=b, end__device__building=b
|
||||
)
|
||||
.select_related()
|
||||
.order_by('id')
|
||||
.order_by("id")
|
||||
)
|
||||
self.assertEqual(
|
||||
[(c.id, str(c.start), str(c.end)) for c in connections],
|
||||
[(c1.id, 'router/4', 'switch/7'), (c2.id, 'switch/7', 'server/1')]
|
||||
[(c1.id, "router/4", "switch/7"), (c2.id, "switch/7", "server/1")],
|
||||
)
|
||||
|
||||
# This final query should only have seven tables (port, device and building
|
||||
@@ -85,20 +106,26 @@ class SelectRelatedRegressTests(TestCase):
|
||||
for country before getting status.
|
||||
"""
|
||||
|
||||
Country.objects.create(name='Australia')
|
||||
active = ClientStatus.objects.create(name='active')
|
||||
client = Client.objects.create(name='client', status=active)
|
||||
Country.objects.create(name="Australia")
|
||||
active = ClientStatus.objects.create(name="active")
|
||||
client = Client.objects.create(name="client", status=active)
|
||||
|
||||
self.assertEqual(client.status, active)
|
||||
self.assertEqual(Client.objects.select_related()[0].status, active)
|
||||
self.assertEqual(Client.objects.select_related('state')[0].status, active)
|
||||
self.assertEqual(Client.objects.select_related('state', 'status')[0].status, active)
|
||||
self.assertEqual(Client.objects.select_related('state__country')[0].status, active)
|
||||
self.assertEqual(Client.objects.select_related('state__country', 'status')[0].status, active)
|
||||
self.assertEqual(Client.objects.select_related('status')[0].status, active)
|
||||
self.assertEqual(Client.objects.select_related("state")[0].status, active)
|
||||
self.assertEqual(
|
||||
Client.objects.select_related("state", "status")[0].status, active
|
||||
)
|
||||
self.assertEqual(
|
||||
Client.objects.select_related("state__country")[0].status, active
|
||||
)
|
||||
self.assertEqual(
|
||||
Client.objects.select_related("state__country", "status")[0].status, active
|
||||
)
|
||||
self.assertEqual(Client.objects.select_related("status")[0].status, active)
|
||||
|
||||
def test_multi_table_inheritance(self):
|
||||
""" Exercising select_related() with multi-table model inheritance. """
|
||||
"""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")
|
||||
@@ -115,90 +142,120 @@ class SelectRelatedRegressTests(TestCase):
|
||||
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')
|
||||
australia = Country.objects.create(name="Australia")
|
||||
active = ClientStatus.objects.create(name="active")
|
||||
|
||||
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')
|
||||
Client.objects.create(name="Brian Burke", state=wa, status=active)
|
||||
burke = (
|
||||
Client.objects.select_related("state")
|
||||
.defer("state__name")
|
||||
.get(name="Brian Burke")
|
||||
)
|
||||
|
||||
self.assertEqual(burke.name, 'Brian Burke')
|
||||
self.assertEqual(burke.state.name, 'Western Australia')
|
||||
self.assertEqual(burke.name, "Brian Burke")
|
||||
self.assertEqual(burke.state.name, "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')
|
||||
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.assertEqual(troy.name, 'Troy Buswell')
|
||||
self.assertEqual(troy.name, "Troy Buswell")
|
||||
self.assertEqual(troy.value, 42)
|
||||
self.assertEqual(troy.state.name, 'Western Australia')
|
||||
self.assertEqual(troy.state.name, "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 = (
|
||||
SpecialClient.objects.select_related("state")
|
||||
.defer("value", "state__name")
|
||||
.get(name="Troy Buswell")
|
||||
)
|
||||
|
||||
self.assertEqual(troy.name, 'Troy Buswell')
|
||||
self.assertEqual(troy.name, "Troy Buswell")
|
||||
self.assertEqual(troy.value, 42)
|
||||
self.assertEqual(troy.state.name, 'Western Australia')
|
||||
self.assertEqual(troy.state.name, "Western Australia")
|
||||
|
||||
# Also works if you use only, rather than defer
|
||||
troy = SpecialClient.objects.select_related('state').only('name', 'state').get(name='Troy Buswell')
|
||||
troy = (
|
||||
SpecialClient.objects.select_related("state")
|
||||
.only("name", "state")
|
||||
.get(name="Troy Buswell")
|
||||
)
|
||||
|
||||
self.assertEqual(troy.name, 'Troy Buswell')
|
||||
self.assertEqual(troy.name, "Troy Buswell")
|
||||
self.assertEqual(troy.value, 42)
|
||||
self.assertEqual(troy.state.name, 'Western Australia')
|
||||
self.assertEqual(troy.state.name, "Western Australia")
|
||||
|
||||
def test_null_join_promotion(self):
|
||||
australia = Country.objects.create(name='Australia')
|
||||
active = ClientStatus.objects.create(name='active')
|
||||
australia = Country.objects.create(name="Australia")
|
||||
active = ClientStatus.objects.create(name="active")
|
||||
|
||||
wa = State.objects.create(name="Western Australia", country=australia)
|
||||
bob = Client.objects.create(name='Bob', status=active)
|
||||
jack = Client.objects.create(name='Jack', status=active, state=wa)
|
||||
qs = Client.objects.filter(state=wa).select_related('state')
|
||||
bob = Client.objects.create(name="Bob", status=active)
|
||||
jack = Client.objects.create(name="Jack", status=active, state=wa)
|
||||
qs = Client.objects.filter(state=wa).select_related("state")
|
||||
with self.assertNumQueries(1):
|
||||
self.assertEqual(list(qs), [jack])
|
||||
self.assertEqual(qs[0].state, wa)
|
||||
# The select_related join wasn't promoted as there was already an
|
||||
# existing (even if trimmed) inner join to state.
|
||||
self.assertNotIn('LEFT OUTER', str(qs.query))
|
||||
qs = Client.objects.select_related('state').order_by('name')
|
||||
self.assertNotIn("LEFT OUTER", str(qs.query))
|
||||
qs = Client.objects.select_related("state").order_by("name")
|
||||
with self.assertNumQueries(1):
|
||||
self.assertEqual(list(qs), [bob, jack])
|
||||
self.assertIs(qs[0].state, None)
|
||||
self.assertEqual(qs[1].state, wa)
|
||||
# The select_related join was promoted as there is already an
|
||||
# existing join.
|
||||
self.assertIn('LEFT OUTER', str(qs.query))
|
||||
self.assertIn("LEFT OUTER", str(qs.query))
|
||||
|
||||
def test_regression_19870(self):
|
||||
hen = Hen.objects.create(name='Hen')
|
||||
Chick.objects.create(name='Chick', mother=hen)
|
||||
hen = Hen.objects.create(name="Hen")
|
||||
Chick.objects.create(name="Chick", mother=hen)
|
||||
|
||||
self.assertEqual(Chick.objects.all()[0].mother.name, 'Hen')
|
||||
self.assertEqual(Chick.objects.select_related()[0].mother.name, 'Hen')
|
||||
self.assertEqual(Chick.objects.all()[0].mother.name, "Hen")
|
||||
self.assertEqual(Chick.objects.select_related()[0].mother.name, "Hen")
|
||||
|
||||
def test_regression_10733(self):
|
||||
a = A.objects.create(name='a', lots_of_text='lots_of_text_a', a_field='a_field')
|
||||
b = B.objects.create(name='b', lots_of_text='lots_of_text_b', b_field='b_field')
|
||||
c = C.objects.create(name='c', lots_of_text='lots_of_text_c', is_published=True,
|
||||
c_a=a, c_b=b)
|
||||
results = C.objects.all().only('name', 'lots_of_text', 'c_a', 'c_b', 'c_b__lots_of_text',
|
||||
'c_a__name', 'c_b__name').select_related()
|
||||
a = A.objects.create(name="a", lots_of_text="lots_of_text_a", a_field="a_field")
|
||||
b = B.objects.create(name="b", lots_of_text="lots_of_text_b", b_field="b_field")
|
||||
c = C.objects.create(
|
||||
name="c", lots_of_text="lots_of_text_c", is_published=True, c_a=a, c_b=b
|
||||
)
|
||||
results = (
|
||||
C.objects.all()
|
||||
.only(
|
||||
"name",
|
||||
"lots_of_text",
|
||||
"c_a",
|
||||
"c_b",
|
||||
"c_b__lots_of_text",
|
||||
"c_a__name",
|
||||
"c_b__name",
|
||||
)
|
||||
.select_related()
|
||||
)
|
||||
self.assertSequenceEqual(results, [c])
|
||||
with self.assertNumQueries(0):
|
||||
qs_c = results[0]
|
||||
self.assertEqual(qs_c.name, 'c')
|
||||
self.assertEqual(qs_c.lots_of_text, 'lots_of_text_c')
|
||||
self.assertEqual(qs_c.c_b.lots_of_text, 'lots_of_text_b')
|
||||
self.assertEqual(qs_c.c_a.name, 'a')
|
||||
self.assertEqual(qs_c.c_b.name, 'b')
|
||||
self.assertEqual(qs_c.name, "c")
|
||||
self.assertEqual(qs_c.lots_of_text, "lots_of_text_c")
|
||||
self.assertEqual(qs_c.c_b.lots_of_text, "lots_of_text_b")
|
||||
self.assertEqual(qs_c.c_a.name, "a")
|
||||
self.assertEqual(qs_c.c_b.name, "b")
|
||||
|
||||
def test_regression_22508(self):
|
||||
building = Building.objects.create(name='101')
|
||||
building = Building.objects.create(name="101")
|
||||
device = Device.objects.create(name="router", building=building)
|
||||
Port.objects.create(port_number='1', device=device)
|
||||
Port.objects.create(port_number="1", device=device)
|
||||
|
||||
device = Device.objects.get()
|
||||
port = device.port_set.select_related('device__building').get()
|
||||
port = device.port_set.select_related("device__building").get()
|
||||
with self.assertNumQueries(0):
|
||||
port.device.building
|
||||
|
||||
Reference in New Issue
Block a user