mirror of
https://github.com/django/django.git
synced 2024-11-18 07:26:04 +00:00
266de5f9ae
This patch iproved two major parts in Django. First, the fields.related was refactored. The main addition there was ForeignObject. Second, the ORM now handles multicolumn joins in most cases, though there are still cases that do not work correcly (split_exclude() for example). In addition there were extesive changes to how GenericRelation works. Before it was a fake m2m field, now it is a pure virtual fields and is based on ForeignObject. There is still much room for improvement. The related fields code is still somewhat confusing, and how fields are represented in model._meta should also be revisited. This patch was written mostly by Jeremy Tillman with some final polish by the committer.
339 lines
14 KiB
Python
339 lines
14 KiB
Python
import datetime
|
|
from operator import attrgetter
|
|
|
|
from .models import Country, Person, Group, Membership, Friendship, Article, ArticleTranslation
|
|
from django.test import TestCase
|
|
from django.utils.translation import activate
|
|
from django import forms
|
|
|
|
class MultiColumnFKTests(TestCase):
|
|
def setUp(self):
|
|
# Creating countries
|
|
self.usa = Country.objects.create(name="United States of America")
|
|
self.soviet_union = Country.objects.create(name="Soviet Union")
|
|
Person()
|
|
# Creating People
|
|
self.bob = Person()
|
|
self.bob.name = 'Bob'
|
|
self.bob.person_country = self.usa
|
|
self.bob.save()
|
|
self.jim = Person.objects.create(name='Jim', person_country=self.usa)
|
|
self.george = Person.objects.create(name='George', person_country=self.usa)
|
|
|
|
self.jane = Person.objects.create(name='Jane', person_country=self.soviet_union)
|
|
self.mark = Person.objects.create(name='Mark', person_country=self.soviet_union)
|
|
self.sam = Person.objects.create(name='Sam', person_country=self.soviet_union)
|
|
|
|
# Creating Groups
|
|
self.kgb = Group.objects.create(name='KGB', group_country=self.soviet_union)
|
|
self.cia = Group.objects.create(name='CIA', group_country=self.usa)
|
|
self.republican = Group.objects.create(name='Republican', group_country=self.usa)
|
|
self.democrat = Group.objects.create(name='Democrat', group_country=self.usa)
|
|
|
|
def test_get_succeeds_on_multicolumn_match(self):
|
|
# Membership objects have access to their related Person if both
|
|
# country_ids match between them
|
|
membership = Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.bob.id, group_id=self.cia.id)
|
|
|
|
person = membership.person
|
|
self.assertEqual((person.id, person.name), (self.bob.id, "Bob"))
|
|
|
|
def test_get_fails_on_multicolumn_mismatch(self):
|
|
# Membership objects returns DoesNotExist error when the there is no
|
|
# Person with the same id and country_id
|
|
membership = Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.jane.id, group_id=self.cia.id)
|
|
|
|
self.assertRaises(Person.DoesNotExist, getattr, membership, 'person')
|
|
|
|
def test_reverse_query_returns_correct_result(self):
|
|
# Creating a valid membership because it has the same country has the person
|
|
Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.bob.id, group_id=self.cia.id)
|
|
|
|
# Creating an invalid membership because it has a different country has the person
|
|
Membership.objects.create(
|
|
membership_country_id=self.soviet_union.id, person_id=self.bob.id,
|
|
group_id=self.republican.id)
|
|
|
|
self.assertQuerysetEqual(
|
|
self.bob.membership_set.all(), [
|
|
self.cia.id
|
|
],
|
|
attrgetter("group_id")
|
|
)
|
|
|
|
def test_query_filters_correctly(self):
|
|
|
|
# Creating a to valid memberships
|
|
Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.bob.id, group_id=self.cia.id)
|
|
Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.jim.id,
|
|
group_id=self.cia.id)
|
|
|
|
# Creating an invalid membership
|
|
Membership.objects.create(membership_country_id=self.soviet_union.id,
|
|
person_id=self.george.id, group_id=self.cia.id)
|
|
|
|
self.assertQuerysetEqual(
|
|
Membership.objects.filter(person__name__contains='o'), [
|
|
self.bob.id
|
|
],
|
|
attrgetter("person_id")
|
|
)
|
|
|
|
def test_reverse_query_filters_correctly(self):
|
|
|
|
timemark = datetime.datetime.utcnow()
|
|
timedelta = datetime.timedelta(days=1)
|
|
|
|
# Creating a to valid memberships
|
|
Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.bob.id,
|
|
group_id=self.cia.id, date_joined=timemark - timedelta)
|
|
Membership.objects.create(
|
|
membership_country_id=self.usa.id, person_id=self.jim.id,
|
|
group_id=self.cia.id, date_joined=timemark + timedelta)
|
|
|
|
# Creating an invalid membership
|
|
Membership.objects.create(
|
|
membership_country_id=self.soviet_union.id, person_id=self.george.id,
|
|
group_id=self.cia.id, date_joined=timemark + timedelta)
|
|
|
|
self.assertQuerysetEqual(
|
|
Person.objects.filter(membership__date_joined__gte=timemark), [
|
|
'Jim'
|
|
],
|
|
attrgetter('name')
|
|
)
|
|
|
|
def test_forward_in_lookup_filters_correctly(self):
|
|
Membership.objects.create(membership_country_id=self.usa.id, person_id=self.bob.id,
|
|
group_id=self.cia.id)
|
|
Membership.objects.create(membership_country_id=self.usa.id, person_id=self.jim.id,
|
|
group_id=self.cia.id)
|
|
|
|
# Creating an invalid membership
|
|
Membership.objects.create(
|
|
membership_country_id=self.soviet_union.id, person_id=self.george.id,
|
|
group_id=self.cia.id)
|
|
|
|
self.assertQuerysetEqual(
|
|
Membership.objects.filter(person__in=[self.george, self.jim]), [
|
|
self.jim.id,
|
|
],
|
|
attrgetter('person_id')
|
|
)
|
|
|
|
self.assertQuerysetEqual(
|
|
Membership.objects.filter(person__in=Person.objects.filter(name='Jim')), [
|
|
self.jim.id,
|
|
],
|
|
attrgetter('person_id')
|
|
)
|
|
|
|
def test_select_related_foreignkey_forward_works(self):
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
|
|
|
|
with self.assertNumQueries(1):
|
|
people = [m.person for m in Membership.objects.select_related('person')]
|
|
|
|
normal_people = [m.person for m in Membership.objects.all()]
|
|
self.assertEqual(people, normal_people)
|
|
|
|
def test_prefetch_foreignkey_forward_works(self):
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
|
|
|
|
with self.assertNumQueries(2):
|
|
people = [m.person for m in Membership.objects.prefetch_related('person')]
|
|
|
|
normal_people = [m.person for m in Membership.objects.all()]
|
|
self.assertEqual(people, normal_people)
|
|
|
|
def test_prefetch_foreignkey_reverse_works(self):
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
|
|
with self.assertNumQueries(2):
|
|
membership_sets = [list(p.membership_set.all())
|
|
for p in Person.objects.prefetch_related('membership_set')]
|
|
|
|
normal_membership_sets = [list(p.membership_set.all()) for p in Person.objects.all()]
|
|
self.assertEqual(membership_sets, normal_membership_sets)
|
|
|
|
def test_m2m_through_forward_returns_valid_members(self):
|
|
# We start out by making sure that the Group 'CIA' has no members.
|
|
self.assertQuerysetEqual(
|
|
self.cia.members.all(),
|
|
[]
|
|
)
|
|
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.cia)
|
|
|
|
# Let's check to make sure that it worked. Bob and Jim should be members of the CIA.
|
|
|
|
self.assertQuerysetEqual(
|
|
self.cia.members.all(), [
|
|
'Bob',
|
|
'Jim'
|
|
], attrgetter("name")
|
|
)
|
|
|
|
def test_m2m_through_reverse_returns_valid_members(self):
|
|
# We start out by making sure that Bob is in no groups.
|
|
self.assertQuerysetEqual(
|
|
self.bob.groups.all(),
|
|
[]
|
|
)
|
|
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob,
|
|
group=self.republican)
|
|
|
|
# Bob should be in the CIA and a Republican
|
|
self.assertQuerysetEqual(
|
|
self.bob.groups.all(), [
|
|
'CIA',
|
|
'Republican'
|
|
], attrgetter("name")
|
|
)
|
|
|
|
def test_m2m_through_forward_ignores_invalid_members(self):
|
|
# We start out by making sure that the Group 'CIA' has no members.
|
|
self.assertQuerysetEqual(
|
|
self.cia.members.all(),
|
|
[]
|
|
)
|
|
|
|
# Something adds jane to group CIA but Jane is in Soviet Union which isn't CIA's country
|
|
Membership.objects.create(membership_country=self.usa, person=self.jane, group=self.cia)
|
|
|
|
# There should still be no members in CIA
|
|
self.assertQuerysetEqual(
|
|
self.cia.members.all(),
|
|
[]
|
|
)
|
|
|
|
def test_m2m_through_reverse_ignores_invalid_members(self):
|
|
# We start out by making sure that Jane has no groups.
|
|
self.assertQuerysetEqual(
|
|
self.jane.groups.all(),
|
|
[]
|
|
)
|
|
|
|
# Something adds jane to group CIA but Jane is in Soviet Union which isn't CIA's country
|
|
Membership.objects.create(membership_country=self.usa, person=self.jane, group=self.cia)
|
|
|
|
# Jane should still not be in any groups
|
|
self.assertQuerysetEqual(
|
|
self.jane.groups.all(),
|
|
[]
|
|
)
|
|
|
|
def test_m2m_through_on_self_works(self):
|
|
self.assertQuerysetEqual(
|
|
self.jane.friends.all(),
|
|
[]
|
|
)
|
|
|
|
Friendship.objects.create(
|
|
from_friend_country=self.jane.person_country, from_friend=self.jane,
|
|
to_friend_country=self.george.person_country, to_friend=self.george)
|
|
|
|
self.assertQuerysetEqual(
|
|
self.jane.friends.all(),
|
|
['George'], attrgetter("name")
|
|
)
|
|
|
|
def test_m2m_through_on_self_ignores_mismatch_columns(self):
|
|
self.assertQuerysetEqual(self.jane.friends.all(), [])
|
|
|
|
# Note that we use ids instead of instances. This is because instances on ForeignObject
|
|
# properties will set all related field off of the given instance
|
|
Friendship.objects.create(
|
|
from_friend_id=self.jane.id, to_friend_id=self.george.id,
|
|
to_friend_country_id=self.jane.person_country_id,
|
|
from_friend_country_id=self.george.person_country_id)
|
|
|
|
self.assertQuerysetEqual(self.jane.friends.all(), [])
|
|
|
|
def test_prefetch_related_m2m_foward_works(self):
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
|
|
|
|
with self.assertNumQueries(2):
|
|
members_lists = [list(g.members.all())
|
|
for g in Group.objects.prefetch_related('members')]
|
|
|
|
normal_members_lists = [list(g.members.all()) for g in Group.objects.all()]
|
|
self.assertEqual(members_lists, normal_members_lists)
|
|
|
|
def test_prefetch_related_m2m_reverse_works(self):
|
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
|
Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
|
|
|
|
with self.assertNumQueries(2):
|
|
groups_lists = [list(p.groups.all()) for p in Person.objects.prefetch_related('groups')]
|
|
|
|
normal_groups_lists = [list(p.groups.all()) for p in Person.objects.all()]
|
|
self.assertEqual(groups_lists, normal_groups_lists)
|
|
|
|
def test_translations(self):
|
|
activate('fi')
|
|
a1 = Article.objects.create(pub_date=datetime.date.today())
|
|
at1_fi = ArticleTranslation(article=a1, lang='fi', title='Otsikko', body='Diipadaapa')
|
|
at1_fi.save()
|
|
at2_en = ArticleTranslation(article=a1, lang='en', title='Title', body='Lalalalala')
|
|
at2_en.save()
|
|
with self.assertNumQueries(1):
|
|
fetched = Article.objects.select_related('active_translation').get(
|
|
active_translation__title='Otsikko')
|
|
self.assertTrue(fetched.active_translation.title == 'Otsikko')
|
|
a2 = Article.objects.create(pub_date=datetime.date.today())
|
|
at2_fi = ArticleTranslation(article=a2, lang='fi', title='Atsikko', body='Diipadaapa',
|
|
abstract='dipad')
|
|
at2_fi.save()
|
|
a3 = Article.objects.create(pub_date=datetime.date.today())
|
|
at3_en = ArticleTranslation(article=a3, lang='en', title='A title', body='lalalalala',
|
|
abstract='lala')
|
|
at3_en.save()
|
|
# Test model initialization with active_translation field.
|
|
a3 = Article(id=a3.id, pub_date=a3.pub_date, active_translation=at3_en)
|
|
a3.save()
|
|
self.assertEqual(
|
|
list(Article.objects.filter(active_translation__abstract=None)),
|
|
[a1, a3])
|
|
self.assertEqual(
|
|
list(Article.objects.filter(active_translation__abstract=None,
|
|
active_translation__pk__isnull=False)),
|
|
[a1])
|
|
activate('en')
|
|
self.assertEqual(
|
|
list(Article.objects.filter(active_translation__abstract=None)),
|
|
[a1, a2])
|
|
|
|
class FormsTests(TestCase):
|
|
# ForeignObjects should not have any form fields, currently the user needs
|
|
# to manually deal with the foreignobject relation.
|
|
class ArticleForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Article
|
|
|
|
def test_foreign_object_form(self):
|
|
# A very crude test checking that the non-concrete fields do not get form fields.
|
|
form = FormsTests.ArticleForm()
|
|
self.assertIn('id_pub_date', form.as_table())
|
|
self.assertNotIn('active_translation', form.as_table())
|
|
form = FormsTests.ArticleForm(data={'pub_date': str(datetime.date.today())})
|
|
self.assertTrue(form.is_valid())
|
|
a = form.save()
|
|
self.assertEqual(a.pub_date, datetime.date.today())
|
|
form = FormsTests.ArticleForm(instance=a, data={'pub_date': '2013-01-01'})
|
|
a2 = form.save()
|
|
self.assertEqual(a.pk, a2.pk)
|
|
self.assertEqual(a2.pub_date, datetime.date(2013, 1, 1))
|