/tests/regressiontests/custom_managers_regress/tests.py
https://code.google.com/p/mango-py/ · Python · 47 lines · 22 code · 9 blank · 16 comment · 1 complexity · 61247b5ed1e7020fb6942d97c015601b MD5 · raw file
- from django.test import TestCase
- from models import RelatedModel, RestrictedModel, OneToOneRestrictedModel
- class CustomManagersRegressTestCase(TestCase):
- def test_filtered_default_manager(self):
- """Even though the default manager filters out some records,
- we must still be able to save (particularly, save by updating
- existing records) those filtered instances. This is a
- regression test for #8990, #9527"""
- related = RelatedModel.objects.create(name="xyzzy")
- obj = RestrictedModel.objects.create(name="hidden", related=related)
- obj.name = "still hidden"
- obj.save()
- # If the hidden object wasn't seen during the save process,
- # there would now be two objects in the database.
- self.assertEqual(RestrictedModel.plain_manager.count(), 1)
- def test_delete_related_on_filtered_manager(self):
- """Deleting related objects should also not be distracted by a
- restricted manager on the related object. This is a regression
- test for #2698."""
- related = RelatedModel.objects.create(name="xyzzy")
- for name, public in (('one', True), ('two', False), ('three', False)):
- RestrictedModel.objects.create(name=name, is_public=public, related=related)
- obj = RelatedModel.objects.get(name="xyzzy")
- obj.delete()
- # All of the RestrictedModel instances should have been
- # deleted, since they *all* pointed to the RelatedModel. If
- # the default manager is used, only the public one will be
- # deleted.
- self.assertEqual(len(RestrictedModel.plain_manager.all()), 0)
- def test_delete_one_to_one_manager(self):
- # The same test case as the last one, but for one-to-one
- # models, which are implemented slightly different internally,
- # so it's a different code path.
- obj = RelatedModel.objects.create(name="xyzzy")
- OneToOneRestrictedModel.objects.create(name="foo", is_public=False, related=obj)
- obj = RelatedModel.objects.get(name="xyzzy")
- obj.delete()
- self.assertEqual(len(OneToOneRestrictedModel.plain_manager.all()), 0)