mirror of https://github.com/django/django.git
144 lines
5.0 KiB
Python
144 lines
5.0 KiB
Python
from django.core.exceptions import FieldError
|
|
from django.test import TestCase
|
|
|
|
from .models import (
|
|
Entry,
|
|
Line,
|
|
Post,
|
|
RegressionModelSplit,
|
|
SelfRefer,
|
|
SelfReferChild,
|
|
SelfReferChildSibling,
|
|
Tag,
|
|
TagCollection,
|
|
Worksheet,
|
|
)
|
|
|
|
|
|
class M2MRegressionTests(TestCase):
|
|
def test_multiple_m2m(self):
|
|
# Multiple m2m references to model must be distinguished when
|
|
# accessing the relations through an instance attribute.
|
|
|
|
s1 = SelfRefer.objects.create(name="s1")
|
|
s2 = SelfRefer.objects.create(name="s2")
|
|
s3 = SelfRefer.objects.create(name="s3")
|
|
s1.references.add(s2)
|
|
s1.related.add(s3)
|
|
|
|
e1 = Entry.objects.create(name="e1")
|
|
t1 = Tag.objects.create(name="t1")
|
|
t2 = Tag.objects.create(name="t2")
|
|
|
|
e1.topics.add(t1)
|
|
e1.related.add(t2)
|
|
|
|
self.assertSequenceEqual(s1.references.all(), [s2])
|
|
self.assertSequenceEqual(s1.related.all(), [s3])
|
|
|
|
self.assertSequenceEqual(e1.topics.all(), [t1])
|
|
self.assertSequenceEqual(e1.related.all(), [t2])
|
|
|
|
def test_internal_related_name_not_in_error_msg(self):
|
|
# The secret internal related names for self-referential many-to-many
|
|
# fields shouldn't appear in the list when an error is made.
|
|
msg = (
|
|
"Choices are: id, name, references, related, selfreferchild, "
|
|
"selfreferchildsibling"
|
|
)
|
|
with self.assertRaisesMessage(FieldError, msg):
|
|
SelfRefer.objects.filter(porcupine="fred")
|
|
|
|
def test_m2m_inheritance_symmetry(self):
|
|
# Test to ensure that the relationship between two inherited models
|
|
# with a self-referential m2m field maintains symmetry
|
|
|
|
sr_child = SelfReferChild(name="Hanna")
|
|
sr_child.save()
|
|
|
|
sr_sibling = SelfReferChildSibling(name="Beth")
|
|
sr_sibling.save()
|
|
sr_child.related.add(sr_sibling)
|
|
|
|
self.assertSequenceEqual(sr_child.related.all(), [sr_sibling.selfrefer_ptr])
|
|
self.assertSequenceEqual(sr_sibling.related.all(), [sr_child.selfrefer_ptr])
|
|
|
|
def test_m2m_pk_field_type(self):
|
|
# Regression for #11311 - The primary key for models in a m2m relation
|
|
# doesn't have to be an AutoField
|
|
|
|
w = Worksheet(id="abc")
|
|
w.save()
|
|
w.delete()
|
|
|
|
def test_create_copy_with_m2m(self):
|
|
t1 = Tag.objects.create(name="t1")
|
|
Entry.objects.create(name="e1")
|
|
entry = Entry.objects.first()
|
|
entry.topics.set([t1])
|
|
old_topics = entry.topics.all()
|
|
entry.pk = None
|
|
entry._state.adding = True
|
|
entry.save()
|
|
entry.topics.set(old_topics)
|
|
entry = Entry.objects.get(pk=entry.pk)
|
|
self.assertCountEqual(entry.topics.all(), old_topics)
|
|
self.assertSequenceEqual(entry.topics.all(), [t1])
|
|
|
|
def test_add_m2m_with_base_class(self):
|
|
# Regression for #11956 -- You can add an object to a m2m with the
|
|
# base class without causing integrity errors
|
|
|
|
t1 = Tag.objects.create(name="t1")
|
|
t2 = Tag.objects.create(name="t2")
|
|
|
|
c1 = TagCollection.objects.create(name="c1")
|
|
c1.tags.set([t1, t2])
|
|
c1 = TagCollection.objects.get(name="c1")
|
|
|
|
self.assertCountEqual(c1.tags.all(), [t1, t2])
|
|
self.assertCountEqual(t1.tag_collections.all(), [c1])
|
|
|
|
def test_manager_class_caching(self):
|
|
e1 = Entry.objects.create()
|
|
e2 = Entry.objects.create()
|
|
t1 = Tag.objects.create()
|
|
t2 = Tag.objects.create()
|
|
|
|
# Get same manager twice in a row:
|
|
self.assertIs(t1.entry_set.__class__, t1.entry_set.__class__)
|
|
self.assertIs(e1.topics.__class__, e1.topics.__class__)
|
|
|
|
# Get same manager for different instances
|
|
self.assertIs(e1.topics.__class__, e2.topics.__class__)
|
|
self.assertIs(t1.entry_set.__class__, t2.entry_set.__class__)
|
|
|
|
def test_m2m_abstract_split(self):
|
|
# Regression for #19236 - an abstract class with a 'split' method
|
|
# causes a TypeError in add_lazy_relation
|
|
m1 = RegressionModelSplit(name="1")
|
|
m1.save()
|
|
|
|
def test_assigning_invalid_data_to_m2m_doesnt_clear_existing_relations(self):
|
|
t1 = Tag.objects.create(name="t1")
|
|
t2 = Tag.objects.create(name="t2")
|
|
c1 = TagCollection.objects.create(name="c1")
|
|
c1.tags.set([t1, t2])
|
|
|
|
with self.assertRaisesMessage(TypeError, "'int' object is not iterable"):
|
|
c1.tags.set(7)
|
|
|
|
c1.refresh_from_db()
|
|
self.assertSequenceEqual(c1.tags.order_by("name"), [t1, t2])
|
|
|
|
def test_multiple_forwards_only_m2m(self):
|
|
# Regression for #24505 - Multiple ManyToManyFields to same "to"
|
|
# model with related_name set to '+'.
|
|
foo = Line.objects.create(name="foo")
|
|
bar = Line.objects.create(name="bar")
|
|
post = Post.objects.create()
|
|
post.primary_lines.add(foo)
|
|
post.secondary_lines.add(bar)
|
|
self.assertSequenceEqual(post.primary_lines.all(), [foo])
|
|
self.assertSequenceEqual(post.secondary_lines.all(), [bar])
|