summaryrefslogtreecommitdiff
path: root/tests/known_related_objects
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/known_related_objects
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
downloaddjango-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/known_related_objects')
-rw-r--r--tests/known_related_objects/tests.py51
1 files changed, 31 insertions, 20 deletions
diff --git a/tests/known_related_objects/tests.py b/tests/known_related_objects/tests.py
index deef6560c5..0270220061 100644
--- a/tests/known_related_objects/tests.py
+++ b/tests/known_related_objects/tests.py
@@ -4,18 +4,25 @@ from .models import Organiser, Pool, PoolStyle, Tournament
class ExistingRelatedInstancesTests(TestCase):
-
@classmethod
def setUpTestData(cls):
- cls.t1 = Tournament.objects.create(name='Tourney 1')
- cls.t2 = Tournament.objects.create(name='Tourney 2')
- cls.o1 = Organiser.objects.create(name='Organiser 1')
- cls.p1 = Pool.objects.create(name='T1 Pool 1', tournament=cls.t1, organiser=cls.o1)
- cls.p2 = Pool.objects.create(name='T1 Pool 2', tournament=cls.t1, organiser=cls.o1)
- cls.p3 = Pool.objects.create(name='T2 Pool 1', tournament=cls.t2, organiser=cls.o1)
- cls.p4 = Pool.objects.create(name='T2 Pool 2', tournament=cls.t2, organiser=cls.o1)
- cls.ps1 = PoolStyle.objects.create(name='T1 Pool 2 Style', pool=cls.p2)
- cls.ps2 = PoolStyle.objects.create(name='T2 Pool 1 Style', pool=cls.p3)
+ cls.t1 = Tournament.objects.create(name="Tourney 1")
+ cls.t2 = Tournament.objects.create(name="Tourney 2")
+ cls.o1 = Organiser.objects.create(name="Organiser 1")
+ cls.p1 = Pool.objects.create(
+ name="T1 Pool 1", tournament=cls.t1, organiser=cls.o1
+ )
+ cls.p2 = Pool.objects.create(
+ name="T1 Pool 2", tournament=cls.t1, organiser=cls.o1
+ )
+ cls.p3 = Pool.objects.create(
+ name="T2 Pool 1", tournament=cls.t2, organiser=cls.o1
+ )
+ cls.p4 = Pool.objects.create(
+ name="T2 Pool 2", tournament=cls.t2, organiser=cls.o1
+ )
+ cls.ps1 = PoolStyle.objects.create(name="T1 Pool 2 Style", pool=cls.p2)
+ cls.ps2 = PoolStyle.objects.create(name="T2 Pool 1 Style", pool=cls.p3)
def test_foreign_key(self):
with self.assertNumQueries(2):
@@ -25,13 +32,17 @@ class ExistingRelatedInstancesTests(TestCase):
def test_foreign_key_prefetch_related(self):
with self.assertNumQueries(2):
- tournament = (Tournament.objects.prefetch_related('pool_set').get(pk=self.t1.pk))
+ tournament = Tournament.objects.prefetch_related("pool_set").get(
+ pk=self.t1.pk
+ )
pool = tournament.pool_set.all()[0]
self.assertIs(tournament, pool.tournament)
def test_foreign_key_multiple_prefetch(self):
with self.assertNumQueries(2):
- tournaments = list(Tournament.objects.prefetch_related('pool_set').order_by('pk'))
+ tournaments = list(
+ Tournament.objects.prefetch_related("pool_set").order_by("pk")
+ )
pool1 = tournaments[0].pool_set.all()[0]
self.assertIs(tournaments[0], pool1.tournament)
pool2 = tournaments[1].pool_set.all()[0]
@@ -85,25 +96,25 @@ class ExistingRelatedInstancesTests(TestCase):
def test_one_to_one_select_related(self):
with self.assertNumQueries(1):
- style = PoolStyle.objects.select_related('pool').get(pk=self.ps1.pk)
+ style = PoolStyle.objects.select_related("pool").get(pk=self.ps1.pk)
pool = style.pool
self.assertIs(style, pool.poolstyle)
def test_one_to_one_multi_select_related(self):
with self.assertNumQueries(1):
- poolstyles = list(PoolStyle.objects.select_related('pool').order_by('pk'))
+ poolstyles = list(PoolStyle.objects.select_related("pool").order_by("pk"))
self.assertIs(poolstyles[0], poolstyles[0].pool.poolstyle)
self.assertIs(poolstyles[1], poolstyles[1].pool.poolstyle)
def test_one_to_one_prefetch_related(self):
with self.assertNumQueries(2):
- style = PoolStyle.objects.prefetch_related('pool').get(pk=self.ps1.pk)
+ style = PoolStyle.objects.prefetch_related("pool").get(pk=self.ps1.pk)
pool = style.pool
self.assertIs(style, pool.poolstyle)
def test_one_to_one_multi_prefetch_related(self):
with self.assertNumQueries(2):
- poolstyles = list(PoolStyle.objects.prefetch_related('pool').order_by('pk'))
+ poolstyles = list(PoolStyle.objects.prefetch_related("pool").order_by("pk"))
self.assertIs(poolstyles[0], poolstyles[0].pool.poolstyle)
self.assertIs(poolstyles[1], poolstyles[1].pool.poolstyle)
@@ -115,24 +126,24 @@ class ExistingRelatedInstancesTests(TestCase):
def test_reverse_one_to_one_select_related(self):
with self.assertNumQueries(1):
- pool = Pool.objects.select_related('poolstyle').get(pk=self.p2.pk)
+ pool = Pool.objects.select_related("poolstyle").get(pk=self.p2.pk)
style = pool.poolstyle
self.assertIs(pool, style.pool)
def test_reverse_one_to_one_prefetch_related(self):
with self.assertNumQueries(2):
- pool = Pool.objects.prefetch_related('poolstyle').get(pk=self.p2.pk)
+ pool = Pool.objects.prefetch_related("poolstyle").get(pk=self.p2.pk)
style = pool.poolstyle
self.assertIs(pool, style.pool)
def test_reverse_one_to_one_multi_select_related(self):
with self.assertNumQueries(1):
- pools = list(Pool.objects.select_related('poolstyle').order_by('pk'))
+ pools = list(Pool.objects.select_related("poolstyle").order_by("pk"))
self.assertIs(pools[1], pools[1].poolstyle.pool)
self.assertIs(pools[2], pools[2].poolstyle.pool)
def test_reverse_one_to_one_multi_prefetch_related(self):
with self.assertNumQueries(2):
- pools = list(Pool.objects.prefetch_related('poolstyle').order_by('pk'))
+ pools = list(Pool.objects.prefetch_related("poolstyle").order_by("pk"))
self.assertIs(pools[1], pools[1].poolstyle.pool)
self.assertIs(pools[2], pools[2].poolstyle.pool)