summaryrefslogtreecommitdiff
path: root/tests/reverse_single_related/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/reverse_single_related/tests.py')
-rw-r--r--tests/reverse_single_related/tests.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/tests/reverse_single_related/tests.py b/tests/reverse_single_related/tests.py
new file mode 100644
index 0000000000..0c755c4db6
--- /dev/null
+++ b/tests/reverse_single_related/tests.py
@@ -0,0 +1,39 @@
+from __future__ import absolute_import
+
+from django.test import TestCase
+
+from .models import Source, Item
+
+
+class ReverseSingleRelatedTests(TestCase):
+ """
+ Regression tests for an object that cannot access a single related
+ object due to a restrictive default manager.
+ """
+
+ def test_reverse_single_related(self):
+
+ public_source = Source.objects.create(is_public=True)
+ public_item = Item.objects.create(source=public_source)
+
+ private_source = Source.objects.create(is_public=False)
+ private_item = Item.objects.create(source=private_source)
+
+ # Only one source is available via all() due to the custom default manager.
+ self.assertQuerysetEqual(
+ Source.objects.all(),
+ ["<Source: Source object>"]
+ )
+
+ self.assertEqual(public_item.source, public_source)
+
+ # Make sure that an item can still access its related source even if the default
+ # manager doesn't normally allow it.
+ self.assertEqual(private_item.source, private_source)
+
+ # If the manager is marked "use_for_related_fields", it'll get used instead
+ # of the "bare" queryset. Usually you'd define this as a property on the class,
+ # but this approximates that in a way that's easier in tests.
+ Source.objects.use_for_related_fields = True
+ private_item = Item.objects.get(pk=private_item.pk)
+ self.assertRaises(Source.DoesNotExist, lambda: private_item.source)