Resolve test timezone warnings

This commit is contained in:
shamoon 2025-05-21 00:51:53 -07:00
parent bd5f05ff2b
commit 1583783a0b
No known key found for this signature in database

View File

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