diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/prefetch_related/tests.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/prefetch_related/tests.py b/tests/prefetch_related/tests.py index 35c2a1c6b8..226dfb8ab9 100644 --- a/tests/prefetch_related/tests.py +++ b/tests/prefetch_related/tests.py @@ -815,15 +815,17 @@ class CustomPrefetchTests(TestCase): self.traverse_qs(list(houses), [['occupants', 'houses', 'main_room']]) def test_values_queryset(self): - msg = 'Prefetch querysets cannot use raw() and values().' + msg = 'Prefetch querysets cannot use raw(), values(), and values_list().' with self.assertRaisesMessage(ValueError, msg): Prefetch('houses', House.objects.values('pk')) + with self.assertRaisesMessage(ValueError, msg): + Prefetch('houses', House.objects.values_list('pk')) # That error doesn't affect managers with custom ModelIterable subclasses self.assertIs(Teacher.objects_custom.all()._iterable_class, ModelIterableSubclass) Prefetch('teachers', Teacher.objects_custom.all()) def test_raw_queryset(self): - msg = 'Prefetch querysets cannot use raw() and values().' + msg = 'Prefetch querysets cannot use raw(), values(), and values_list().' with self.assertRaisesMessage(ValueError, msg): Prefetch('houses', House.objects.raw('select pk from house')) |