diff --git a/src/documents/tests/test_migration_created.py b/src/documents/tests/test_migration_created.py index 7aced9789..89e97cbe1 100644 --- a/src/documents/tests/test_migration_created.py +++ b/src/documents/tests/test_migration_created.py @@ -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))