diff options
author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
---|---|---|
committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/known_related_objects | |
parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
download | django-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/known_related_objects')
-rw-r--r-- | tests/known_related_objects/tests.py | 51 |
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) |