From 402e6d292fd8aa5dfd49166e80098122950053f7 Mon Sep 17 00:00:00 2001 From: Hasan Ramezani Date: Thu, 11 Jul 2019 21:13:36 +0200 Subject: Fixed #30602 -- Made Extract raise ValueError when using unsupported lookups for DurationField. --- tests/db_functions/datetime/test_extract_trunc.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'tests') diff --git a/tests/db_functions/datetime/test_extract_trunc.py b/tests/db_functions/datetime/test_extract_trunc.py index 2b7ee5befd..f2f75ee620 100644 --- a/tests/db_functions/datetime/test_extract_trunc.py +++ b/tests/db_functions/datetime/test_extract_trunc.py @@ -273,6 +273,13 @@ class DateFunctionTests(TestCase): with self.assertRaisesMessage(ValueError, msg): list(DTModel.objects.annotate(extracted=Extract('duration', 'second'))) + def test_extract_duration_unsupported_lookups(self): + msg = "Cannot extract component '%s' from DurationField 'duration'." + for lookup in ('year', 'iso_year', 'month', 'week', 'week_day', 'quarter'): + with self.subTest(lookup): + with self.assertRaisesMessage(ValueError, msg % lookup): + DTModel.objects.annotate(extracted=Extract('duration', lookup)) + def test_extract_year_func(self): start_datetime = datetime(2015, 6, 15, 14, 30, 50, 321) end_datetime = datetime(2016, 6, 15, 14, 10, 50, 123) -- cgit v1.2.1