62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
from django.test import TestCase
|
|
|
|
from .models import (
|
|
CompetingTeam,
|
|
Event,
|
|
Group,
|
|
IndividualCompetitor,
|
|
Membership,
|
|
Person,
|
|
)
|
|
|
|
|
|
class MultiTableTests(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.alice = Person.objects.create(name="Alice")
|
|
cls.bob = Person.objects.create(name="Bob")
|
|
cls.chris = Person.objects.create(name="Chris")
|
|
cls.dan = Person.objects.create(name="Dan")
|
|
cls.team_alpha = Group.objects.create(name="Alpha")
|
|
Membership.objects.create(person=cls.alice, group=cls.team_alpha)
|
|
Membership.objects.create(person=cls.bob, group=cls.team_alpha)
|
|
cls.event = Event.objects.create(name="Exposition Match")
|
|
IndividualCompetitor.objects.create(event=cls.event, person=cls.chris)
|
|
IndividualCompetitor.objects.create(event=cls.event, person=cls.dan)
|
|
CompetingTeam.objects.create(event=cls.event, team=cls.team_alpha)
|
|
|
|
def test_m2m_query(self):
|
|
result = self.event.teams.all()
|
|
self.assertCountEqual(result, [self.team_alpha])
|
|
|
|
def test_m2m_reverse_query(self):
|
|
result = self.chris.event_set.all()
|
|
self.assertCountEqual(result, [self.event])
|
|
|
|
def test_m2m_query_proxied(self):
|
|
result = self.event.special_people.all()
|
|
self.assertCountEqual(result, [self.chris, self.dan])
|
|
|
|
def test_m2m_reverse_query_proxied(self):
|
|
result = self.chris.special_event_set.all()
|
|
self.assertCountEqual(result, [self.event])
|
|
|
|
def test_m2m_prefetch_proxied(self):
|
|
result = Event.objects.filter(name="Exposition Match").prefetch_related(
|
|
"special_people"
|
|
)
|
|
with self.assertNumQueries(2):
|
|
self.assertCountEqual(result, [self.event])
|
|
self.assertEqual(
|
|
sorted(p.name for p in result[0].special_people.all()), ["Chris", "Dan"]
|
|
)
|
|
|
|
def test_m2m_prefetch_reverse_proxied(self):
|
|
result = Person.objects.filter(name="Dan").prefetch_related("special_event_set")
|
|
with self.assertNumQueries(2):
|
|
self.assertCountEqual(result, [self.dan])
|
|
self.assertEqual(
|
|
[event.name for event in result[0].special_event_set.all()],
|
|
["Exposition Match"],
|
|
)
|