diff options
Diffstat (limited to 'tests/generic_relations/tests.py')
| -rw-r--r-- | tests/generic_relations/tests.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/generic_relations/tests.py b/tests/generic_relations/tests.py index 7c0db95908..683efaddfb 100644 --- a/tests/generic_relations/tests.py +++ b/tests/generic_relations/tests.py @@ -564,6 +564,19 @@ class GenericRelationsTests(TestCase): for tag in tags: self.assertSequenceEqual(tag.content_object.tags.all(), [tag]) + def test_prefetch_related_custom_object_id(self): + tiger = Animal.objects.create(common_name='tiger') + cheetah = Animal.objects.create(common_name='cheetah') + Comparison.objects.create( + first_obj=cheetah, other_obj=tiger, comparative='faster', + ) + Comparison.objects.create( + first_obj=tiger, other_obj=cheetah, comparative='cooler', + ) + qs = Comparison.objects.prefetch_related('first_obj__comparisons') + for comparison in qs: + self.assertSequenceEqual(comparison.first_obj.comparisons.all(), [comparison]) + class ProxyRelatedModelTest(TestCase): def test_default_behavior(self): |