| from __future__ import absolute_import |
| |
| 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) |
| |
