|
1 | | -from datetime import datetime |
| 1 | +from datetime import UTC, datetime |
2 | 2 | from decimal import Decimal |
3 | 3 | from math import pi |
4 | 4 |
|
|
15 | 15 | ) |
16 | 16 | from django.db.models.functions import Collate |
17 | 17 | from django.db.models.lookups import GreaterThan |
18 | | -from django.test import SimpleTestCase, TestCase, override_settings, skipUnlessDBFeature |
19 | | -from django.utils import timezone |
| 18 | +from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature |
20 | 19 |
|
21 | 20 | from .models import ( |
22 | 21 | Article, |
@@ -77,15 +76,14 @@ def test_null_db_default(self): |
77 | 76 | self.assertIsNone(obj2.null) |
78 | 77 |
|
79 | 78 | @skipUnlessDBFeature("supports_expression_defaults") |
80 | | - @override_settings(USE_TZ=True) |
81 | 79 | def test_db_default_function(self): |
82 | 80 | m = DBDefaultsFunction.objects.create() |
83 | 81 | expected_num_queries = ( |
84 | 82 | 0 if connection.features.can_return_columns_from_insert else 4 |
85 | 83 | ) |
86 | 84 | with self.assertNumQueries(expected_num_queries): |
87 | 85 | self.assertAlmostEqual(m.number, pi) |
88 | | - self.assertEqual(m.year, timezone.now().year) |
| 86 | + self.assertEqual(m.year, datetime.now(UTC).year) |
89 | 87 | self.assertAlmostEqual(m.added, pi + 4.5) |
90 | 88 | self.assertEqual(m.multiple_subfunctions, 4.5) |
91 | 89 |
|
@@ -175,13 +173,12 @@ def test_bulk_create_mixed_db_defaults(self): |
175 | 173 | self.assertCountEqual(headlines, ["Default headline", "Something else"]) |
176 | 174 |
|
177 | 175 | @skipUnlessDBFeature("supports_expression_defaults") |
178 | | - @override_settings(USE_TZ=True) |
179 | 176 | def test_bulk_create_mixed_db_defaults_function(self): |
180 | 177 | instances = [DBDefaultsFunction(), DBDefaultsFunction(year=2000)] |
181 | 178 | DBDefaultsFunction.objects.bulk_create(instances) |
182 | 179 |
|
183 | 180 | years = DBDefaultsFunction.objects.values_list("year", flat=True) |
184 | | - self.assertCountEqual(years, [2000, timezone.now().year]) |
| 181 | + self.assertCountEqual(years, [2000, datetime.now(UTC).year]) |
185 | 182 |
|
186 | 183 | @skipUnlessDBFeature("supports_expression_defaults") |
187 | 184 | def test_full_clean(self): |
|
0 commit comments