From 9c19aff7c7561e3a82978a272ecdaad40dda5c00 Mon Sep 17 00:00:00 2001 From: django-bot Date: Thu, 3 Feb 2022 20:24:19 +0100 Subject: Refs #33476 -- Reformatted code with Black. --- tests/select_related_regress/models.py | 12 ++- tests/select_related_regress/tests.py | 187 +++++++++++++++++++++------------ 2 files changed, 130 insertions(+), 69 deletions(-) (limited to 'tests/select_related_regress') diff --git a/tests/select_related_regress/models.py b/tests/select_related_regress/models.py index 83e9321ef7..9bae754196 100644 --- a/tests/select_related_regress/models.py +++ b/tests/select_related_regress/models.py @@ -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. diff --git a/tests/select_related_regress/tests.py b/tests/select_related_regress/tests.py index 47e5e4f3c9..5ce331d332 100644 --- a/tests/select_related_regress/tests.py +++ b/tests/select_related_regress/tests.py @@ -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 -- cgit v1.2.1