tests

src/documents/tests/test_management_archiver.py

@@ -0,0 +1,28 @@
+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()