mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
e8527ba723
Cleans up some command help text and adds more control over process count for command with a Pool
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
import os
|
|
import shutil
|
|
from unittest import mock
|
|
|
|
from django.core.management import call_command
|
|
from django.test import TestCase
|
|
|
|
from documents.management.commands.document_thumbnails import _process_document
|
|
from documents.models import Document
|
|
from documents.parsers import get_default_thumbnail
|
|
from documents.tests.utils import DirectoriesMixin
|
|
from documents.tests.utils import FileSystemAssertsMixin
|
|
|
|
|
|
class TestMakeThumbnails(DirectoriesMixin, FileSystemAssertsMixin, TestCase):
|
|
def make_models(self):
|
|
self.d1 = Document.objects.create(
|
|
checksum="A",
|
|
title="A",
|
|
content="first document",
|
|
mime_type="application/pdf",
|
|
filename="test.pdf",
|
|
)
|
|
shutil.copy(
|
|
os.path.join(os.path.dirname(__file__), "samples", "simple.pdf"),
|
|
self.d1.source_path,
|
|
)
|
|
|
|
self.d2 = Document.objects.create(
|
|
checksum="B",
|
|
title="B",
|
|
content="second document",
|
|
mime_type="application/pdf",
|
|
filename="test2.pdf",
|
|
)
|
|
shutil.copy(
|
|
os.path.join(os.path.dirname(__file__), "samples", "simple.pdf"),
|
|
self.d2.source_path,
|
|
)
|
|
|
|
self.d3 = Document.objects.create(
|
|
checksum="C",
|
|
title="C",
|
|
content="third document",
|
|
mime_type="application/pdf",
|
|
filename="test3.pdf",
|
|
)
|
|
shutil.copy(
|
|
os.path.join(os.path.dirname(__file__), "samples", "password-is-test.pdf"),
|
|
self.d3.source_path,
|
|
)
|
|
|
|
def setUp(self) -> None:
|
|
super().setUp()
|
|
self.make_models()
|
|
|
|
def test_process_document(self):
|
|
self.assertIsNotFile(self.d1.thumbnail_path)
|
|
_process_document(self.d1.id)
|
|
self.assertIsFile(self.d1.thumbnail_path)
|
|
|
|
def test_process_document_password_protected(self):
|
|
self.assertIsFile(get_default_thumbnail())
|
|
self.assertIsNotFile(self.d3.thumbnail_path)
|
|
_process_document(self.d3.id)
|
|
# Ensure default thumbnail is still there
|
|
self.assertIsFile(get_default_thumbnail())
|
|
self.assertIsFile(self.d3.thumbnail_path)
|
|
|
|
@mock.patch("documents.management.commands.document_thumbnails.shutil.move")
|
|
def test_process_document_invalid_mime_type(self, m: mock.Mock):
|
|
self.d1.mime_type = "asdasdasd"
|
|
self.d1.save()
|
|
|
|
# .save() triggers filename handling
|
|
m.reset_mock()
|
|
|
|
_process_document(self.d1.id)
|
|
|
|
# Not called during processing of document
|
|
m.assert_not_called()
|
|
|
|
def test_command(self):
|
|
self.assertIsNotFile(self.d1.thumbnail_path)
|
|
self.assertIsNotFile(self.d2.thumbnail_path)
|
|
call_command("document_thumbnails", "--processes", "1")
|
|
self.assertIsFile(self.d1.thumbnail_path)
|
|
self.assertIsFile(self.d2.thumbnail_path)
|
|
|
|
def test_command_documentid(self):
|
|
self.assertIsNotFile(self.d1.thumbnail_path)
|
|
self.assertIsNotFile(self.d2.thumbnail_path)
|
|
call_command("document_thumbnails", "--processes", "1", "-d", f"{self.d1.id}")
|
|
self.assertIsFile(self.d1.thumbnail_path)
|
|
self.assertIsNotFile(self.d2.thumbnail_path)
|