mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-09-26 01:12:43 -05:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
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
|
|
|
|
|
|
class TestMigrateDocumentCreated(DirectoriesMixin, TestMigrations):
|
|
migrate_from = "1066_alter_workflowtrigger_schedule_offset_days"
|
|
migrate_to = "1067_alter_document_created"
|
|
|
|
def setUpBeforeMigration(self, apps):
|
|
# 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=make_aware(naive, timezone=UTC),
|
|
checksum=i,
|
|
)
|
|
|
|
def testDocumentCreatedMigrated(self):
|
|
Document = self.apps.get_model("documents", "Document")
|
|
|
|
doc = Document.objects.get(id=1)
|
|
self.assertEqual(doc.created, date(2023, 10, 1))
|