/tests/regressiontests/m2m_regress/tests.py

https://code.google.com/p/mango-py/ · Python · 75 lines · 45 code · 20 blank · 10 comment · 0 complexity · d7f4ef172bfd33a61cde86f7a774e8db MD5 · raw file

  1. from django.core.exceptions import FieldError
  2. from django.test import TestCase
  3. from models import (SelfRefer, Tag, TagCollection, Entry, SelfReferChild,
  4. SelfReferChildSibling, Worksheet)
  5. class M2MRegressionTests(TestCase):
  6. def test_multiple_m2m(self):
  7. # Multiple m2m references to model must be distinguished when
  8. # accessing the relations through an instance attribute.
  9. s1 = SelfRefer.objects.create(name='s1')
  10. s2 = SelfRefer.objects.create(name='s2')
  11. s3 = SelfRefer.objects.create(name='s3')
  12. s1.references.add(s2)
  13. s1.related.add(s3)
  14. e1 = Entry.objects.create(name='e1')
  15. t1 = Tag.objects.create(name='t1')
  16. t2 = Tag.objects.create(name='t2')
  17. e1.topics.add(t1)
  18. e1.related.add(t2)
  19. self.assertQuerysetEqual(s1.references.all(), ["<SelfRefer: s2>"])
  20. self.assertQuerysetEqual(s1.related.all(), ["<SelfRefer: s3>"])
  21. self.assertQuerysetEqual(e1.topics.all(), ["<Tag: t1>"])
  22. self.assertQuerysetEqual(e1.related.all(), ["<Tag: t2>"])
  23. def test_internal_related_name_not_in_error_msg(self):
  24. # The secret internal related names for self-referential many-to-many
  25. # fields shouldn't appear in the list when an error is made.
  26. self.assertRaisesRegexp(FieldError,
  27. "Choices are: id, name, references, related, selfreferchild, selfreferchildsibling$",
  28. lambda: SelfRefer.objects.filter(porcupine='fred')
  29. )
  30. def test_m2m_inheritance_symmetry(self):
  31. # Test to ensure that the relationship between two inherited models
  32. # with a self-referential m2m field maintains symmetry
  33. sr_child = SelfReferChild(name="Hanna")
  34. sr_child.save()
  35. sr_sibling = SelfReferChildSibling(name="Beth")
  36. sr_sibling.save()
  37. sr_child.related.add(sr_sibling)
  38. self.assertQuerysetEqual(sr_child.related.all(), ["<SelfRefer: Beth>"])
  39. self.assertQuerysetEqual(sr_sibling.related.all(), ["<SelfRefer: Hanna>"])
  40. def test_m2m_pk_field_type(self):
  41. # Regression for #11311 - The primary key for models in a m2m relation
  42. # doesn't have to be an AutoField
  43. w = Worksheet(id='abc')
  44. w.save()
  45. w.delete()
  46. def test_add_m2m_with_base_class(self):
  47. # Regression for #11956 -- You can add an object to a m2m with the
  48. # base class without causing integrity errors
  49. t1 = Tag.objects.create(name='t1')
  50. t2 = Tag.objects.create(name='t2')
  51. c1 = TagCollection.objects.create(name='c1')
  52. c1.tags = [t1,t2]
  53. c1 = TagCollection.objects.get(name='c1')
  54. self.assertQuerysetEqual(c1.tags.all(), ["<Tag: t1>", "<Tag: t2>"])
  55. self.assertQuerysetEqual(t1.tag_collections.all(), ["<TagCollection: c1>"])