From f997c81472b96a1cf48a1a19a4fe974683455a50 Mon Sep 17 00:00:00 2001 From: Hasan Ramezani Date: Thu, 23 Sep 2021 18:40:54 +0200 Subject: Fixed #33127 -- Added error messages on | and & operators with combined querysets. --- tests/queries/test_qs_combinators.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'tests/queries') diff --git a/tests/queries/test_qs_combinators.py b/tests/queries/test_qs_combinators.py index 351a401902..89d08ef6db 100644 --- a/tests/queries/test_qs_combinators.py +++ b/tests/queries/test_qs_combinators.py @@ -441,3 +441,24 @@ class QuerySetSetOperationTests(TestCase): with self.subTest(combinator=combinator): with self.assertRaisesMessage(NotSupportedError, msg % combinator): getattr(qs, combinator)(qs).get(num=2) + + def test_operator_on_combined_qs_error(self): + qs = Number.objects.all() + msg = 'Cannot use %s operator with combined queryset.' + combinators = ['union'] + if connection.features.supports_select_difference: + combinators.append('difference') + if connection.features.supports_select_intersection: + combinators.append('intersection') + operators = [ + ('|', operator.or_), + ('&', operator.and_), + ] + for combinator in combinators: + combined_qs = getattr(qs, combinator)(qs) + for operator_, operator_func in operators: + with self.subTest(combinator=combinator): + with self.assertRaisesMessage(TypeError, msg % operator_): + operator_func(qs, combined_qs) + with self.assertRaisesMessage(TypeError, msg % operator_): + operator_func(combined_qs, qs) -- cgit v1.2.1