mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2026-05-20 21:45:23 +00:00
jonaswinkler
29 lines
1001 B
Python
29 lines
1001 B
Python
import os
|
|
import shutil
|
|
from unittest import mock
|
|
|
|
from django.core.management import call_command
|
|
from django.test import TestCase
|
|
|
|
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):
|
|
self.d1 = Document.objects.create(checksum="A", title="A", content="first document", pk=1, mime_type="application/pdf")
|
|
#self.d2 = Document.objects.create(checksum="B", title="B", content="second document")
|
|
#self.d3 = Document.objects.create(checksum="C", title="C", content="unrelated document")
|
|
|
|
@mock.patch("documents.management.commands.document_archiver.handle_document")
|
|
def test_archiver(self, m):
|
|
|
|
shutil.copy(sample_file, os.path.join(self.dirs.originals_dir, "0000001.pdf"))
|
|
self.make_models()
|
|
|
|
call_command('document_archiver')
|
|
|
|
m.assert_called()
|