]> git.ipfire.org Git - thirdparty/paperless-ngx.git/commitdiff
Resolve test timezone warnings
authorshamoon <4887959+shamoon@users.noreply.github.com>
Wed, 21 May 2025 07:51:53 +0000 (00:51 -0700)
committershamoon <4887959+shamoon@users.noreply.github.com>
Wed, 21 May 2025 07:51:53 +0000 (00:51 -0700)
src/documents/tests/test_migration_created.py

index 7aced97893f54caf1e0ab47602873017346ab4f6..89e97cbe12a551eed30dfcd71af4b00599a03316 100644 (file)
@@ -1,6 +1,10 @@
+from datetime import date
 from datetime import datetime
 from datetime import timedelta
 
+from django.utils.timezone import make_aware
+from pytz import UTC
+
 from documents.tests.utils import DirectoriesMixin
 from documents.tests.utils import TestMigrations
 
@@ -13,19 +17,12 @@ class TestMigrateDocumentCreated(DirectoriesMixin, TestMigrations):
         # create 600 documents
         for i in range(600):
             Document = apps.get_model("documents", "Document")
+            naive = datetime(2023, 10, 1, 12, 0, 0) + timedelta(days=i)
             Document.objects.create(
                 title=f"test{i}",
                 mime_type="application/pdf",
                 filename=f"file{i}.pdf",
-                created=datetime(
-                    2023,
-                    10,
-                    1,
-                    12,
-                    0,
-                    0,
-                )
-                + timedelta(days=i),
+                created=make_aware(naive, timezone=UTC),
                 checksum=i,
             )
 
@@ -33,4 +30,4 @@ class TestMigrateDocumentCreated(DirectoriesMixin, TestMigrations):
         Document = self.apps.get_model("documents", "Document")
 
         doc = Document.objects.get(id=1)
-        self.assertEqual(doc.created, datetime(2023, 10, 1, 12, 0, 0).date())
+        self.assertEqual(doc.created, date(2023, 10, 1))