+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
# 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,
)
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))