mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
more test.
This commit is contained in:
jonaswinkler
parent
fad3df1e39
commit
bb814da95b
135
src/documents/tests/test_management.py
Normal file
135
src/documents/tests/test_management.py
Normal file
@ -0,0 +1,135 @@
|
|||||||
|
import hashlib
|
||||||
|
import tempfile
|
||||||
|
import filecmp
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
from pathlib import Path
|
||||||
|
from unittest import mock
|
||||||
|
|
||||||
|
from django.test import TestCase, override_settings
|
||||||
|
|
||||||
|
|
||||||
|
from django.core.management import call_command
|
||||||
|
|
||||||
|
from documents.file_handling import generate_filename
|
||||||
|
from documents.management.commands.document_archiver import handle_document
|
||||||
|
from documents.models import Document
|
||||||
|
from documents.tests.utils import DirectoriesMixin
|
||||||
|
|
||||||
|
|
||||||
|
sample_file = os.path.join(os.path.dirname(__file__), "samples", "simple.pdf")
|
||||||
|
|
||||||
|
|
||||||
|
class TestArchiver(DirectoriesMixin, TestCase):
|
||||||
|
|
||||||
|
def make_models(self):
|
||||||
|
return Document.objects.create(checksum="A", title="A", content="first document", mime_type="application/pdf")
|
||||||
|
|
||||||
|
def test_archiver(self):
|
||||||
|
|
||||||
|
doc = self.make_models()
|
||||||
|
shutil.copy(sample_file, os.path.join(self.dirs.originals_dir, f"{doc.id:07}.pdf"))
|
||||||
|
|
||||||
|
call_command('document_archiver')
|
||||||
|
|
||||||
|
def test_handle_document(self):
|
||||||
|
|
||||||
|
doc = self.make_models()
|
||||||
|
shutil.copy(sample_file, os.path.join(self.dirs.originals_dir, f"{doc.id:07}.pdf"))
|
||||||
|
|
||||||
|
handle_document(doc.pk)
|
||||||
|
|
||||||
|
doc = Document.objects.get(id=doc.id)
|
||||||
|
|
||||||
|
self.assertIsNotNone(doc.checksum)
|
||||||
|
self.assertTrue(os.path.isfile(doc.archive_path))
|
||||||
|
self.assertTrue(os.path.isfile(doc.source_path))
|
||||||
|
self.assertTrue(filecmp.cmp(sample_file, doc.source_path))
|
||||||
|
|
||||||
|
|
||||||
|
class TestDecryptDocuments(TestCase):
|
||||||
|
|
||||||
|
@override_settings(
|
||||||
|
ORIGINALS_DIR=os.path.join(os.path.dirname(__file__), "samples", "originals"),
|
||||||
|
THUMBNAIL_DIR=os.path.join(os.path.dirname(__file__), "samples", "thumb"),
|
||||||
|
PASSPHRASE="test",
|
||||||
|
PAPERLESS_FILENAME_FORMAT=None
|
||||||
|
)
|
||||||
|
@mock.patch("documents.management.commands.decrypt_documents.input")
|
||||||
|
def test_decrypt(self, m):
|
||||||
|
|
||||||
|
media_dir = tempfile.mkdtemp()
|
||||||
|
originals_dir = os.path.join(media_dir, "documents", "originals")
|
||||||
|
thumb_dir = os.path.join(media_dir, "documents", "thumbnails")
|
||||||
|
os.makedirs(originals_dir, exist_ok=True)
|
||||||
|
os.makedirs(thumb_dir, exist_ok=True)
|
||||||
|
|
||||||
|
override_settings(
|
||||||
|
ORIGINALS_DIR=originals_dir,
|
||||||
|
THUMBNAIL_DIR=thumb_dir,
|
||||||
|
PASSPHRASE="test"
|
||||||
|
).enable()
|
||||||
|
|
||||||
|
doc = Document.objects.create(checksum="9c9691e51741c1f4f41a20896af31770", title="wow", filename="0000002.pdf.gpg", mime_type="application/pdf", storage_type=Document.STORAGE_TYPE_GPG)
|
||||||
|
|
||||||
|
shutil.copy(os.path.join(os.path.dirname(__file__), "samples", "documents", "originals", "0000002.pdf.gpg"), os.path.join(originals_dir, "0000002.pdf.gpg"))
|
||||||
|
shutil.copy(os.path.join(os.path.dirname(__file__), "samples", "documents", "thumbnails", f"0000002.png.gpg"), os.path.join(thumb_dir, f"{doc.id:07}.png.gpg"))
|
||||||
|
|
||||||
|
call_command('decrypt_documents')
|
||||||
|
|
||||||
|
doc.refresh_from_db()
|
||||||
|
|
||||||
|
self.assertEqual(doc.storage_type, Document.STORAGE_TYPE_UNENCRYPTED)
|
||||||
|
self.assertEqual(doc.filename, "0000002.pdf")
|
||||||
|
self.assertTrue(os.path.isfile(os.path.join(originals_dir, "0000002.pdf")))
|
||||||
|
self.assertTrue(os.path.isfile(doc.source_path))
|
||||||
|
self.assertTrue(os.path.isfile(os.path.join(thumb_dir, f"{doc.id:07}.png")))
|
||||||
|
self.assertTrue(os.path.isfile(doc.thumbnail_path))
|
||||||
|
|
||||||
|
with doc.source_file as f:
|
||||||
|
checksum = hashlib.md5(f.read()).hexdigest()
|
||||||
|
self.assertEqual(checksum, doc.checksum)
|
||||||
|
|
||||||
|
|
||||||
|
class TestMakeIndex(TestCase):
|
||||||
|
|
||||||
|
@mock.patch("documents.management.commands.document_index.index_reindex")
|
||||||
|
def test_reindex(self, m):
|
||||||
|
call_command("document_index", "reindex")
|
||||||
|
m.assert_called_once()
|
||||||
|
|
||||||
|
@mock.patch("documents.management.commands.document_index.index_optimize")
|
||||||
|
def test_optimize(self, m):
|
||||||
|
call_command("document_index", "optimize")
|
||||||
|
m.assert_called_once()
|
||||||
|
|
||||||
|
|
||||||
|
class TestRenamer(DirectoriesMixin, TestCase):
|
||||||
|
|
||||||
|
def test_rename(self):
|
||||||
|
doc = Document.objects.create(title="test", mime_type="application/pdf")
|
||||||
|
doc.filename = generate_filename(doc)
|
||||||
|
doc.save()
|
||||||
|
|
||||||
|
Path(doc.source_path).touch()
|
||||||
|
|
||||||
|
old_source_path = doc.source_path
|
||||||
|
|
||||||
|
with override_settings(PAPERLESS_FILENAME_FORMAT="{title}"):
|
||||||
|
call_command("document_renamer")
|
||||||
|
|
||||||
|
doc2 = Document.objects.get(id=doc.id)
|
||||||
|
|
||||||
|
self.assertEqual(doc2.filename, "test.pdf")
|
||||||
|
self.assertFalse(os.path.isfile(old_source_path))
|
||||||
|
self.assertFalse(os.path.isfile(doc.source_path))
|
||||||
|
self.assertTrue(os.path.isfile(doc2.source_path))
|
||||||
|
|
||||||
|
|
||||||
|
class TestCreateClassifier(TestCase):
|
||||||
|
|
||||||
|
@mock.patch("documents.management.commands.document_create_classifier.train_classifier")
|
||||||
|
def test_create_classifier(self, m):
|
||||||
|
call_command("document_create_classifier")
|
||||||
|
|
||||||
|
m.assert_called_once()
|
||||||
@ -1,40 +0,0 @@
|
|||||||
import filecmp
|
|
||||||
import os
|
|
||||||
import shutil
|
|
||||||
|
|
||||||
from django.core.management import call_command
|
|
||||||
from django.test import TestCase
|
|
||||||
|
|
||||||
from documents.management.commands.document_archiver import handle_document
|
|
||||||
from documents.models import Document
|
|
||||||
from documents.tests.utils import DirectoriesMixin
|
|
||||||
|
|
||||||
|
|
||||||
sample_file = os.path.join(os.path.dirname(__file__), "samples", "simple.pdf")
|
|
||||||
|
|
||||||
|
|
||||||
class TestArchiver(DirectoriesMixin, TestCase):
|
|
||||||
|
|
||||||
def make_models(self):
|
|
||||||
return Document.objects.create(checksum="A", title="A", content="first document", mime_type="application/pdf")
|
|
||||||
|
|
||||||
def test_archiver(self):
|
|
||||||
|
|
||||||
doc = self.make_models()
|
|
||||||
shutil.copy(sample_file, os.path.join(self.dirs.originals_dir, f"{doc.id:07}.pdf"))
|
|
||||||
|
|
||||||
call_command('document_archiver')
|
|
||||||
|
|
||||||
def test_handle_document(self):
|
|
||||||
|
|
||||||
doc = self.make_models()
|
|
||||||
shutil.copy(sample_file, os.path.join(self.dirs.originals_dir, f"{doc.id:07}.pdf"))
|
|
||||||
|
|
||||||
handle_document(doc.pk)
|
|
||||||
|
|
||||||
doc = Document.objects.get(id=doc.id)
|
|
||||||
|
|
||||||
self.assertIsNotNone(doc.checksum)
|
|
||||||
self.assertTrue(os.path.isfile(doc.archive_path))
|
|
||||||
self.assertTrue(os.path.isfile(doc.source_path))
|
|
||||||
self.assertTrue(filecmp.cmp(sample_file, doc.source_path))
|
|
||||||
@ -1,57 +0,0 @@
|
|||||||
import hashlib
|
|
||||||
import json
|
|
||||||
import os
|
|
||||||
import shutil
|
|
||||||
import tempfile
|
|
||||||
from unittest import mock
|
|
||||||
|
|
||||||
from django.core.management import call_command
|
|
||||||
from django.test import TestCase, override_settings
|
|
||||||
|
|
||||||
from documents.management.commands import document_exporter
|
|
||||||
from documents.models import Document, Tag, DocumentType, Correspondent
|
|
||||||
|
|
||||||
|
|
||||||
class TestDecryptDocuments(TestCase):
|
|
||||||
|
|
||||||
@override_settings(
|
|
||||||
ORIGINALS_DIR=os.path.join(os.path.dirname(__file__), "samples", "originals"),
|
|
||||||
THUMBNAIL_DIR=os.path.join(os.path.dirname(__file__), "samples", "thumb"),
|
|
||||||
PASSPHRASE="test",
|
|
||||||
PAPERLESS_FILENAME_FORMAT=None
|
|
||||||
)
|
|
||||||
@mock.patch("documents.management.commands.decrypt_documents.input")
|
|
||||||
def test_decrypt(self, m):
|
|
||||||
|
|
||||||
media_dir = tempfile.mkdtemp()
|
|
||||||
originals_dir = os.path.join(media_dir, "documents", "originals")
|
|
||||||
thumb_dir = os.path.join(media_dir, "documents", "thumbnails")
|
|
||||||
os.makedirs(originals_dir, exist_ok=True)
|
|
||||||
os.makedirs(thumb_dir, exist_ok=True)
|
|
||||||
|
|
||||||
override_settings(
|
|
||||||
ORIGINALS_DIR=originals_dir,
|
|
||||||
THUMBNAIL_DIR=thumb_dir,
|
|
||||||
PASSPHRASE="test"
|
|
||||||
).enable()
|
|
||||||
|
|
||||||
doc = Document.objects.create(checksum="9c9691e51741c1f4f41a20896af31770", title="wow", filename="0000002.pdf.gpg", mime_type="application/pdf", storage_type=Document.STORAGE_TYPE_GPG)
|
|
||||||
|
|
||||||
shutil.copy(os.path.join(os.path.dirname(__file__), "samples", "documents", "originals", "0000002.pdf.gpg"), os.path.join(originals_dir, "0000002.pdf.gpg"))
|
|
||||||
shutil.copy(os.path.join(os.path.dirname(__file__), "samples", "documents", "thumbnails", f"0000002.png.gpg"), os.path.join(thumb_dir, f"{doc.id:07}.png.gpg"))
|
|
||||||
|
|
||||||
call_command('decrypt_documents')
|
|
||||||
|
|
||||||
doc.refresh_from_db()
|
|
||||||
|
|
||||||
self.assertEqual(doc.storage_type, Document.STORAGE_TYPE_UNENCRYPTED)
|
|
||||||
self.assertEqual(doc.filename, "0000002.pdf")
|
|
||||||
self.assertTrue(os.path.isfile(os.path.join(originals_dir, "0000002.pdf")))
|
|
||||||
self.assertTrue(os.path.isfile(doc.source_path))
|
|
||||||
self.assertTrue(os.path.isfile(os.path.join(thumb_dir, f"{doc.id:07}.png")))
|
|
||||||
self.assertTrue(os.path.isfile(doc.thumbnail_path))
|
|
||||||
|
|
||||||
with doc.source_file as f:
|
|
||||||
checksum = hashlib.md5(f.read()).hexdigest()
|
|
||||||
self.assertEqual(checksum, doc.checksum)
|
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user