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 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))